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

2014-04-02
标签: PYTHON
本文发布至今已有10年零17天,可能不再适用,请谨慎对待。

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}

如果您对本站内容有疑问或者寻求合作,欢迎 联系邮箱邮箱已到剪贴板

标签: PYTHON

欢迎转载本文,惟请保留 原文出处 ,且不得用于商业用途。
本站 是个人网站,若无特别说明,所刊文章均为原创,并采用 署名协议 CC-BY-NC 授权。

相关文章