在 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)邀请你们进群喔。

关注公众号，测试干货及时送达

小博测试成长之路

将在本公众号上持续为大家分享自己从小白开始入门学习软件测试的一些学习笔记和心得，包括业务、功能、自动化和性能测试等方面。通过此平台也结识了很多人脉，关注我，带你进群和小伙伴一起交流。

如果想查看博主的所有文章列表，可以在公众号的菜单那里进行点击查看：