マスクされた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では、2つの等しい形状の配列の中から選択していますAおよびBを使用して、AおよびBと同じ形状の新しい配列を生成します。この選択は、maskの要素に基づいて行われ、AおよびBと同じ形状です。したがって、の要素がmaskにある場合は、Aを選択します。それ以外の場合はBとなります。これを当社のケースに翻訳すると、Aは(img*0.5).astype(int)となり、Bはimgとなります。

アプローチ＃3：内蔵のこの正確なタスクに最も近いように思わやインプレースの設定を行うために使用することができ、そのようなnp.putmaskすることがあります -

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