本文共 1256 字，大约阅读时间需要 4 分钟。

Python 的 yield 语句：浅入浅出

作为一个开发者，你或许已经听说过 Python 中的 yield 语句。这个语句在编写迭代器（iterator）时非常有用，它可以帮助你定义懒序列（Lazy Sequence），从而在需要时才生成数据，避免了一次性占用大量内存。

生成器与迭代器

在 Python 中，任何迭代器都可以用 list 或列表推导式（list comprehension）来处理。然而，当面对非常大的列表或无限列表时，直接生成完整的列表可能并不可行。这时候，懒序列的概念就派上用场了。yield 语句正是用来方便地定义这样的迭代器。

传统方法的局限

以 Fibonacci 序列为例，传统的做法是将每个数添加到一个列表中并返回。这种方法虽然简单，但在生成完整列表时必须执行所有计算，导致内存占用增加。如果你只需要前几项或处理非常大的数据，这种方法显然不够灵活。

生成器的优势

生成器可以帮助你创建灵活的迭代器。例如，下面的代码可以生成 Fibonacci 序列的前 n 项：

class Fab(object):    def __init__(self, max):        self.max = max        self.n, self.a, self.b = 0, 0, 1    def __iter__(self):        return self    def next(self):        if self.n < self.max:            r = self.b            self.a, self.b = self.b, self.a + self.b            self.n = self.n + 1            return r        raise StopIteration()

使用生成器，你可以逐个获取所需的项，非常适合处理大文件或其他输入流。

使用 yield 优化代码

实际上，yield 语句可以显著简化上述生成器代码。例如，下面是使用 yield 优化后的 Fibonacci 生成器：

def fab(max):    n, a, b = 0, 0, 1    while n < max:        yield b        a, b = b, a + b        n = n + 1

这段代码与传统生成器代码相比，更加简洁易读。通过 yield 语句，你可以按需获取数据，而无需预先生成完整的列表。

懒序列的应用场景

懒序列的概念不仅适用于 Fibonacci 序列，还可以扩展到文件读取、网络请求等场景。你可以按需读取数据，避免一次性加载过多内容，这对于处理大型数据或无限流数据尤为重要。

通过上述方法，你可以轻松地创建高效的迭代器，满足各种懒序列需求。yield 语句不仅简化了代码编写，还提升了性能和灵活性，是每个Python 开发人员的必备工具。

转载地址：http://ngafk.baihongyu.com/