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]

方法二,集合运算

还可以用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}

相关文章   欢迎到 留言板 写下你的看法。
  本页面内容采用 署名协议 CC-BY 授权。欢迎转载,请保留原文链接