4-D Numpy数组的互换顺序

Question

我有一个尺寸为96x96x3x1000的4-D Numpy数组-这些对应于我已导入的图像数据集：96X96像素各1000幅图像，每个像素RGB值。

但是，我需要遍历每个图像的展平数组，即。 仅是二维数组[1000] [96 * 96 * 3]。 我设法通过先做来变换给定的数组

    a.reshape(-1,a.size[3])

然后使用循环将每一列分配给图像。 我想问一下是否有一种更简单的切片方法来交换ndarray的顺序？

谢谢

Answer 1

您可以使用numpy.swapaxes更改轴的numpy.swapaxes

a.reshape(-1,1000).swapaxes(0,1)

或简单地转译

a.reshape(-1,1000).T

您也可以更改轴的顺序在与开始numpy.transpose ，然后应用reshape

a.transpose([3,0,1,2]).reshape(1000,-1)