带有返回语句的生成器[重复]

Question

在我的报道中，我对以下案例（python 3.4）摸不着头脑

def simple_gen_function(str_in, sep=""):
    if sep == "":
        yield str_in[0]
        for c in str_in[1:]:
            yield c
    else:
        return str_in
        # yield from str_in

str_in = "je teste "
t = "".join(simple_gen_function(str_in))
p = "".join(simple_gen_function(str_in, "\n"))

print("%r %r" % (t, p))
# 'je teste' ''

在生成器中使用 return 时，使用 yield from str_in 时没有“达到”返回值我得到了预期的结果。

这个问题看起来很简单，但我相信在生成器中使用 return 就可以了。

Answer 1

yield 在函数体中的存在会将其变成生成器函数，而不是普通函数。在生成器函数中，使用return 是一种表示“生成器已结束，没有更多元素”的方式。通过将生成器方法的第一条语句设为return str_in，可以保证生成器不会返回任何元素。

正如评论中提到的，返回值 用作生成器结束时引发的StopIteration 异常的参数。见：

>>> gen = simple_gen_function("hello", "foo")
>>> next(gen)
Traceback (most recent call last):
  File "<stdin>", line 1, in <module>
StopIteration: hello

如果您的def 中有yield，它就是一个生成器！

在 cmets 中，提问者提到他们认为函数在执行 yield 语句时动态地变成了生成器。但这不是它的工作原理！决定是在代码执行之前做出的。如果 Python 在您的 def 下的任何地方找到 yield，它会将 def 转换为生成器函数。

请看这个超浓缩的例子：

>>> def foo():
...     if False:
...         yield "bar"
...     return "baz"
>>> foo()
<generator object foo at ...>
>>> # The return value "baz" is only exposed via StopIteration
>>> # You probably shouldn't use this behavior.
>>> next(foo())
Traceback (most recent call last):
  ...
StopIteration: baz
>>> # Nothing is ever yielded from the generator, so it generates no values.
>>> list(foo())
[]