在 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()方法除了获取下一个值，还可以向生成器发送一个值，通常用于更复杂的生成器逻辑。
这两个方法可以结合使用，根据具体需求选择合适的方法。
End
如果觉得文章对你有帮助的话，欢迎点赞转发~
如果还没有进粉丝交流群的小伙伴，赶快添加好友(xiaobotester)邀请你们进群喔。
👇👇👇
关注公众号，测试干货及时送达
如果想查看博主的所有文章列表，可以在公众号的菜单那里进行点击查看：