Python idiom to chain (flatten) an infinite iterable of finite iterables?
本问题已经有最佳答案,请猛点这里访问。
假设我们有一个迭代器(无限迭代器)返回列表(或有限迭代器),例如由
1 | infinite = itertools.cycle([[1,2,3]]) |
获取迭代器(显然是无限的)的好的python习惯用法是什么?它将从第一个迭代器返回每个元素,然后从第二个迭代器返回每个元素,等等。在上面的示例中,它将返回
- 扁平化python中的浅列表
从python 2.6开始,可以使用
1 | itertools.chain.from_iterable(iterables) |
您还可以使用嵌套生成器理解来执行此操作:
1 2 | def flatten(iterables): return (elem for iterable in iterables for elem in iterable) |
使用发电机:
1 | (item for it in infinite for item in it) |
*构造解包成一个元组以便传递参数,因此无法使用它。