numpy.moveaxis

numpy.moveaxis(a, source, destination)[源代码]

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

其他轴保持其原始顺序。

1.11.0 新版功能.

参数
anp.ndarray

其轴应重新排序的数组。

sourceint或int序列

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

destinationint或int序列

每个原始轴的目标位置。它们也必须是独一无二的。

返回
resultnp.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.swapaxes(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)