numpy.moveaxis

numpy.moveaxis(a, source, destination)

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

其他轴保持其原始顺序。

版本 1.11.0 中新增。

参数:

anp.ndarray

要重新排序轴的数组。

sourceint 或 int 序列

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

destinationint 或 int 序列

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

返回值:

resultnp.ndarray

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

另请参见

transpose

置换数组的维度。

swapaxes

交换数组的两个轴。

示例

>>> import numpy as np
>>> 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)