从2d创建3通道蒙版的numpy方法

Question

我有一个numpy数组scale_down代表RGB图像（形状（500,500,3））和蒙版2d numpy数组输出（形状（500,500）），并且需要用灰色填充其中scaled_down大于给定值的所有像素。目前，我通过以下方式进行操作。

for i in range(out.shape[0]):
  for j in range(out.shape[1]):
    if out[i][j] > 10:
       scale_down[i][j][:] = 100

但是我希望numpy中还有另一种方法可以实现，而无需for循环。有什么帮助吗？

Answer 1

我在这里重复@Warren Weckesser提供的答案。

scale_down[out > 10] = 100