Python 提取两个列表的共同元素

Python如何提取两个列表的共同元素？也就是取两个集合的交集。除了常规的循环外，Python提供了一些更简洁更美的表达法，比如可以用列表解析的方法，或者用 Python 内置的集合运算。

比如已有如下两个列表，Python如何提取两个列表的共同元素？

[1, 2, 3, 4, 5]

和

[4, 5, 6, 7, 8]

希望能得到

[4, 5]

的结果？

方法一，列表解析

可以用列表解析， List comprehension 下的方法：

A = [1, 2, 3, 4, 5]
B = [4, 5, 6, 7, 8]
print([x for x in A if x in B]) # [4, 5]

方法二，集合运算

还可以用Python 的集合（set）运算，s.intersection(t) 返回s与t的交集

A = [1, 2, 3, 4, 5]
B = [4, 5, 6, 7, 8]
print(set(A).intersection(set(B))) # {4, 5}

各位读后有什么想法，请在下方留言吧！如果对本文有疑问或者寻求合作，欢迎 联系邮箱 。邮箱已到剪贴板

精彩评论