当mask为1时，numpy更改数组值

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]