python 兩個list 求交集,並集,差集
阿新 • • 發佈:2018-01-20
pytho 列表解析 int __main__ class clas etc intersect run 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 "retC1 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()
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
讓code run起來
retA is: [3, 4, 5]
retB is: [3, 4, 5]
retC1 is: [1, 2, 3, 4, 5, 6, 7]
retD is: [6, 7]
retE is: [6, 7]
1
2
3
4
5
結合代碼來看,大體上是兩種思路:
1.使用列表解析式。列表解析式一般來說比循環更快,而且更pythonic顯得更牛逼。
2.將list轉成set以後,使用set的各種方法去處理。
python 兩個list 求交集,並集,差集