在numpy中重塑阵列

Question

我有一个大小为5000x32x32x3的numpy数组。 数字5000是图像的数量，每个图像的宽度和高度为32x32 ，并且具有3颜色通道。

现在我想以保存数据的方式创建一个形状为5000x3x32x32的numpy数组。 我保留数据的意思是：

1. 结果数组中应该有5000个数据点
2. 阵列的第二维（ 3 ）正确确定颜色通道，即第二维为0所有元素属于红色通道，第二维为1属于绿色通道，第二维为2属于蓝色通道。

简单地重塑np.reshape(data,(5000,3,32,32))将不起作用，因为它不会保留通道，只是将数据重新整形为所需的形状。

Answer 1

我认为你正在寻找轴的排列， numpy.transpose可以完成这项任务：

data = np.transpose( data, (0, 3, 1, 2))