根据条件设置 numpy 数组值

Question

我有 3 通道图像存储在 numpy 数组中。如果这些像素的 red_channel 大于特定值，我想将这些像素设置为白色

最有效的方法是什么？

注：图像形状为100×100×3

谢谢

Answer 1

如果最后一个维度是 r,g,b（例如使用 时的情况imageio.imread），那么您可以执行以下操作：x = image[:, :, 0] > cutoffimage[x, :] = 255例子：import imageiocutoff = 200image = imageio.imread("in.jpg")x = image[:, :, 0] > cutoffimage[x, :] = 255imageio.imwrite("out.jpg", image)