numpy數組的變平

Question

我正在對圖像執行數據挖掘。每個像素都被視爲一個數據點。圖像被如下：

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

根據圖像是否是彩色或灰度，im.shape是（10,10，3）或（10,10,1）

之後中，圖像是平坦的一個特徵矩陣如下：

如果（10,10,3），然後--->（100,3）

如果（10,10,1），然後 - - >（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]))