在 Python 中,生成器(generator)是一种特殊的迭代器,通过 yield
关键字定义。生成器函数在调用时不会立即执行代码,而是返回一个生成器对象。这个对象实现了迭代器协议,可以逐步执行函数代码。
生成器对象有两个主要方法:next()
和 send()
。它们的作用和使用场景有所不同。
next()
- 作用: 从生成器中获取下一个值。
- 调用:
next(generator)
或generator.__next__()
- 使用场景: 当你只需要简单地获取生成器的下一个值时使用
next()
。 - 行为: 执行生成器函数的代码直到遇到下一个
yield
语句,并返回yield
语句中指定的值。如果生成器已经执行完所有代码,会抛出StopIteration
异常。
send()
- 作用: 向生成器发送一个值,并获取下一个
yield
语句的值。 - 调用:
generator.send(value)
- 使用场景: 当你需要向生成器中发送一个值(通常用于协程或更复杂的生成器逻辑)时使用
send()
。 - 行为: 执行生成器函数的代码直到遇到下一个
yield
语句,并返回yield
语句中指定的值。send()
方法会将传递的值作为生成器中上一次暂停的yield
表达式的结果。首次调用生成器时,必须传递None
,因为生成器还没有遇到任何yield
表达式。
总结
next()
方法用于简单地获取生成器的下一个值。send()
方法除了获取下一个值,还可以向生成器发送一个值,通常用于更复杂的生成器逻辑。
这两个方法可以结合使用,根据具体需求选择合适的方法。
End
关注公众号,测试干货及时送达