yield语句用于定义生成器函数,在函数中可以使用send、close和throw方法来控制生成器的执行。
send(value):将value发送给生成器,该值会成为生成器当前暂停位置yield表达式的结果,并且在下一次迭代时作为yield语句的返回值。如果在第一次调用send方法之前调用了next方法,则该方法的效果与next方法相同。
close():关闭生成器,并向其发送一个GeneratorExit异常。当生成器被关闭后,任何进一步尝试发送数据到它或从它接收数据都将引发StopIteration异常。
throw(type, value=None, traceback=None):向正在运行的生成器抛出指定类型和值的异常,并终止该生成器。如果指定了值,则它将成为异常对象的参数传递给捕获该异常的except子句。如果未指定值,则使用默认值None。如果指定了traceback,则应该是一个traceback对象,否则Python会自动创建一个traceback对象。
以下是一个简单示例,展示如何使用这三个方法来控制生成器:
def my_generator():
while True:
try:
x = yield
print("Received:", x)
except ValueError as e:
print("Caught an exception:", e)
except GeneratorExit:
print("Closing the generator")
return
g = my_generator()
next(g) # 等价于 g.send(None)
g.send(10) # 输出 "Received: 10"
g.throw(ValueError("Something went wrong")) # 输出 "Caught an exception: Something went wrong"
g.close() # 输出 "Closing the generator"