扁平的numpy数组

Question

我正在对图像进行数据挖掘。每个像素都被视为数据点。图像如下：

im=Image.open('lena.bmp')
im=numpy.array(im)
print im.shape

根据图像是彩色还是灰度，im.shape可以是（10,10,3）或（10,10,1）

之后，图像被平铺为特征矩阵，如下所示：

if（10,10,3），然后---＆gt;（100,3）

if（10,10,1），然后---＆gt;（100,1）

如何为此编写多态函数？我目前的做法是：

obs=reshape(im,(im.shape[0]*im.shape[1],1, im.size/(im.shape[0]*im.shape[1])))

Answer 1

你可以这样做：

obs = np.reshape(im, (-1, im.shape[-1]))