在数据处理中经常需要使用 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 ,1