Joh*_*nck 17 python recursion yield generator
我有一个Python生成器,可以调用自己来获得更多元素.它看起来像这样:
def gen(list):
# ...
if list:
for x in gen(list[1:]):
yield x
Run Code Online (Sandbox Code Playgroud)
我的问题是关于最后两行:是否有更简洁的方式来表达这一点?我希望有这样的东西(理解这不是有效的Python原样):
def gen(list):
# ...
if list:
yield each in gen(list[1:])
Run Code Online (Sandbox Code Playgroud)
Python 3.3添加了yield from关键字. 这是您当前拥有的内容与使用 new 关键字的代码之间的比较:
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])))
Run Code Online (Sandbox Code Playgroud)
» python3 yield_from_test.py
[1, 2, 3]