numpy.rollaxis() function | Python

numpy.rollaxis() function roll the specified axis backwards, until it lies in a given position.

Syntax : numpy.rollaxis(arr, axis, start=0)
Parameters :
arr : [ndarray] Input array.
axis : [int] The axis to roll backwards. The positions of the other axes do not change relative to one another.
start : [int, optional] The axis is rolled until it lies before this position. The default, 0, results in a “complete” roll.
Return : [ndarray] In earlier NumPy versions, arr is returned only if the order of the axes is changed, otherwise the input array is returned. For NumPy >= 1.10.0, a view of arr is always returned.

Code #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)

Output :

(1, 4, 2, 3)

Code #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)

Output :

(3, 1, 2, 4)