举例详解Python中yield生成器的用法

下面是针对Python中yield生成器的用法的完整攻略。

什么是yield生成器

在Python中，使用yield可以定义生成器函数(generator function)。

生成器函数每次调用时，返回一个生成器(generator)对象。生成器支持迭代器协议，能够被for-in循环调用，也能用next()函数获取下一个值。生成器每次返回一个值后会“暂停”，保留当前状态，并保存临时变量；下次调用时，会从当前状态继续往下执行。

yield是Python中用来创建生成器的关键字，用于将函数定义为生成器。yield语句不仅会将值返回给调用方，还会“冻结”生成器的状态，以便下次使用时继续运行。

yield生成器的用法

下面是一个简单的使用yield生成器的例子：

def generate_nums(n):
    for i in range(n):
        yield i

gen = generate_nums(5)
for num in gen:
    print(num)

在上面的例子中，我们定义了一个生成器函数generate_nums，该函数使用for循环遍历一组数字，并逐个返回每个数字。

我们通过调用generate_nums函数并向其传递参数，创建一个生成器对象gen。然后，我们使用for循环遍历生成器对象，并使用print语句打印每个返回的数字，最终输出结果为：

0
1
2
3
4

从输出结果可以看出，每次调用生成器对象返回一个数字，并且在生成下一个数字之前会“暂停”函数的执行状态，保存当前环境并保留相应的变量。

另一个例子如下：

def read_file(filename):
    with open(filename, 'r') as file:
        for line in file:
            yield line

gen = read_file('example.txt')
for line in gen:
    print(line)

在上面的例子中，我们定义了一个生成器函数read_file，该函数用于逐行读取一个文件中的文本内容，并将每一行作为生成器对象的返回值。

我们通过调用read_file函数并传递文件名参数，创建一个生成器对象gen。然后，我们使用for循环遍历生成器对象，并使用print语句打印每个返回的行，最终输出整个文件的内容。

总结

在Python中，使用yield可以创建非常方便的生成器对象，使得我们可以按需生成需要的数据。yield不仅返回数据值，并短暂地挂起函数，暂停执行，等待下次调用之后，继续执行。这种机制不仅可以提高代码执行效率，还能够减少内存占用。