同一函数中的 return 和 Yield

问题描述：

当在 Python 中的同一个函数中使用 Yield 和 Return 时，究竟会发生什么呢？就像这样？

def find_all(a_str, sub):
    start = 0
    while True:
        start = a_str.find(sub, start)
        if start == -1: return
        yield start
        start += len(sub) # use start += 1 to find overlapping matches

它还是发电机吗？

解决方案 1：

是的，它仍然是一个生成器。return几乎相当于提高StopIteration。

PEP 255明确指出：

规格：返回

生成器函数还可以包含以下形式的返回语句：

"return"

请注意，生成器主体中的返回语句上不允许使用 expression_list（当然，它们可以出现在生成器内嵌套的非生成器函数主体中）。

当遇到 return 语句时，控制权将像任何函数返回一样继续执行，执行适当的 finally 子句（如果存在）。然后引发 StopIteration 异常，表示迭代器已耗尽。如果控制权流出生成器的末尾而没有显式返回，也会引发 StopIteration 异常。

请注意，对于生成器函数和非生成器函数来说，return 表示“我已完成，并且没有什么有趣的东西可以返回”。

请注意，return 并不总是等同于引发 StopIteration：区别在于如何处理封闭的 try/except 构造。例如，

>>> def f1():
...     try:
...         return
...     except:
...        yield 1
>>> print list(f1())
[]

因为，就像在任何函数中一样，return 只是退出，但是

>>> def f2():
...     try:
...         raise StopIteration
...     except:
...         yield 42
>>> print list(f2())
[42]

因为 StopIteration 被一个裸的“except”捕获，就像任何异常一样。

解决方案 2：

是的，它仍然是一个生成器。可以使用空的return或return None来结束生成器函数。它相当于引发StopIteration（有关详细信息，请参阅@NPE 的回答）。

SyntaxError请注意，在 Python 3.3 之前的版本中，返回非 None 参数。

正如 @BrenBarn 在评论中指出的那样，从 Python 3.3 开始，返回值现在传递给StopIteration.

来自PEP 380：

在生成器中，语句

return value

在语义上等同于

raise StopIteration(value)

解决方案 3：

有一种方法可以实现函数中的yield和return方法，允许您返回一个值或生成器。

它可能不像您想要的那样干净，但它确实可以达到您的期望。

以下是一个例子：

def six(how_many=None):
    if how_many is None or how_many < 1:
        return None  # returns value

    if how_many == 1:
        return 6  # returns value

    def iter_func():
        for count in range(how_many):
            yield 6
    return iter_func()  # returns generator

解决方案 4：

StopIteration注意：下面的示例不会出现异常。

def odd(max):
    n = 0
    while n < max:
        yield n
        n = n + 1
    return 'done'


for x in odd(3):
    print(x)

环for抓住了它。这是它停止的信号

但你可以用这种方法捕捉它：

g = odd(3)

while True:
    try:
        x = next(g)
        print(x)
    except StopIteration as e:
        print("g return value:", e.value)
        break