將函數應用於蒙版numpy數組

Question

我有一個圖像作爲numpy數組和圖像蒙版。

from scipy.misc import face 

img = face(gray=True) 
mask = img > 250 

如何將函數應用於所有被屏蔽的元素？

def foo(x): 
    return int(x*0.5) 

Answer 1

對於該特定功能，可以列出的方法很少。

方法1：可以使用boolean indexing就地設置 -

img[mask] = (img[mask]*0.5).astype(int) 

方法2：您還可以使用np.where一個可能更直觀的解決方案 -

img_out = np.where(mask,(img*0.5).astype(int),img) 

使用具有np.where(mask,A,B)語法的np.where，我們在兩個相等形狀的數組之間進行選擇生成與A和B相同形狀的新陣列。根據mask中的元素進行選擇，該元素的形狀與A和B的形狀相同。因此，對於mask中的每個True元素，我們選擇A，否則B。將此翻譯成我們的情況，A將是(img*0.5).astype(int)和B是img。

方法3：有一個內置的np.putmask這似乎是最接近這個確切的任務，可以用來做就地設置，像這樣 -

np.putmask(img, mask, (img*0.5).astype('uint8'))