numpy.swapaxes() in Python
numpy.swapaxes() in Python
The numpy.swapaxes() function interchanges the two specified axes of the given array.
Syntax
numpy.swapaxes(a, axis1, axis2)
Parameter
The numpy.swapaxes() method consists of three parameters, which are as follows:
a : This parameter represents the Input array.
axis1 : It represents the first axis.
axis2 : It signifies the second axis.
Return
This function returns the view of parameter ‘a’ if ‘a’ is a ndarray, else a new array is created.
Example 1
# Python Program explaining
# numpy.swapaxes() function
import numpy as np
array = np.arange(8).reshape(2,2,2)
print ('Original array:')
print (array,"\n")
# swaping the numbers between axis 1 and axis 2 (along width)
print ('The array after applying the swapaxes() function:')
print (np.swapaxes(array, 2, 1))
Output
Original array: [[[0 1] [2 3]] [[4 5] [6 7]]] The array after applying the swapaxes() function: [[[0 2] [1 3]] [[4 6] [5 7]]]