Numpy:有效地将满足条件的所有像素设置为黑色

import numpy as np

def set_pixels_to_black(image, condition):
    # 将满足条件的像素设置为黑色
    image[condition] = [0, 0, 0]

# 假设我们有一个RGB图像，存储在一个ndarray对象中
image = np.array([
    [[255, 255, 255], [100, 100, 100], [200, 200, 200]],
    [[150, 150, 150], [50, 50, 50], [250, 250, 250]],
    [[180, 180, 180], [120, 120, 120], [80, 80, 80]]
])

# 设置条件，选择像素值小于200的像素
condition = image < 200

# 调用函数，将满足条件的像素设置为黑色
set_pixels_to_black(image, condition)

print(image)

[[[  0   0   0]
  [100 100 100]
  [  0   0   0]]

 [[150 150 150]
  [ 50  50  50]
  [  0   0   0]]

 [[  0   0   0]
  [120 120 120]
  [ 80  80  80]]]