如何用另一个数组的相同索引中的值替换一个数组中的值？

Question

我有两个表示两个图像的3D numpy数组。每个数组的形状是（1080，1920，3）。数字3代表图像中每个像素的RGB值。

我的目标是将第一个数组中的每个非黑色像素替换为另一个数组中“平行”像素（在同一索引中）的值。

如何仅使用numpy方法执行此操作？

Answer 1

使用具有True / False值的蒙版

# All pixels should be normalized 0..1 or 0..254
first_img = np.random.rand(1920,1080,3)
second_img = np.random.rand(1920,1080,3)

eps = 0.01  # Black pixel threshold
mask = first_img.sum(axis=2) > eps

for i in range(first_img.shape[2]):
    first_img[:,:,i] = (first_img[:, :, i] * mask) + ((1 - mask) * second_img[:, :, i])