Python numpy rollaxis用法及代码示例

本文简要介绍 python 语言中 numpy.rollaxis 的用法。

用法:

numpy.rollaxis(a, axis, start=0)

向后滚动指定的轴，直到它位于给定位置。

为了向后兼容，继续支持此函数，但您应该更喜欢 moveaxis 。 moveaxis 函数是在 NumPy 1.11 中添加的。

参数：

a： ndarray

输入数组。

axis： int

要滚动的轴。其他轴的位置相对于彼此不会改变。

start： 整数，可选

当 start <= axis 时，轴会回滚，直到它位于该位置。当 start > axis 时，轴会滚动，直到它位于该位置之前。默认值 0 会产生 “complete” 滚动。下表说明了如何解释 start 的负值：

start	标准化start
-(arr.ndim+1)	提高AxisError
-arr.ndim	0
⋮	⋮
-1	arr.ndim-1
0	0
⋮	⋮
arr.ndim	arr.ndim
arr.ndim + 1	提高AxisError

返回：

res： ndarray

对于 NumPy >= 1.10.0 始终返回 a 的视图。对于早期的 NumPy 版本，仅当轴的顺序更改时才返回 a 的视图，否则返回输入数组。

例子：

>>> a = np.ones((3,4,5,6))
>>> np.rollaxis(a, 3, 1).shape
(3, 6, 4, 5)
>>> np.rollaxis(a, 2).shape
(5, 3, 4, 6)
>>> np.rollaxis(a, 1, 4).shape
(3, 5, 6, 4)