Python生成器是一种特殊的迭代器,它允许你在需要时生成值,而不是一次性生成所有值
yield
关键字:在定义生成器函数时,使用yield
关键字而不是return
关键字来返回一个值。当生成器函数被调用时,它返回一个生成器对象,但不会立即执行函数体。当生成器对象被迭代时,函数体将被执行,直到遇到yield
关键字,此时函数将返回yield
后面的值,并暂停执行。下次迭代时,函数将从暂停的地方继续执行,直到再次遇到yield
关键字。def my_generator():
yield 1
yield 2
yield 3
gen = my_generator()
print(next(gen)) # 输出 1
print(next(gen)) # 输出 2
print(next(gen)) # 输出 3
for
循环:使用for
循环迭代生成器对象可以确保每次迭代都获取一个新的值,而不是一次性获取所有值。这样可以避免内存不足的问题,特别是在处理大量数据时。def my_generator():
yield 1
yield 2
yield 3
gen = my_generator()
for value in gen:
print(value) # 依次输出 1、2、3
next()
函数:通过调用生成器对象的next()
方法,可以逐个获取生成器中的值。当生成器中没有更多值时,next()
方法将引发StopIteration
异常,你可以使用try-except
语句来处理这种情况。def my_generator():
yield 1
yield 2
yield 3
gen = my_generator()
try:
while True:
value = next(gen)
print(value) # 依次输出 1、2、3
except StopIteration:
pass
通过以上方法,Python生成器可以确保在迭代过程中准确地生成和返回值。