Python的习惯用法是把有限的iterable链起来（扁平化）成无限的iterable？

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(iterables)

您还可以通过嵌套生成器理解来执行此操作：

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