python - 带有 if 的列表理解双循环

Question

我有两个列表 A 和 B，其中 B 是 A 的超集。

我想要指定 A 的元素是否等于 B 的元素的 bool 列表。超集列表 B 是有序的。

如果试过这个:

res = [(1 for a in A if a is b) for b in B]

根据其他答案，这应该可行。但在我的例子中，它返回了一个生成器列表。我怎样才能得到一个普通列表。像这样的东西:

A = ['a', 'b', 'c', 'e']
B = ['a', 'b', 'c', 'd', 'e', 'f']
res = [1, 1, 1, 0, 1, 0]

Answer 1

你得到生成器对象的原因是因为你使用了 () ，因此那些生成器表达式，在你遍历它们之前不会被评估。

如果简单的 in 运算符，您真正应该使用的是什么。示例 -

>>> A = ['a', 'b', 'c', 'e']
>>> B = ['a', 'b', 'c', 'd', 'e', 'f']
>>> res = [1 if x in A else 0 for x in B]
>>> res
[1, 1, 1, 0, 1, 0]

---

如果您的用例更复杂并且您必须使用 is 运算符来比较 A 和 B 的元素。然后，您可以使用 any() 函数遍历生成器表达式，如果找到匹配则返回 True ，否则返回 False 。示例 -

res = [1 if any(a is b for a in A) else 0 for b in B]

演示 -

>>> A = ['a', 'b', 'c', 'e']
>>> B = ['a', 'b', 'c', 'd', 'e', 'f']
>>> res = [1 if any(a is b for a in A) else 0 for b in B]
>>> res
[1, 1, 1, 0, 1, 0]

---

另外，根据你的问题——

I would like boolean list of that specifies whether an element of A is equal to an element of B.

如果您真正想要的是 bool 值(而不是1 或0)，那么您只需将上面的示例更改为-

示例 1 -

res = [x in A for x in B]

示例 2 -

res = [any(a is b for a in A) for b in B]