Python生成器是一种特殊的迭代器,它允许你在需要时才生成值,从而节省内存并提高效率
yield
关键字:在定义生成器函数时,使用yield
关键字返回一个值。当生成器函数被调用时,它返回一个生成器对象,而不是直接执行函数体。当生成器对象被迭代时,函数体将被执行,直到遇到yield
关键字,此时函数将返回yield
后面的值,并暂停执行。下次迭代时,函数将从暂停的地方继续执行,直到再次遇到yield
关键字。def simple_generator():
yield 1
yield 2
yield 3
gen = simple_generator()
for value in gen:
print(value)
for
循环迭代:生成器对象可以直接用于for
循环进行迭代。def simple_generator():
yield 1
yield 2
yield 3
for value in simple_generator():
print(value)
next()
函数迭代:可以使用next()
函数获取生成器对象的下一个值。当生成器对象中没有更多值时,next()
函数将引发StopIteration
异常。def simple_generator():
yield 1
yield 2
yield 3
gen = simple_generator()
while True:
try:
value = next(gen)
print(value)
except StopIteration:
break
send()
方法发送值:生成器对象还支持send()
方法,允许你在迭代过程中向生成器发送值。当生成器接收到一个值时,它将在下一次yield
处返回该值。def simple_generator():
x = 0
while True:
value = yield x
if value is not None:
x = value
else:
break
gen = simple_generator()
print(next(gen)) # 输出 0
print(gen.send(5)) # 输出 5
print(next(gen)) # 输出 5
通过这些方法,Python生成器可以实现高效的迭代,节省内存并提高程序性能。