替換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