python itertools

目录

• itertools.chain(list_str,list_int)

itertools.chain(list_str,list_int)

你想在多个对象执行相同的操作，但是这些对象在不同的容器中，你希望代码在不失可读性的情况下避免写重复的循环.

from itertools import chain
a = [1, 2, 3, 4]
b = [‘x’, ‘y’, ‘z’]
for x in chain(a, b):
  print(x)

1
2
3
4
x
y
z

好处：

如果采用a+b的方式遍历，那么要求a和b的类型一致，如果数据再大一点会，会消耗内存。
而chain,是通过创建迭代器，依次返回可迭代对象的元素。