numpy.moveaxis

numpy.moveaxis(a, source, destination)

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

其他轴保持其原始顺序。

参数：

a：np.ndarray

其轴应重新排序的数组。

源：int或int的序列

要移动的轴的原始位置。这些必须是唯一的。

destination：int或int的序列

每个原始轴的目标位置。这些也必须是唯一的。

返回：

result：np.ndarray

移动轴的数组。此数组是输入数组的视图。

也可以看看

transpose

允许数组的尺寸。

swapaxes

交换数组的两个轴。

例子

>>> x = np.zeros((3, 4, 5))
>>> np.moveaxis(x, 0, -1).shape
(4, 5, 3)
>>> np.moveaxis(x, -1, 0).shape
(5, 3, 4)

这些都达到相同的结果：

>>> np.transpose(x).shape
(5, 4, 3)
>>> np.swapaxis(x, 0, -1).shape
(5, 4, 3)
>>> np.moveaxis(x, [0, 1], [-1, -2]).shape
(5, 4, 3)
>>> np.moveaxis(x, [0, 1, 2], [-1, -2, -3]).shape
(5, 4, 3)