是否有更短/更简单的更改多维数组的版本？

Question

我有一个由 RGBA 图像制成的多维数组

data

，在这段代码中，我使用原始颜色（从黑色到白色）作为新颜色的亮度来更改图像的颜色：

r

，

g 

、

b

、

a

data[:,:,0] = data[:,:,0]/255*r
data[:,:,1] = data[:,:,1]/255*g
data[:,:,2] = data[:,:,2]/255*b
data[:,:,3] = data[:,:,3]/255*a

代码工作正常。 但我认为改变

data

数组可以写得更短更好

我尝试使用mask做类似

this

的事情，但不幸的是我仍然不明白如何使这个简短的代码适应我的需求

Answer 1

使用广播：

data = data / 255 * [r, g, b, a]

您可以将结果与您拥有的结果进行比较。