深入理解Python中的 `yield` 和 `yield from`

引言

在Python中，yield 和 yield from 都是用于生成器函数的关键字。虽然它们看起来相似，但实际上功能和使用场景大相径庭。通过一些具体的例子，我们来深入了解这两个关键字的区别。

yield 的使用及例子

yield 关键字用于一个生成器函数中，以产生一个值。当生成器遇到 yield 时，它会返回一个值给调用者，并在下次调用时从当前位置继续执行。

例子1: 简单的数字生成器

def simple_generator():
    yield 1
    yield 2
    yield 3

for value in simple_generator():
    print(value)

输出:

1
2
3

在这个例子中，生成器 simple_generator 依次产生1, 2, 3。每次调用时，它通过 yield 返回当前的值，并在下次迭代时从停止的地方继续。

yield from 的使用及例子

yield from 关键字在Python 3.3中引入，用于在一个生成器中委派另一个生成器。它不仅返回子生成器的所有值，还能处理子生成器抛出的异常。

例子2: 委派生成器

def sub_generator():
    yield 1
    yield 2

def delegating_generator():
    yield from sub_generator()
    yield 3

for value in delegating_generator():
    print(value)

输出:

1
2
3

在这个例子中，delegating_generator 通过 yield from 将执行委派给 sub_generator。sub_generator 首先产生1和2，然后控制权回到 delegating_generator 继续产生3。

yield 和 yield from 的关键区别

1. 委派操作: yield from 主要用于委派操作，当一个生成器需要调用另一个生成器时。这在处理嵌套生成器时非常有用。

2. 异常传递: yield from 还负责传递异常，这对于编写涉及多个生成器的复杂代码非常重要。

3. 代码简化: 使用 yield from 可以简化代码，使其更易于阅读和维护。

结论

理解 yield 和 yield from 的区别对于编写高效且易于维护的Python生成器函数至关重要。yield 适用于简单的单步生成操作，而 yield from 更适用于复杂的生成器委派和管理。掌握这些概念将有助于您更好地利用Python的强大功能。