numpy數組變化指數

Question

我有一個numpy數組與3234 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