python - Python成语链接（展平）有限迭代的无限迭代？

Question

假设我们有一个返回列表（或有限迭代器）的迭代器（无限迭代器），例如由

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

什么是获得迭代器（显然是无限的）的好 Python 习惯用法，它将从第一个迭代器返回每个元素，然后从第二个迭代器返回每个元素，依此类推。在上面的示例中，它将返回1,2,3,1,2,3,.... 迭代器是无限的，所以itertools.chain(*infinite)不会工作。

有关的

• 在python中展平一个浅表

Answer 1

从 Python 2.6 开始，您可以使用itertools.chain.from_iterable：

itertools.chain.from_iterable(iterables)

您也可以使用嵌套生成器理解来执行此操作：

def flatten(iterables):
    return (elem for iterable in iterables for elem in iterable)

Answer 2

使用生成器：

(item for it in infinite for item in it)

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