numpy.permute_dims

numpy.permute_dims(a, axes=None)

返回一个轴被转置的数组。

对于一维数组，这将返回原始数组的未更改视图，因为转置向量只是相同的向量。要将一维数组转换为二维列向量，必须添加一个额外的维度，例如 np.atleast_2d(a).T 可以实现此目的， a[:, np.newaxis] 也可以。对于二维数组，这是标准矩阵转置。对于 n 维数组，如果指定了 axes，则它们的顺序指示如何置换轴（参见示例）。如果未提供 axes，则 transpose(a).shape == a.shape[::-1]。

参数：

aarray_like

输入数组。

axes整数元组或列表，可选

如果指定，它必须是一个元组或列表，其中包含 [0, 1, …, N-1] 的排列，其中 N 是 a 的轴数。也可以使用负索引来指定轴。返回数组的第 i 个轴将对应于输入的编号为 axes[i] 的轴。如果未指定，则默认为 range(a.ndim)[::-1]，这会反转轴的顺序。

返回值：

pndarray

其轴被置换的 a。只要可能，就会返回视图。

参见

ndarray.transpose

等效方法。

moveaxis

将数组的轴移动到新的位置。

argsort

返回将对数组进行排序的索引。

备注

使用 transpose(a, argsort(axes)) 来反转使用 axes 关键字参数时张量的转置。

示例

>>> import numpy as np
>>> a = np.array([[1, 2], [3, 4]])
>>> a
array([[1, 2],
       [3, 4]])
>>> np.transpose(a)
array([[1, 3],
       [2, 4]])

>>> a = np.array([1, 2, 3, 4])
>>> a
array([1, 2, 3, 4])
>>> np.transpose(a)
array([1, 2, 3, 4])

>>> a = np.ones((1, 2, 3))
>>> np.transpose(a, (1, 0, 2)).shape
(2, 1, 3)

>>> a = np.ones((2, 3, 4, 5))
>>> np.transpose(a).shape
(5, 4, 3, 2)

>>> a = np.arange(3*4*5).reshape((3, 4, 5))
>>> np.transpose(a, (-1, 0, -2)).shape
(5, 3, 4)