在python中,一边循环一边计算的机制,就是生成器(generator)。
生成器的实现方法:
1、第一种:
gen = (i for i in range(10)) print(gen) print(type(gen)) # result:at 0x0000015C28282A20> # result:
我们可以通过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:
我们可以通过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



