Пример функции функции Python numpy ollaxis
# welcome to softhunt.net
# Python program explaining
# numpy.rollaxis() function
# importing numpy as np
import numpy as np
arr = np.ones((4, 3, 2, 1))
softhunt = np.rollaxis(arr, 3, 1).shape
print (softhunt)
Outrageous Ostrich