Python生成器实现斐波那契数列

Posted 2020-12-26 小学弟-

            比如，斐波那契数列：1，1，2，3，5，8，13，21，34....
            用列表生成式写不出来，但是我们可以用函数把它打印出来：
            def fib(number):
                n, a, b = 0, 0, 1
                while n < number:
                    print(b)
                    a, b = b, a + b
                    n = n + 1
                return ‘OK!‘
            print(fib(5))
            结果：
                1
                1
                2
                3
                5
                OK!
            我们可以看出从第一个元素开始，推算出后续任意的元素。很像generator。
            要把fib函数变成generator，只需要把 print(b)改为 yield b就可以了：
            def fib(number):
                n, a, b = 0, 0, 1
                while n < number:
                    yield b
                    a, b = b, a + b
                    n = n + 1
                return ‘OK!‘
 
            print(fib(5))#<generator object fib at 0x105606ca8>
            注意：
                这里难理解的就是generator和函数的执行流程是不一样的。
                函数是顺序执行，遇到return语句或者最后一行函数语句就
                返回。
                注意：函数创建一次生成一个生成器，所以我们会将创建的生
                    成器赋值给一个变量。如果直接用函数本身这个生成器，
                    我们没用一次生成一个新的生成器对象，所以，我们一
                    般都将创建的生成器赋给一个变量。
                generetor的函数，在每次调用 next()的时候执行，遇到
                yield语句返回，再次执行时从上次返回的yield语句处继续
                执行。