使用 Python 获取两个列表的交集、并集、差集的常用方法 | Jin''''s Blog

文章目录

1. 1. 交集（Intersection）
2. 2. 并集（Union）
3. 3. 差集（Difference Set）

在数据处理中经常需要使用 Python 来获取两个列表的交集，并集和差集。在 Python 中实现的方法有很多，我平时只使用一两种我所熟悉的，但效率不一定最高，也不一定最优美，所以这次想把常用的方法都搜集总结一下。

intersection-union-difference

交集（Intersection）

>>> a = [1, 2, 3, 4, 5, 6]
>>> b = [2, 4, 6, 8 ,10]
>>> a and b
[2, 4, 6]

方法一：

intersection = list(set(a).intersection(b))

方法二：

intersection = list(set(a) & set(b))

方法三：

intersection = [x for x in b if x in set(a)]   # list a is the larger list b

方法四：

intersection = list((set(a).union(set(b)))^(set(a)^set(b)))

注意：如果不考虑顺序并且一定要使用 loop 的话，不要直接使用 List，而应该使用 Set。在 List 中查找元素相对 Set 慢了非常非常多。

参考资料：

• Python - Intersection of two lists
• How to find list intersection?
• List intersection in python: let’s do it quickly
• python 求两个list的差集，并集和交集

并集（Union）

>>> a = [1, 2, 3, 4, 5, 6]
>>> b = [2, 4, 6, 8 ,10]
>>> a or b
[1, 2, 3, 4, 5, 6, 8, 10]

方法一：

union = list(set(a) | set(b))

方法二：

list(set(a).union(set(b)))

注意：如果不考虑顺序并且一定要使用 loop 的话，不要直接使用 List，而应该使用 Set。在 List 中查找元素相对 Set 慢了非常非常多。

参考资料：

• Python set Union and set Intersection operate differently?
• How to find the intersection and union of two lists in Python
• python 求两个list的差集，并集和交集

差集（Difference Set）

>>> a = [1, 2, 3, 4, 5, 6]
>>> b = [2, 4, 6, 8 ,10]
>>> a or b
[1, 2, 3, 4, 5, 6, 8, 10]

方法一：

difference = list(set(b).difference(set(a)))    # elements in b but not in a
difference = list(set(a).difference(set(b)))    # elements in a but not in b

方法二：

difference = list(set(a_list)^set(b_list))

方法三：

difference = list(set(a)-set(b))    # elements in b but not in a
difference = list(set(a)-set(b))    # elements in a but not in b

方法四：

difference = [x for x in b if x not in set(a)]    # elements in b but not in a
difference = [x for x in b if x not in set(a)]    # elements in a but not in b

注意：如果不考虑顺序并且一定要使用 loop 的话，不要直接使用 List，而应该使用 Set。在 List 中查找元素相对 Set 慢了非常非常多。

参考资料：

• Get difference between two lists
• Retaining order while using Python’s set difference
• How to find the intersection and union of two lists in Python
• python 求两个list的差集，并集和交集