마스크로 선택한 픽셀 목록을 작성하십시오.

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]