版权声明:本文为博主原创文章,未经博主允许不得转载。 https://blog.csdn.net/xqt15538076006/article/details/82320758
第一种
# 斐波拉契数列
def fib(max):
n, a, b = 0, 1, 1
while n < max:
print(b)
a, b = b, a+b
n = n+1
return 'done'
fib(5)
结果为
1
1
2
3
5
第二改成生成器模式
仔细观察,可以看出,fib函数实际上是定义了斐波拉契数列的推算规则,可以从第一个元素开始,推算出后续任意的元素,这种逻辑其实非常类似generator。
也就是说,上面的函数和generator仅一步之遥。要把fib函数变成generator,只需要把print(b)改为yield b就可以了:
def fib(max):
n, a, b = 0, 1, 1
while n < max:
yield b
a, b = b, a+b
n = n+1
return 'done'
f= fib(5)
print(f)
for i in f:
print(i)
结果为
<generator object fib at 0x00000170CFFD6A98>
1
2
3
5
8