Numpy数组更改索引

Question

我有一个带有X_train.shape: (32, 32, 3, 73257)的32 x 32 x 3张照片的numpy-array。但是，我想有以下数组形状：（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