Numpy Array Индексы изменений

Question

У меня есть numpy-массив с 32 x 32 x 3 картинками с X_train.shape: (32, 32, 3, 73257). Однако я хотел бы иметь следующую массивную форму: (73257, 32, 32, 3).

Как это сделать?

Answer 1

Есть два способа архивировать это или np.reshape(x, ndims) или np.transpose(x, dims).

Для фотографий предлагаю np.transpose(x, dims), который может быть использован с использованием X_train = np.transpose(X_train, (3,0,1,2)).

Answer 2

Я думаю, что вы хотите сделать transpose

>>> a = np.array([[1, 2], [3, 4]]) 
>>> a 
array([[1, 2], 
     [3, 4]]) 
>>> a.transpose() 
array([[1, 3], 
     [2, 4]]) 
>>> a.transpose((1, 0)) 
array([[1, 3], 
     [2, 4]]) 
>>> a.transpose(1, 0) 
array([[1, 3], 
     [2, 4]]) 

Answer 3

np.reshape(X_train, (73257, 32, 32, 3)) 

https://docs.scipy.org/doc/numpy-1.10.1/reference/generated/numpy.reshape.html