如果生成器函数需要产出另一个生成器生成的值,传统的解决方法是使用嵌套的 for 循环。

例如,下面是我们自己实现的 chain 生成器:13

13标准库中的 itertools.chain 函数是使用 C 语言编写的。

    >>> def chain(*iterables):
    ...     for it in iterables:
    ...         for i in it:
    ...             yield i
    ...
    >>> s = 'ABC'
    >>> t = tuple(range(3))
    >>> list(chain(s, t))
    ['A', 'B', 'C', 0, 1, 2]

chain 生成器函数把操作依次交给接收到的各个可迭代对象处理。为此,“PEP 380 — Syntax for Delegating to a Subgenerator”(https://www.python.org/dev/peps/pep-0380/)引入了一个新句法,如下述控制台中的代码清单所示:

    >>> def chain(*iterables):
    ...     for i in iterables:
    ...         yield from i
    ...
    >>> list(chain(s, t))
    ['A', 'B', 'C', 0, 1, 2]

可以看出,yield from i 完全代替了内层的 for 循环。在这个示例中使用 yield from 是对的,而且代码读起来更顺畅,不过感觉更像是语法糖。除了代替循环之外,yield from 还会创建通道,把内层生成器直接与外层生成器的客户端联系起来。把生成器当成协程使用时,这个通道特别重要,不仅能为客户端代码生成值,还能使用客户端代码提供的值。第 16 章会深入讲解协程,其中有几页会说明为什么 yield from 不只是语法糖而已。

一瞥 yield from 之后,我们回过头继续复习标准库中善于处理可迭代对象的函数。