numpy在mask爲1時更改數組值

Question

我是numpy的新手，我遇到了麻煩。

我有兩個numpy的陣列，IMG和THR：

>>>img.shape 
(2448, 3264, 3) 
>>>thr.shape 
(2448, 3264) 

，我希望做這樣的事情：設置img[x,y] = [255,255,255]只有當thr[x,y] is not 0

我試圖遍歷數組過來做我自己，但需要很長時間，所以我真的需要C下的numpy。我也看了看屏蔽陣列，但我不明白如何使用它們。

謝謝！

Answer 1

使用NumPy assignment to an indexed array：

img[thr != 0] = [255,255,255]