当前位置：首页 > article >正文

Python生成器（send，close，throw）方法详解

article 2024/11/27 14:47:51

在Python中，生成器（generator）是一种特殊的迭代器，它使用函数来产生序列中的元素。
生成器函数使用yield关键字一次返回一个值，并在每次调用时暂停和恢复其状态。除了标准的迭代协议外，生成器对象还提供了一些额外的方法：send()、close()和throw()。下面是对这些方法的详细解释：

1. send(value)

功能：向生成器发送一个值，并恢复其执行直到下一个yield表达式。该值将成为yield表达式的结果。
参数：value是发送给生成器的值。
返回值：返回生成器中yield表达式的结果（如果有的话）。
异常：如果生成器已经终止（例如，由于StopIteration或异常），则调用send()将引发StopIteration或相应的异常。
示例：

def my_generator():
    value = yield 1
    print(f'Received: {value}')
    yield value + 1
 
gen = my_generator()
print(next(gen))  # 输出: 1，启动生成器并返回第一个yield的值
gen.send(10)     # 输出: Received: 10，并返回第二个yield的值：11

2. close()

功能：关闭生成器，使其不再产生任何值。如果生成器尚未终止，则调用close()将引发GeneratorExit异常到生成器中。生成器可以选择捕获此异常并执行清理操作。
参数：无。
返回值：None。
注意：close()方法不会返回生成器产生的任何值或异常。它只是确保生成器被适当地关闭。
示例：

def my_generator():
    try:
        yield 1
        yield 2
        yield 3
    except GeneratorExit:
        print('Generator is being closed')
 
gen = my_generator()
print(next(gen))  # 输出: 1
print(next(gen))  # 输出: 2
gen.close()       # 输出: Generator is being closed

3. throw(type[, value[, traceback]])

功能：将指定的异常抛入生成器中，并在生成器内部引发。如果生成器未捕获该异常，则它将传播到调用者。
参数：
type：异常的类型。
value（可选）：异常的实例。如果没有提供，则使用type()创建一个新的实例。
traceback（可选）：一个traceback对象，用于表示异常的堆栈跟踪。通常不需要手动提供。
返回值：如果生成器捕获并处理了异常，则返回生成器中yield表达式的结果（如果有的话）。否则，传播异常。
示例：

def my_generator():
    try:
        yield 1
        yield 2
        raise ValueError('This will not be raised')
    except ValueError as e:
        print(f'Caught ValueError: {e}')
        yield 3  # 继续产生值
 
gen = my_generator()
print(next(gen))  # 输出: 1
gen.throw(ValueError, 'Manually thrown exception')
# 输出: Caught ValueError: Manually thrown exception
# 并返回下一个yield的值：3