Python yield方法详解|极客教程

Python yield方法详解

在Python中，yield是一个关键字，用于定义生成器函数。生成器函数是一种特殊类型的函数，它可以在迭代时暂停执行，返回一个中间结果，然后在下次迭代时恢复执行。yield方法非常适合用于处理大量数据或需要延迟计算的情况。在本文中，我们将详细介绍yield方法的用法和示例。

yield的基本用法

首先，让我们来看一个简单的示例，演示yield方法的基本用法：

def my_generator():
    yield 1
    yield 2
    yield 3

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

在上面的代码中，我们定义了一个名为my_generator的生成器函数，该函数使用yield关键字返回了三个值：1、2和3。在调用生成器函数时，我们通过next()方法逐个获取生成器函数返回的值。

yield的用途

1. 生成无限序列

使用yield方法可以方便地生成无限序列，例如生成所有的自然数：

def natural_numbers():
    n = 1
    while True:
        yield n
        n += 1

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

上述代码中，natural_numbers生成器函数会不断产生自然数，直到无限循环结束。

2. 处理大量数据

yield方法也可以用于处理大量数据，避免一次性加载所有数据到内存中，节省内存空间。例如，读取文件中的每一行：

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

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

上述代码中，read_file生成器函数每次从文件中读取一行数据，并使用yield关键字返回。在迭代生成器函数时，只会读取一行数据并打印，而不会一次性读取整个文件。

3. 实现协程

协程是一种轻量级的线程，可以在多个任务之间切换执行。yield方法可以用于实现协程，让程序在不同任务之间切换执行。

def task1():
    for i in range(5):
        print('Task 1 -', i)
        yield

def task2():
    for i in range(5):
        print('Task 2 -', i)
        yield

t1 = task1()
t2 = task2()

for _ in range(5):
    next(t1)
    next(t2)

上述代码中，task1和task2是两个协程函数，通过yield方法可以在两个任务之间切换执行，实现并发操作。