我经常需要写一些比较复杂的代码,常常会遇到各种各样的问题。比如我在使用yield from
表达式时,return
语句的问题。我们知道,在使用 yield from
表达式时,return
语句的作用是在子生成器(被调用的生成器)执行完毕后,返回最终的结果到调用者。这可以让生成器在嵌套结构中更清晰地传递值。具体情况我会一一用文字记录下来,方便后期参考:
1、问题背景
使用 "yield from" 表达式时,return 语句的作用是什么?我在很多示例中都没有找到 return 语句从 yield from 表达式返回的值。我尝试了以下简单的代码,但没有成功:
代码语言:javascript复制def return4():
return 4
def yield_from():
res = yield from range(4)
res = yield from return4()
def test_yield_from():
for x in yield_from():
print(x)
test_yield_from()
运行该代码会产生以下输出:
» python test.py 0 1 2 3 Traceback (most recent call last): File "test.py", line 52, in <module> test_yield_from() File "test.py", line 48, in test_yield_from for x in yield_from(): File "test.py", line 44, in yield_from res = yield from return4() TypeError: 'int' object is not iterable
但我希望的输出是:
» python test.py 0 1 2 3 4
因为 PEP 中有这样一段说明:
此外,当迭代器是另一个生成器时,子生成器被允许执行带有值的 return 语句,该值变为 yield from 表达式的值。
很显然,我没有得到这种解释。在 "yield from" 语句中,子生成器中的 return 语句是如何工作的?
2、解决方案
答案1: 生成器在耗尽时可以返回一个值:
代码语言:javascript复制def my_gen():
yield 0
return "done"
g = my_gen()
next(g)
next(g) # raises StopIteration: "done"
在 yield from 语句中返回的值将是此值。例如:
代码语言:javascript复制def yield_from():
res = yield from my_gen()
assert res == "done"
默认情况下,此值为 None。即 res = yield from range(4) 会将 res 设置为 None。
答案2:
yield from generator 的简写为:
代码语言:javascript复制for i in generator:
yield i
好吧,它比这复杂一点:https://www.python.org/dev/peps/pep-0380/#formal-semantics 。如果 generator = 4,则它将无法正常工作。(你的 return4() 不是一个生成器,而是一个函数。)为了得到你想要的结果,你可以这样做:
代码语言:javascript复制def yield_from():
yield from range(4)
yield 4
答案3:
我为你提供了测试的一个工作示例。return4 函数现在是一个生成器。为实现这一目标,必须在函数中任何地方出现 yield(Python 3.5 中有一个新的相关特性,但现在并不重要)。
正如你已经引用的:
此外,当迭代器是另一个生成器时,子生成器被允许执行带有值的 return 语句,该值变为 yield from 表达式的值。
总结:你将得到一个值。例如,你可以打印它:
代码语言:javascript复制def yield_from():
# ...
val = yield from return4()
print("value:", val) # prints value: 4
但你希望的是 yield 它,而不是打印它。以下是完整的代码:
代码语言:javascript复制def return4():
if False:
yield None
return 4
def yield_from():
yield from range(4)
yield (yield from return4())
def test_yield_from():
for x in yield_from():
print(x)
test_yield_from()
# prints numbers 0 to 4
你可能会问自己,它有什么好处。当你只从生成器接收值时,几乎没有任何优势。但当你向生成器发送值时,yield from 是一个很棒的特性。尝试找到有关 Python 协程的良好解释。它很棒。
其实上面的问题我们知道,具体来说,当一个生成器函数中使用 yield from
调用另一个生成器时,return
语句的返回值会成为调用者获取的值。这就允许子生成器产生的值直接传递给调用者,而不需要在父生成器中一个一个地使用 yield
语句传递。所以后期有啥不懂的都可以评论区留言。