文章目录
  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 慢了非常非常多。

参考资料:

并集(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 慢了非常非常多。

参考资料:

差集(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 慢了非常非常多。

参考资料:

文章目录
  1. 1. 交集(Intersection)
  2. 2. 并集(Union)
  3. 3. 差集(Difference Set)