Python中生成器的next和send方法有什么区别

在 Python 中，生成器（generator）是一种特殊的迭代器，通过 yield 关键字定义。生成器函数在调用时不会立即执行代码，而是返回一个生成器对象。这个对象实现了迭代器协议，可以逐步执行函数代码。

生成器对象有两个主要方法：next() 和 send()。它们的作用和使用场景有所不同。

next()

• 作用: 从生成器中获取下一个值。
• 调用: next(generator) 或 generator.__next__()
• 使用场景: 当你只需要简单地获取生成器的下一个值时使用 next()。
• 行为: 执行生成器函数的代码直到遇到下一个 yield 语句，并返回 yield 语句中指定的值。如果生成器已经执行完所有代码，会抛出 StopIteration 异常。

def simple_generator():
    yield 1
    yield 2
    yield 3


gen = simple_generator()
print(next(gen))  # 输出: 1
print(next(gen))  # 输出: 2
print(gen.__next__())  # 输出: 3
next(gen)  # 如果再调用 next(gen)，将抛出 StopIteration 异常

send()

• 作用: 向生成器发送一个值，并获取下一个 yield 语句的值。
• 调用: generator.send(value)
• 使用场景: 当你需要向生成器中发送一个值（通常用于协程或更复杂的生成器逻辑）时使用 send()。
• 行为: 执行生成器函数的代码直到遇到下一个 yield 语句，并返回 yield 语句中指定的值。send() 方法会将传递的值作为生成器中上一次暂停的 yield 表达式的结果。首次调用生成器时，必须传递 None，因为生成器还没有遇到任何 yield 表达式。

def simple_coroutine():
    print("Start coroutine")
    x = yield 1
    print(f"Received: {x}")
    y = yield 2
    print(f"Received: {y}")


gen = simple_coroutine()
# 首次调用生成器时，必须传递 None，因为生成器还没有遇到任何 yield 表达式。传其他值会报错
print(gen.send(None)) # 输出: Start coroutine, 然后输出: 1
print(gen.send(10))  # 输出: Received: 10, 然后输出: 2
print(gen.send(20))  # 输出: Received: 20, 然后生成器结束，抛出 StopIteration 异常

总结

• next() 方法用于简单地获取生成器的下一个值。
• send() 方法除了获取下一个值，还可以向生成器发送一个值，通常用于更复杂的生成器逻辑。

这两个方法可以结合使用，根据具体需求选择合适的方法。