问题:我们需要对许多对象执行相同的操作,但是这些对象包含在不同的容器内,而我们希望可以避免写出嵌套循环,保持代码的可读性
解决方案:使用itertools.chain()
可以解决这个问题,它接受一系列可迭代对象作为输入并返回一个迭代器(实际上是在对多个容器进行迭代)
import itertools
a = [1, 2, 3, 4]
b = ['x', 'y', 'z']
for x in itertools.chain(a, b):
print(x)
1
2
3
4
x
y
z
总结:
1、itertools.chain()
可接受一个或多个可迭代对象作为参数,然后会创建一个迭代器,该迭代器可连续访问并返回提供的每个可迭代对象中的元素;
2、如果输入的序列很大,在内存使用上chain()就会高效很多,而且当可迭代对象之间不是同一种类型时也可以轻松适用。
网友评论