python - 替换二维 numpy 数组中的向量

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