python实现斐波拉契数列

第一种

# 斐波拉契数列


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