Python成语链接(展平)有限迭代的无限迭代？

Question

这个问题在这里已经有了答案:

How do I make a flat list out of a list of lists? (33 个答案)

关闭 6年前。

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

返回的迭代器

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)