numpy.moveaxis()

numpy.moveaxis

numpy.moveaxis(a, source, destination) [source]

Move axes of an array to new positions.

Other axes remain in their original order.

Parameters:

a : np.ndarray

The array whose axes should be reordered.

source : int or sequence of int

Original positions of the axes to move. These must be unique.

destination : int or sequence of int

Destination positions for each of the origin