将3x32x32大小的numpy图像数组转换为32x32x3并在Python中保存为实际图像

Question

我有一个形状为（3072,1）或（3,32,32）的numpy数组中的图像。这由前1024个值组成，分别对应于红色分量后接1024绿色分量后接1024蓝色分量。这1024个值对应于32x32大小的图像。我想知道是否有任何python方法或快捷方式将这个数组转换为32x32x3大小的数组，以便将数组保存为实际图像？在

这里的问题是，如果我把它的尺寸（32x32x3）改为

x = numpy.reshape(x,(32,32,3))

然后像素值会混乱起来，我不会得到实际的图像保存后，因为原来的图像是形状（3，32，32）

我只想把数组保存到实际图像中。在

Answer 1

您可以使用^{}：

In [5]: np.rollaxis(a,0,3).shape
Out[5]: (32, 32, 3)