python中列表之间求差集、交集、并集

求两个列表的交集、并集、差集

python

def diff(listA, listB):

# 求交集的两种方式

retA = [i for i in listA if i in listB]

retB = list(set(listA).intersection(set(listB)))

print("retA is :", retA)

print("retB is :", retB)

# 求并集

retC = list(set(listA).union(set(listB)))

print("retC is:", retC)

# 求差集,在B中但不在A中

retD = list(set(listB).difference(set(listA)))

print("retD is:", retD)

retE = [i for i in listB if i not in listA]

print("retE is:", retE)

def main():

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

listB = [3, 4, 5, 6, 7]

diff(listA, listB)

if name == 'main':

main()

main()

###直接上运行结果

# retA is : [3, 4, 5]

#retB is : [3, 4, 5]

#retC is: [1, 2, 3, 4, 5, 6, 7]

#retD is: [6,7]

#retE is: [6, 7]

*代码思路：使用列表解析式。列表解析式一般来说比循环快，而且更python，将list转成set以后，使用set的各种方法去处理*