python-3.x 有没有更短/更简单的版本来更改多维数组？

hwamh0ep 于 2023-10-21 发布在 Python

关注(0)|答案(1)|浏览(110)

我有一个由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的事情，但不幸的是，我仍然不明白如何使这段简短的代码适应我的需要

1条答案

使用广播：

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

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