Python 中的折叠生成器

Question

所以我有这样的功能：

def demo_range(limit):
    value = 0
    while value < limit:
        yield value
        value = value + 1

还有这个

def demo_generator_to_list(generator):
    return [x for x in range(generator)]

现在在demo_generator_to_list(generator) 中，我需要填写一些代码来将生成器折叠到：

[0,1,2,3]

来自

demo_generator_to_list(demo_range(4))

Answer 1

只需将生成器传递给list() 调用：

def demo_generator_to_list(generator):
    return list(generator)

list() 函数将遍历生成器并将所有结果添加到新的列表对象中。

你仍然可以使用列表推导式，但你不能使用range()：

return [x for x in generator]

这与使用list() 相比没有优势；它只是更慢。