比如,斐波那契数列: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语句处继续 执行。
原文地址:https://www.cnblogs.com/xiao-xue-di/p/9399314.html
时间: 2024-10-09 08:37:50