带python的numpy：将3d数组转换为2d

Question

假设我有一个彩色图像，很自然，这将由python中的一个三维数组表示，比如shape（nxmx3），并称之为img。

我想要一个新的二维数组，称之为“narray”有一个形状（3，nxm），这样数组的每一行分别包含R、G和B通道的“扁平”版本。而且，它应该具有这样的特性：我可以很容易地通过类似于

narray[0,].reshape(img.shape[0:2])    #so this should reconstruct back the R channel.

问题是如何从img构造“narray”？简单的img.reforme（3，-1）不起作用，因为元素的顺序对我来说是不可取的。

谢谢

Answer 1

假设我们有一个大小为img的数组m x n x 3来转换成大小为new_img的数组3 x (m*n)

new_img = img.reshape((img.shape[0]*img.shape[1]), img.shape[2])
new_img = new_img.transpose()