手记

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

定义

差集: 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下测试通过

原文来源:https://m.pythontab.com/article/1206


0人推荐
随时随地看视频
慕课网APP