NumPy - swapaxes() function
The NumPy swapaxes() function is used to interchange two axes of an array.
Syntax
numpy.swapaxes(a, axis1, axis2)
Parameters
a |
Required. Specify the input array (array_like). |
axis1 |
Required. Specify the first axis. |
axis2 |
Required. Specify the second axis. |
Return Value
If a is an ndarray, then a view of a is returned; otherwise a new array is created.
Example:
In the example below, swapaxes() function is used to interchange two axes of an array.
import numpy as np x = np.array([[1,2,3]]) y = np.array([[[1,2,3]], [[4,5,6]]]) #print x print("x =") print(x) #swap axis of x print("\nnp.swapaxes(x,0,1) =") print(np.swapaxes(x,0,1)) #print y print("\ny =") print(y) #swap axis of y print("\nnp.swapaxes(y,0,2) =") print(np.swapaxes(y,0,2))
The output of the above code will be:
x = [[1 2 3]] np.swapaxes(x,0,1) = [[1] [2] [3]] y = [[[1 2 3]] [[4 5 6]]] np.swapaxes(y,0,2) = [[[1 4]] [[2 5]] [[3 6]]]
❮ NumPy - Functions