itertools.chain如何与生成器函数一起使用？ - python

在下面的内容中，我正在尝试但未获得固定清单:

>>> from itertools import chain
>>> 
>>> def foo():
...     for i in range(3):
...         yield range(5)
... 
>>> 
>>> chain(foo)
<itertools.chain object at 0x7fce499934d0>
>>> 
>>> list(chain(foo()))
[[0, 1, 2, 3, 4], [0, 1, 2, 3, 4], [0, 1, 2, 3, 4]]
>>> 
>>> list(chain([foo()]))
[<generator object foo at 0x7fce49994aa0>]
>>> 
>>> list(chain(list(foo())))
[[0, 1, 2, 3, 4], [0, 1, 2, 3, 4], [0, 1, 2, 3, 4]]

我究竟做错了什么？如何使用itertools.chain获取平面列表[0, 1, 2, 3, 4, 0, 1, 2, 3, 4, 0, 1, 2, 3, 4]？

python大神给出的解决方案

您已经很接近了，可以使用chain.from_iterable中的itertools

>>> list(itertools.chain.from_iterable(foo()))
[0, 1, 2, 3, 4, 0, 1, 2, 3, 4, 0, 1, 2, 3, 4]