在 Python 中,yield 是一个关键字,它用于定义生成器函数。生成器函数是一个特殊的函数,可以返回一个迭代器,当生成器函数被调用时,它不会立即执行,而是返回一个生成器对象,通过迭代生成器对象可以逐个获取生成器函数中的值。与普通函数不同的是,生成器函数不会一次性返回所有的结果,而是只返回当前需要的值,并在下次调用时继续执行函数直到完成所有的操作。
yield 的好处主要有两点:
延迟计算:yield 可以将一个大的计算任务分成若干个小的任务,每次只计算一小部分,然后将计算结果 yield 出来,等到下一次需要计算时再继续计算下一部分,这样可以节省内存,避免一次性计算过多的数据,从而提高程序的效率。
协程:yield 还可以用于实现协程,协程是一种轻量级的线程,它的特点是在一个线程中可以同时运行多个协程,协程之间切换时不需要线程切换的开销,因此可以提高程序的并发性能。通过 yield 可以将一个函数变成协程,将一个大的计算任务拆分成若干个小的任务,每次计算一小部分,并将结果 yield 出来,等到下一次需要计算时再继续计算下一部分,这样就可以实现异步执行,提高程序的并发性能。
以下是一个简单的示例代码:
def my_range(n):i = 0while i < n:yield ii += 1
在这个示例中,my_range 函数使用 yield 关键字创建了一个生成器函数,它可以迭代 0 到 n-1 的整数。当使用 for 循环迭代该生成器对象时,每次循环都会执行 yield 关键字,返回一个值并暂停函数执行,直到下一次迭代调用。
for i in my_range(5):print(i)
输出结果:
1
2
3
4
除了迭代器,yield 还可以用于协程和异步编程中,以实现在执行过程中暂停和恢复函数的执行。下面是一个简单的协程示例,展示了如何使用 yield 关键字暂停和恢复函数的执行。
def coroutine():while True:x = yieldprint(x)cor = coroutine()
next(cor) # 启动生成器
cor.send(1) # 输出 1
cor.send(2) # 输出 2
cor.send(3) # 输出 3
在这个示例中,coroutine 函数是一个协程,通过 yield 关键字实现了暂停和恢复函数执行的能力。当调用 cor.send(x) 方法时,会将 x 值赋值给 yield 左边的变量,同时暂停函数执行并返回结果,直到下一次调用 cor.send 时恢复执行。
执行上述代码后的输出结果如下:
1
2
3