python zip()的使用

zip()函數將可迭代的對象作為參數，將對象對應的元素打包成一個個元祖，然后返回由這些元祖組成的對象，我們可以使用list()轉換成列表

如果各個迭代器的元素個數不一致，則返回列表長度和最小的對象相同，用*好操作符，可以將元祖解壓為列表

1.兩個或者多個list

a=[1,2,3]
b=[4,5,6]
c=[7,8,9,10]    

#兩個list去zip()
zip(a,b)

#三個list去zip(),返回的對象的長度和最小的迭代器一致
zip(a,b,c)

#用list()查看zip后的數據
list(zip(a,b))  #[(1, 4), (2, 5), (3, 6)]
list(zip(a,b,c))  #[(1, 4, 7), (2, 5, 8), (3, 6, 9)]

#使用zip(*zip())解壓
a1,b1=zip(*zip(a,b))

#兩個list迭代輸出
for i,j in zip(a,b):
    print(i,j)
    
輸出：
1 4
2 5
3 6

 

2.如有兩個list，一個是一層，一個是嵌套，要組成一個dict 

#將zip對象轉換成字典看看
a=[1,2,3]
d=[['a','b','c'],['aa','bb','cc'],['aaa','bbb','ccc']]

dict(zip(a,d))  #{1: ['a', 'b', 'c'], 2: ['aa', 'bb', 'cc'], 3: ['aaa', 'bbb', 'ccc']}
#如果一個是key值，一個內層list就是一行value
[dict(zip(a,value)) for value in d]  #[{1: 'a', 2: 'b', 3: 'c'},
 {1: 'aa', 2: 'bb', 3: 'cc'},
 {1: 'aaa', 2: 'bbb', 3: 'ccc'}]

#或者是    
l=[]
for value in d:
    dict={}
    for i in range(len(value)):
        dict[a[i]]=value[i]
    l.append(dict)
    

3.同時遍歷兩個dict

如果長度相同的dict直接用zip函數即可同時for
如果長度不同，可以先把這兩個dict整合進一個list再for

dict_1=dict(zip(a,b))  #{1: 4, 2: 5, 3: 6}
dict_2=dict(zip(a,d))  # {1: ['a', 'b', 'c'], 2: ['aa', 'bb', 'cc'], 3: ['aaa', 'bbb', 'ccc']}


#key的遍歷
for i,j in zip(dict_1,dict_2):
    print(i,j)

輸出：
1 1
2 2
3 3

#values的遍歷
for i ,j in zip(dict_1.values(),dict_2.values()):
    print(i,j)
輸出：
4 ['a', 'b', 'c']
5 ['aa', 'bb', 'cc']
6 ['aaa', 'bbb', 'ccc'] 

#同時遍歷,需要2個dict的value都是多層的
dict_4=dict(zip(b,d))   #{4: ['a', 'b', 'c'], 5: ['aa', 'bb', 'cc'], 6: ['aaa', 'bbb', 'ccc']}
for i,j in zip(dict_4,dict_2):
    for k in dict_4[i]:
        print(i,k)
    for m in dict_2[j]:
        print(j,m)

輸出：
4 a
4 b
4 c
1 a
1 b
1 c
5 aa
5 bb
5 cc
2 aa
2 bb
2 cc
6 aaa
6 bbb
6 ccc
3 aaa
3 bbb
3 ccc