1. 程式人生 > >[Python]集合的交集,並集,差集

[Python]集合的交集,並集,差集

前提:測試中需要給某些應用設定黑名單,所以從.txt檔案中求兩者的差集,就可以篩選出需要測試的應用

思路:將.txt檔案中的資料讀到列表中,求列表的集合,再輸出到指定目錄

集合的思路:

a = [1,2,3,4,5,6]

b = [5,6,7]

c = []

①交集

c = [i for i in a if i in b]

c = [5,6]

'''

for i in a:                      # i 在a中迴圈

    if i in b:                    # 如果 i 在b中也存在

        c.append(i)         # 元素 i 新增至c中

'''

②差集

同理:

c1 = [i for i in a if i not in b]   #在a中但不在b中(某些情況,注意a,b順序)

c1 = [1,2,3,4]

c2 = [i for i in b if i not in a]

c2 = [7]

c = c1+c2

c = [1,2,3,4,7]

③並集

c = list(set(a).union(set(b)))

c = [1,2,3,4,5,6,7]