Python求两个list的差集、交集与并集的方法

2023年12月23日 645次阅读

定义

差集： A，B是两个集合，所有属于A且不属于B的元素构成的集合， 就是差集。

交集： A，B是两个集合，既属于A又属于B的元素构成的集合， 就是交集。

并集： A，B是两个集合，把他们所有的元素合并在一起组成的集合，就是并集。

求两个list差集

如有下面两个列表：

listA = [1,2,3,4]
listB = [2,3,4]

想要的结果是[1]

有3种方法：

1. 循环遍历法

ret = []
for i in listA:
    if i not in listB:
        ret.append(i)
print(ret)

2. 运算符法

ret = list(set(listA) ^ set(listB))
print(ret)

3. difference函数法

list(set(listA).difference(set(listB)))
print(ret)

很明显第二种、第三种方法更加优雅。

求两个list的并集

代码如下:

ret = list(set(listA).union(set(listB)))
print(ret)

求两个list的交集

ret = list(set(listA).intersection(set(listB)))
print(ret)

总结：

这三个集合的求法都可以是，将list转成set以后，使用set的各种方法去处理。

注：以上代码在Python3下测试通过

点赞