列表理解双循环 if

Question

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

我想要一个布尔列表，它指定A的元素是否等于B的元素。 超集列表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]

---

另外，根据您的问题-

我想要一个布尔列表，它指定 A 的元素是否等于 B 的元素。

如果您真正想要的是一个布尔值（而不是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]

Answer 2

我想你要找的是两组的交集，对吧？

这是两个集合的交集（我强烈建议你下次想出更好的名字）：

A = [1, 2, 3, 5, 7, 10, 12, 9]

B = [2, 6, 7]

print [x if x in B else None for x in A]

你需要那里的“其他”，否则它不起作用。

并从字面上解决您的问题：

 res = [True if x in A else False for x in B]

我建议您查看这个答案： 在 Python 的列表理解中的 if/else？