如何简化RGBA图像多维数组data的颜色调整代码？

你可以利用NumPy的广播机制来简化代码，不需要逐个通道手动赋值。核心思路是把目标颜色值(r, g, b, a)构造成一个与数组最后一维匹配的数组，让NumPy自动完成逐通道的运算：

import numpy as np

# 构造颜色系数数组，适配data的最后一维
color_coeff = np.array([r, g, b, a]) / 255
# 一次性完成所有通道的调整（若需原地修改，改用data *= color_coeff）
data = data * color_coeff

为什么这样可行？ #

NumPy的广播机制会自动扩展color_coeff的形状（从(4,)扩展为(1,1,4)），和data的形状（假设为(H,W,4)）匹配，这样每个通道的元素都会对应乘以对应的系数，和你原来的逐通道计算逻辑完全一致，但代码更简洁。

你提到的mask更适合做条件过滤场景（比如只修改满足特定条件的像素），这个需求用广播就能完美解决，不需要用到mask。

内容的提问来源于stack exchange，提问作者Mr. Arceni