替換2D Numpy數組中的向量

Question

我從圖像加載了一個形狀(height, width, 3)的numpy array 。 我想將所有黑色像素[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