问题描述
|
我有一个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:])
解决方法
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,3]))
print(list(gen_yield([1,3])))
» python3 yield_from_test.py
[1,3]
, 有人呼吁“ 2”之类的东西“传递”子生成器返回的所有值。请参阅PEP 380了解一些已经出现的想法。但是,尚未执行任何操作。您的第一个例子是正确的。
, 您的代码示例非常惯用和简洁,没有必要,也没有真正的机会进行进一步的改进,尤其是在可读性方面。
, 您的第一个示例是正确的解决方案。