Python求两个list的交集、并集、差(补)集、对称差集的方法
摘要
本文主要介绍在Python下求两个list的交集、并集、差(补)集、对称差集的方法。首先,总结了实现上述功能主要的两种方法:1.使用set集合运算符,2.使用set集合的方法(推荐第2种方法);接着,依次对同一功能的不同种实现方式罗列出具体例子(不限于上述两种方法)。
1. 总结
1.1 求两个list的交、并、差(补)、对称差集 - 使用set集合运算符
输入:
a = [0,1,2,3,4]
b = [0,2,6]
list(set(a) & set(b)) # 使用 "&" 运算求a与b的交集,输出:[0, 2]
list(set(a) | set(b)) # 使用 "|" 运算求a与b的并集,输出:[0, 1, 2, 3, 4, 6]
list(set(b) - set(a)) # 使用 "-" 运算求a与b的差(补)集: 求b中有而a中没有的元素,输出:[6]
list(set(a) - set(b)) # 使用 "-" 运算求a与b的差(补)集: 求a中有而b中没有的元素,输出:[1, 3, 4]
list(set(a) ^ set(b)) # 使用 "^" 运算求a与b的对称差集,输出:[1, 3, 4, 6]
输出:
求交集: list(set(a) & set(b)) 输出 -> [0, 2]
求并集: list(set(a) | set(b)) 输出 -> [0, 1, 2, 3, 4, 6]
求差(补)集: list(set(b) - set(a)) 输出 -> [6]
求差(补)集: list(set(a) - set(b)) 输出 -> [1, 3, 4]
求对称差集: list(set(a) ^ set(b)) 输出 -> [1, 3, 4, 6]
1.2 求两个list的交、并、差(补)、对称差集 - 使用set集合的方法 - 高效率
输入:
a = [0,1,2,3,4]
b = [0,2,6]
list(set(a).intersection(set(b))) # 使用 intersection 求a与b的交集,输出:[0, 2]
list(set(a).union(b)) # 使用 union 求a与b的并集,输出:[0, 1, 2, 3, 4, 6]
list(set(b).difference(set(a))) # 使用 difference 求a与b的差(补)集:求b中有而a中没有的元素,输出: [6]
list(set(a).difference(set(b))) # 使用 difference 求a与b的差(补)集:求a中有而b中没有的元素,输出:[1, 3, 4]
list(set(a).symmetric_difference(b)) # 使用 symmetric_difference 求a与b的对称差集,输出:[1, 3, 4, 6]
输出:
求交集:list(set(a).intersection(set(b)) 输出 --> [0, 2]
求并集:list(set(a).union(b)) 输出 --> [0, 1, 2, 3, 4, 6]
求差(补)集: list(set(b).difference(set(a)) 输出 --> [6]
求差(补)集: list(set(a).difference(set(b))) 输出 --> [1, 3, 4]
求对称差集: list(set(a).symmetric_difference(b)) 输出 --> [1, 3, 4, 6]
2.求两list的交集 - 不同实现方式
输入:
"""以a、b为实验对象"""
a = [0,1,2,3,4]
b = [0,2,6]
"""方法1:使用 intersection 求a与b的交集"""
c1 = list(set(a).intersection(set(b)))
print('c1 -->', c1) #输出:c1 --> [0, 2]
"""方法2:在list里使用循环 求a与b的交集 - 不推荐使用"""
c2 = [c_i for c_i in a if c_i in b]
print('c2 -->', c2) #输出:c2 --> [0, 2]
"""方法3:使用 "&" 运算 求a与b的交集"""
c3 = list(set(a) & set(b))
print('c3 -->', c3) #输出:c3 --> [0, 2]
"""方法4:使用 append 求a与b的交集 - 不推荐使用"""
c4 = []
for a_i in a:
if a_i in b:
c4.append(a_i)
print('c4 -->', c4) # 输出:c4 --> [0, 2]
输出:
c1 --> [0, 2]
c2 --> [0, 2]
c3 --> [0, 2]
c4 --> [0, 2]
3.求两个list的并集 - 不同实现方式
输入:
"""以a、b为实验对象"""
a = [0,1,2,3,4]
b = [0,2,6]
"""方法1:使用 union 求a与b的并集"""
c1 = list(set(a).union(set(b)))
print('c1 -->', c1) # 输出c1 --> [0, 1, 2, 3, 4, 6]
"""方法2:使用 "|" 求a与b的并集"""
c2 = list(set(a) | set(b))
print('c2 -->', c2) # 输出c2 --> [0, 1, 2, 3, 4, 6]
"""方法3:使用 append 求a与b的并集 - 不推荐使用"""
import copy
c3 = copy.deepcopy(a)
for b_i in set(b):
if b_i not in set(a):
c3.append(b_i)
print('c3 -->', c3) # 输出c3 --> [0, 1, 2, 3, 4, 6]
输出:
c1 --> [0, 1, 2, 3, 4, 6]
c2 --> [0, 1, 2, 3, 4, 6]
c3 --> [0, 1, 2, 3, 4, 6]
4.求两个list的差(补)集 - 不同实现方式
输入:
"""以a、b为实验对象"""
a = [0,1,2,3,4]
b = [0,2,6]
"""方法1:使用 difference 求a与b的差(补)集"""
c1 = list(set(b).difference(set(a))) # 求b中有而a中没有的元素
c2 = list(set(a).difference(set(b))) # 求a中有而b中没有的元素
print('c1 -->', c1) # 输出c1 --> [6]
print('c2 -->', c2) # 输出c2 --> [1, 3, 4]
"""方法2:使用 "-" 求a与b的差(补)集"""
c3 = list(set(b) - set(a)) # 求b中有而a中没有的
c4 = list(set(a) - set(b)) # 求a中有而b中没有的
print('c3 -->', c3) # 输出c3 --> [6]
print('c4 -->', c4) # 输出c4 --> [1, 3, 4]
输出:
c1 --> [6]
c2 --> [1, 3, 4]
c3 --> [6]
c4 --> [1, 3, 4]
5.求两个list的对称差集 - 不同实现方式
对称差集:只属于其中一个集合,而不属于另一个集合的元素组成的集合(求不同时属于两集合的元素组成的集合)
输入:
"""以a、b为实验对象"""
a = [0,1,2,3,4]
b = [0,2,6]
"""方法1:使用 symmetric_difference 求对称差集"""
c1 = list(set(a).symmetric_difference(set(b))) """c1 与 c2 的结果一样"""
print('c1 -->', c1) """输出c1 --> [1, 3, 4, 6]"""
c2 = list(set(b).symmetric_difference(set(a))) """c1 与 c2 的结果一样"""
print('c2 -->', c2) """输出c2 --> {1, 3, 4, 6}"""
"""方法2:使用 "^" 求对称差集"""
c3 = list(set(a) ^ set(b))
print('c3 -->', c3) """输出c3 --> {1, 3, 4, 6}"""
输出:
c1 --> [1, 3, 4, 6]
c2 --> [1, 3, 4, 6]
c3 --> [1, 3, 4, 6]