Python numpy.rollaxis()函数

numpy.rollaxis()函数将指定的轴向后滚动，直到它位于一个给定的位置。

语法: numpy.rollaxis(arr, axis, start=0)
参数 :
arr : [ndarray] 输入阵列。
axis : [int] 要向后滚动的轴。其他轴的位置不会相对于彼此改变。
start : [int, optional] 轴被滚动直到它位于这个位置之前。默认值为0，导致一个 “完整 “的滚动。
返回 : [ndarray] 在早期的NumPy版本中，只有当轴的顺序改变时才会返回arr，否则会返回输入数组。对于NumPy >= 1.10.0，arr的视图总是被返回。

代码#1：

# Python program explaining
# numpy.rollaxis() function
  
# importing numpy as geek 
import numpy as geek
  
arr = geek.ones((1, 2, 3, 4))
  
gfg = geek.rollaxis(arr, 3, 1).shape
  
print (gfg)

输出 :

(1, 4, 2, 3)

代码#2：

# Python program explaining
# numpy.rollaxis() function
  
# importing numpy as geek 
import numpy as geek
  
arr = geek.ones((1, 2, 3, 4))
  
gfg = geek.rollaxis(arr, 2).shape
  
print (gfg)