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

Posted 2023-02-23

【中文标题】Python成语链接（展平）有限迭代的无限迭代？ [复制]

【英文标题】：Python idiom to chain (flatten) an infinite iterable of finite iterables? [duplicate]

【发布时间】：2010-09-12 08:42:06

【问题描述】：

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

返回的迭代器

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

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

相关

Flattening a shallow list in python

【参考方案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)

【讨论】：

我可以用';'加入列表吗?

@diffracteD - 加入无限数量的项目？不，不是

【参考方案2】：

使用生成器：

(item for it in infinite for item in it)

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