Python生成器从另一个生成器调用中产生一切

时间:2011-06-06 20:28:23

标签: python recursion generator yield

我有一个Python生成器,可以调用自己来获得更多的元素。它看起来像这样:

def gen(list):
    # ...
    if list:
        for x in gen(list[1:]):
            yield x

我的问题是关于最后两行:是否有更简洁的方式来表达这一点?我希望有这样的东西(理解这不是有效的Python):

def gen(list):
    # ...
    if list:
        yield each in gen(list[1:])

4 个答案:

答案 0 :(得分:21)

有人要求yield from之类的“通过”子发电机返回的所有值。有关已被反弹的一些想法,请参阅PEP 380。但是,还没有任何实施。你的第一个例子是正确的。

答案 1 :(得分:5)

您的代码示例非常惯用且简洁,没有必要也没有真正的机会进行进一步的改进,尤其是在可读性方面。

答案 2 :(得分:3)

你的第一个例子是正确的解决方案。

答案 3 :(得分:3)

Python 3.3 added the yield from keyword。以下是您目前拥有的内容与使用新关键字的代码之间的比较:

yield_from_test.py

def gen_for(a_list):
    if a_list:
        yield a_list[0]
        for x in gen_for(a_list[1:]):
            yield x

def gen_yield(a_list):
    if a_list:
        yield a_list[0]
        yield from gen_yield(a_list[1:])

if __name__ == '__main__':
    assert list(gen_for([1,2,3])) == list(gen_yield([1,2,3]))
    print(list(gen_yield([1,2,3])))

» python3 yield_from_test.py [1, 2, 3]