在生成器中返回和返回 None：PEP 指南

Question

根据 PEP 8，我们应该在函数声明中保持一致，并确保它们都具有相同的返回模式，即所有都应该返回一个表达式，或者都不应该返回。但是，我不确定如何将其应用于生成器。

只要代码到达它们，生成器就会yield 值，除非遇到return 语句，在这种情况下它将停止迭代。但是，我看不到任何从生成器函数返回值的用例。本着这种精神，从 PEP 8 的角度来看，我不明白为什么用明确的 return None 结束这样的功能是有用的。换句话说，如果只有在 yield'ing 结束时才达到 return 表达式，为什么我们应该为生成器表达一个 return 语句？

示例：在以下代码中，我看不到如何使用hello() 将100 分配给变量（因此使用return 语句）。那么为什么 PEP 8 期望我们写一个 return 语句（无论是 100 还是 None）。

def hello():
    for i in range(5):
      yield i

    return 100


h = [x for x in hello()]
g = hello()

print(h)    
# [0, 1, 2, 3, 4]
print(g)
# <generator object hello at 0x7fd2f285a7d8>
# can we ever get 100?

Answer 1

您误读了 PEP8。 PEP8 声明：

在返回语句中保持一致。 函数中的所有 return 语句都应该返回一个表达式，或者它们都不应该返回。

（我的粗体强调）

您应该与在单个函数中使用 return 的方式保持一致，而不是在整个项目中使用。

使用return，它是函数中唯一的return 语句。

但是，我没有看到任何从生成器函数返回值的用例。

生成器的返回值附加到引发的StopIteration 异常：

>>> def gen():
...     if False: yield
...     return 'Return value'
...
>>> try:
...     next(gen())
... except StopIteration as ex:
...     print(ex.value)
...
Return value

这也是yield from产生值的机制； yield from 的返回值是 StopIteration 异常的 value 属性。因此，生成器可以使用return result 将结果返回给使用result = yield from generator 的代码：

>>> def bar():
...     result = yield from gen()
...     print('gen() returned', result)
...
>>> next(bar(), None)
gen() returned Return value

此功能在 Python 标准库中使用；例如在asyncio 库中，StopIteration 的值用于传递Task 结果，@coroutine 装饰器使用res = yield from ... 运行包装的生成器或等待并传递返回值。

所以，从 PEP-8 的角度来看，对于生成器，有两种可能性：

• 您正在使用return 提前退出生成器，例如在使用if 的循环中。使用return，无需添加None：

  def foo():
      while bar:
          yield ham
          if spam:
              return
  

• 您正在使用return <something> 退出并设置StopIteration.value。在整个生成器中始终使用return <something>，即使返回None：

  def foo():
      for bar in baz:
          yield bar
          if spam:
              return 'The bar bazzed the spam'
      return None