替换2d numpy数组中的向量

Question

我有一个形状为(height, width, 3)的numpy array从图像中加载。我想用特定颜色[r, g, b]替换所有黑色像素[0, 0, 0]。有没有numpy的方法来做到这一点？

Answer 1

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

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