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

Question

我有两个代表两个图像的 3D numpy 数组。每个数组的形状是 (1080, 1920, 3)。数字 3 代表图像中每个像素的 RGB 值。

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

我怎样才能只使用 numpy 方法来做到这一点？

标签： pythonnumpy

Answer 1

使用具有真/假值的掩码

# 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])