修改 numpy 数组的形状

Modify the shape of a numpy array

我有一个形状图像数据集的 numpy 数组: (32, 32, 3, 7000) (高度、宽度、通道、图像数量)

但是,我想要具有相同形状的数据集: (7000, 32, 32, 3) (图像数量、高度、宽度、通道) 不改变图像

我能做什么? 谢谢!


我用过: np.moveaxis(X_train, -1, 0)

请使用np.transpose函数(https://numpy.org/doc/stable/reference/generated/numpy.transpose.html)

np.transpose(x, (3, 0, 1,2))