在python中,一边循环一边计算的机制,就是生成器(generator)。
生成器的实现方法:
1、第一种:
gen = (i for i in range(10))
print(gen)
print(type(gen))
# result:<generator object <genexpr> at 0x0000015C28282A20>
# result:<class 'generator'>
我们可以通过next()
函数获得generator的返回值
print(next(gen))
# result: 0
print(next(gen))
# result: 1
print(next(gen))
# result: 2
print(next(gen))
# result: 3
print(next(gen))
# result: 4
print(next(gen))
# result: 5
print(next(gen))
# result: 6
print(next(gen))
# result: 7
print(next(gen))
# result: 8
print(next(gen))
# result: 9
每次调用next()
函数就会计算出生成器gen的下一个元素的值,当没有更多元素时,再执行next()
函数,就会抛出StopIteration
的错误。
同时,我们也可以使用for
循环来遍历生成器
for i in gen:
print(i,end=' ')
# result: 0 1 2 3 4 5 6 7 8 9()
2、第二种
def func(number=0):
n,a,b = 0,0,1
while n < number:
yield b
a,b = b,a+b
n += 1
f = func(6)
print(type(f))
# result:<class 'generator'>
我们可以通过next()
函数获得generator的返回值
print(next(f))
# result: 1
print(next(f))
# result: 1
print(next(f))
# result: 2
生成器函数和普通函数的执行流程不一样。普通函数是顺序执行,遇到return语句或者最后一行函数语句就返回。而生成器函数,在每次调用next()的时候执行,遇到yield语句返回,再次执行时从上次返回的yield语句处继续执行。
每次调用next()
函数就会计算出生成器gen的下一个元素的值,当没有更多元素时,再执行next()
函数,就会抛出StopIteration
的错误。
同时,我们也可以使用for
循环来遍历生成器
for i in f:
print(i,end=' '))
# result: 1 1 2 3 5 8