Самый простой способ перевернуть размер фильма с помощью Python

Question

У меня есть фильм, загруженный из tif-файла с использованием skimage.external.tifffile.imread() в массив с формой (frames, width, height). Что будет лучшим способом изменить порядок фильма в форме ( width, height, frames)?

Я могу построить функцию, которая делает это, используя цикл for, но есть ли лучший способ изменить форму, избегая реализации цикла for? Какая-то векторизация проблемы?

Answer 1

Вы можете сделать транспонирование с последующим свапами :

import numpy as np

movies = np.zeros((10, 250, 100))

print(movies.shape)
print(np.swapaxes(movies.T, 0, 1).shape)

выход

(10, 250, 100)
(250, 100, 10)

Answer 2

Вы можете использовать numpy.moveaxis :

movie = np.moveaxis(movie, 0, 2)