时间戳

首页 > 解决方案 > 从 2d 创建 3 通道蒙版的 numpy 方法

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

问题描述

我有一个代表 RGB 图像（形状（500,500,3））的 numpy 数组 scale_down 和掩码 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 循环。有什么帮助吗？

标签： pythonnumpy

解决方案

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

scale_down[out > 10] = 100

推荐阅读