关于迭代器:python习语to chain(flatten)a infinite iterables of infinite iterables?

Python idiom to chain (flatten) an infinite iterable of finite iterables?

本问题已经有最佳答案,请猛点这里访问。

假设我们有一个迭代器(无限迭代器)返回列表(或有限迭代器),例如由

1
infinite = itertools.cycle([[1,2,3]])

获取迭代器(显然是无限的)的好的python习惯用法是什么?它将从第一个迭代器返回每个元素,然后从第二个迭代器返回每个元素,等等。在上面的示例中,它将返回1,2,3,1,2,3,...。迭代器是无限的,因此itertools.chain(*infinite)将不起作用。

相关的

  • 扁平化python中的浅列表


从python 2.6开始,可以使用itertools.chain.from_iterable

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)

*构造解包成一个元组以便传递参数,因此无法使用它。