python - 2dnumpy配列のベクトルを置き換えます

Question

numpy array画像から形状(height, width, 3)を読み込んでいます。[0, 0, 0]すべての黒のピクセルを特定の色に置き換えたい[r, g, b]。これを行う方法はあり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