NumPy 函数

NumPy rollaxis() 函数向后滚动指定的轴，直到它位于给定位置。使用该函数的语法如下:

语法

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

参数

a必填。 指定输入数组（ndarray）。
axis必填。 指定要滚动的轴。其他轴的位置相对于彼此不会改变。
start可选。 当start <= axis时，轴会回滚，直到到达此位置。当启动> axis时，轴会滚动直到位于该位置之前。默认值 0 会导致"完整"滚动。

返回值

返回a的视图.

示例:

在下面的示例中，rollaxis() 函数用于将轴向后滚动给定数组。 p>

import numpy as np

#创建形状为(3,4,5,6)的数组
arr = np.ones((3,4,5,6))
print(arr.shape)

#向后滚动轴
print(np.rollaxis(arr, 3, 1).shape)
print(np.rollaxis(arr, 2).shape)
print(np.rollaxis(arr, 1, 4).shape)

上述代码的输出将是:

(3, 4, 5, 6)
(3, 6, 4, 5)
(5, 3, 4, 6)
(3, 5, 6, 4)

示例:

再考虑一个示例来理解此函数。

import numpy as np

#创建形状为(2,2,2)的数组
arr = np.arange(8).reshape(2,2,2)
print("The original array:")
print(arr)

#向后滚动轴
print("\nThe array after rollaxis():")
print(np.rollaxis(arr, 1, 2))

上述代码的输出将是:

The original array:
[[[0 1]
  [2 3]]

 [[4 5]
  [6 7]]]

The array after rollaxis():
[[[0 1]
  [2 3]]

 [[4 5]
  [6 7]]]