numpy.moveaxis

原文:https://docs.scipy.org/doc/numpy/reference/generated/numpy.moveaxis.html

译者:飞龙 UsyiyiCN

校对:(虚位以待)

numpy.moveaxis(a, source, destination)[source]

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

其他轴保持其原始顺序。

参数:

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)