マスクで選択されたピクセルのリストを作成する

Question

私はイメージとそのイメージからピクセルを選択するマスクを持っています。これらのピクセルのリストを作成し、scipy.kmeansを使用してそれらのクラスター分析を行いたいと思います。 は、より良い方法はあります（たとえば、私が知らない機能）を使用して、この

pixels = [] 
for irow,mrow in zip(img,mask): 
    pixels += [pixel for pixel,mbyte in zip(irow,mrow) if mbyte==255] 

Answer 1

よりnumpyからboolean or "mask" index array：

>>> import numpy as np 
>>> img = np.array([[1, 2, 3], 
...     [4, 5, 6]]) 
>>> mask = np.array([[0, 255, 0], 
...     [255, 0, 0]]) 
>>> img[mask == 255] 
array([2, 4]) 

あなたの代わりにnumpy.arrayのlistをしたい場合は、にlistを使用CONVER結果：

>>> list(img[mask == 255]) 
[2, 4]