Python中【yield】使用浅析

最新推荐文章于 2024-12-17 19:16:15 发布

原创

最新推荐文章于 2024-12-17 19:16:15 发布 · 241 阅读

0 ·

CC 4.0 BY-SA版权

本文介绍了Python中的yield关键字，通过斐波那契数列的例子展示了如何使用yield将函数转换为生成器，从而实现迭代功能。使用yield可以保持代码简洁并有效控制内存占用。同时，文章提到了isgeneratorfunction函数用于判断一个函数是否是生成器函数，以及生成器实例的特性，如在for循环中自动调用next()方法。最后，通过文件读取的例子进一步说明yield的用途。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

您可能听说过，带有 yield 的函数在 Python 中被称之为 generator（生成器），何谓 generator ？

我们先抛开 generator，以一个常见的编程题目来展示 yield 的概念。

如何生成斐波那契數列
斐波那契（Fibonacci）數列是一个非常简单的递归数列，除第一个和第二个数外，任意一个数都可由前两个数相加得到。用计算机程序输出斐波那契數列的前 N 个数是一个非常简单的问题，许多初学者都可以轻易写出如下函数：

清单 1. 简单输出斐波那契數列前 N 个数
1
2
3
4
5
6
def fab(max):
n, a, b = 0, 0, 1
while n < max:
print b
a, b = b, a + b
n = n + 1
执行 fab(5)，我们可以得到如下输出：

1
2
3
4
5
6

fab(5)
1
1
2
3
5
结果没有问题，但有经验的开发者会指出，直接在 fab 函数中用 print 打印数字会导致该函数可复用性较差，因为 fab 函数返回 None，其他函数无法获得该函数生成的数列。

要提高 fab 函数的可复用性，最好不要直接打印出数列，而是返回一个 List。以下是 fab 函数改写后的第二个版本：

清单 2. 输出斐波那契數列前 N 个数第二版
1
2
3
4
5
6
7
8
def fab(max):
n, a, b = 0, 0, 1
L = []
while n < max:
L.append(b)
a, b = b, a + b
n = n + 1
return L
可以使用如下方式打印出 fab 函数返回的 List：