Измените матрицу в numpy

Question

У меня есть массив numpy размером 5000x32x32x3. Число 5000 - это количество изображений, и каждое изображение равно 32x32 по ширине и высоте и имеет цветные каналы 3.

Теперь я хотел бы создать массив numpy формы 5000x3x32x32 таким образом, чтобы данные сохранялись. Что я имею в виду сохранения данных являются:

1. Там должен быть 5000 точек данных в результирующем массиве
2. 2-й размер (3) массива правильно определяет цветовой канал, т.е. всех элементов, второе измерение 0 принадлежат к красному каналу, второе измерение которого равно 1, принадлежат к зеленому каналу, второе измерение которого 2 принадлежит к синему каналу.

Простое изменение формы на np.reshape(data,(5000,3,32,32)) не будет работать, поскольку оно не сохранит каналы, а просто изменит данные в желаемую форму.

Answer 1

Я думаю, что вы ищете перестановки осей, numpy.transpose может получить эту работу:

data = np.transpose(data, (0, 3, 1, 2))