Python成语链接（展平）有限迭代的无限迭代？ [复制]

Question

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

返回的迭代器

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

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

相关

• Flattening a shallow list in 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)

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