在Python中组合3个布尔掩码

Question

我有3个列表:

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

当我输入

a or b or c

我想找回一份清单

[True, True, True]

但我回来了

[True, False, True]

有什么想法吗？我该如何组合这些面具？

Answer 1

您的or操作员将列表比较为整个对象,而不是其元素.由于a不是一个空列表,它的计算结果为true,并成为结果or.b和c甚至不进行评估.

要按位置生成三个列表的逻辑OR,您必须遍历其内容并对每个位置的值进行OR运算.要将一堆可迭代转换为其分组元素的列表,请使用zip().要检查iterable中的任何元素是否为真(其整个内容的OR),请使用any().用列表理解一次完成这两个:

mask = [any(tup) for tup in zip(a, b, c)]

Answer 2

这个怎么样:

from numpy import asarray as ar

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

尝试:

>>> ar(a) | ar(b) | ar(c)                  #note also the use `|` instead of `or`
array([ True, True, True], dtype=bool)

所以不需要zip等等

Answer 3

or如果它的计算结果为true,则返回第一个操作数,非空列表的计算结果为true; 因此,如果它是非空列表,a or b or c将始终返回a.

可能你想要的

[any(t) for t in zip(a, b, c)]

(这个作品也是逐元素and如果更换any用all)