如何定义一个生成器函数?

生成器函数是一种特殊的函数，它通过 yield 语句将结果产生给调用者，而不是通过 return 语句返回结果。

生成器函数的定义与普通函数类似，使用 def 关键字，并在函数体内部使用 yield 关键字来生成值。当生成器函数被调用时，它会返回一个生成器对象，而不是立即执行函数体内的代码。

下面是一个简单的例子来定义一个生成器函数：

def count_up_to(n):
    i = 1
    while i <= n:
        yield i
        i += 1

在这个例子中，count_up_to 是一个生成器函数，它的作用是产生从 1 到 n 之间的所有整数。它使用一个 while 循环来递增变量 i，并通过 yield 关键字将当前的 i 值产生给调用者。此时函数会暂停，并保留当前的状态，等待下一次的调用。

调用生成器函数时，会得到一个生成器对象：

gen = count_up_to(5)

生成器对象是一个迭代器，可以使用 next 函数或 for 循环进行迭代。每次调用 next 函数时，生成器函数会从上一次暂停的地方继续执行，直到遇到下一个 yield 语句为止。

print(next(gen))  # 输出 1
print(next(gen))  # 输出 2
print(next(gen))  # 输出 3

for num in gen:
    print(num)    # 输出 4 和 5

生成器函数的特点是它只会在需要时生成值，而不是一次性计算出所有的结果。这在处理大量数据或需要逐步生成结果的情况下非常有用，可以减少内存的使用并提高程序的效率。

此外，生成器函数还可以使用多个 yield 语句来生成多个结果，通过在循环中不断调用 next 函数可以逐步获取这些结果。

综上所述，生成器函数是一种通过 yield 语句逐步生成结果的特殊函数，它的调用会返回一个生成器对象，可以通过 next 函数或 for 循环进行迭代。生成器函数的特点是它只会在需要时生成值，这让它成为处理大量数据或逐步生成结果的优秀工具。