python - chain函数

python 官方提供了itertools包，方便操作迭代器，这个包提供chain函数，兼顾内存效率和写法优雅

# 基本使用
from itertools import chain
a = [1,2,3,4,5]
b = [6,7,8,9,10]
c = {"a":100,"b":200}

for i in chain(a,b):
    print(i)     #1,2,3,4,5,6,7,8,9,10
    
for i in chain(a,c.values()):
    print(i)     #1,2,3,4,5,100,200

1
2
3
4
5
6
7
8
9
10
1
2
3
4
5
100
200

# 去除内嵌列表
from itertools import chain
b = [[1,2],[3,4],[5,6]]

# tmp = []
# for i in b:
#     for j in i:
#         tmp.append(j)

for i in chain(*b):  
    print(i)
    

1
2
3
4
5
6