Python 替换二维numpy数组中的向量_Python_Numpy

Python 替换二维numpy数组中的向量

python numpy

Python 替换二维numpy数组中的向量,python,numpy,Python,Numpy,我有一个numpy数组，形状（高度、宽度，3），是从图像加载的。我想用特定的颜色替换所有的黑色像素[0,0,0]。有没有一种numpy方法可以做到这一点 import numpy as np orig_color = (0, 0, 0) replacement_color = (r, g, b) data[(data == orig_color).all(axis = -1)] = replacement_color

我有一个

numpy数组

，形状

（高度、宽度，3）

，是从图像加载的。我想用特定的颜色替换所有的黑色像素

[0,0,0]

。有没有一种

numpy

方法可以做到这一点

import numpy as np
orig_color = (0, 0, 0)
replacement_color = (r, g, b)

data[(data == orig_color).all(axis = -1)] = replacement_color