替换具有某些条件语句的两个数组项

Question

有两个 28*28 像素的图像数组给出这样的掩码值。 Image1 显示肺的左面罩，Image2 显示肺的右面罩。重塑这些图像后，结果就是这种类型

img1=[............,False,  True, False, False, False,
   False, False, False, False, False, False, False, False, False,
   False, False, False, False, False, False, False, False, False,
   False, False, False, False,  True,  True,..................]
img2=[.........................,True,  True,  True, False, False, False, False,
   False, False, False, False, False, False, False, False, False,
   False, False, False, False, False, False, False, False, False,
   False, False,  True,  True,  True,  True,  True, False, False,
   False, False, False, False,............................]

基本上，我想合并这两个掩码以制作 1 个掩码，该掩码将是一个 new_image 并显示左右掩码的组合结果。我的想法是替换 img2 的索引值，它在 Img1 中显示为 True。这是我的代码

import numpy as np
from PIL import Image
img1=Image.open(path)
img2=Image.open(path)
img1_array=np.asarray(img1)
img2_array=np.asarray(img2)
img1_array_reshape=img1_array.reshape(784)
img2_array_reshape=img2_array.reshape(784)
print(len(img1_array_reshape))
print(len(img2_array_reshape))

indices=[i for i,val in enumerate(img2_array_reshape) if val==True]
print(indices)
for index in indices:
    print(index)
    img1_array_reshape[indices[index]] = img2_array_reshape[index]

但我无法运行最后一行，有没有直接的方法可以做到这一点？

Answer 1

如果您尝试将这两个数组与True 组合，那么您可以使用zip。

a = [True,False,False]
b = [True,True,True]

combined = [i[0] or i[1] for i in zip(a,b)]

print(combined) //[True, True, True]

Answer 2

我建议使用布尔掩码来索引img2_array 中的单元格，这些单元格是img1_array 中的True：

img1 = Image.open(path)
img2 = Image.open(path)
img1_array = np.asarray(img1)
img2_array = np.asarray(img2)
img2_array[img1_array] = True

由于img1_array 似乎已经是dtype=bool 的数组，所以可以直接将其用作索引数组。
这避免了缓慢而复杂的循环以及通过重塑和列表理解创建复杂的indices。 如果你无论如何都想重塑你的图像，你仍然可以做img1_array_reshape = img1_array.reshape(-1)。这在应用布尔掩码操作之前和之后都有效。