本文通过一个很简单的例子,即杨辉三角,来说明python的生成器与迭代器语法,熟悉它的运行流程以及注意事项。
生成器:generator,主要是处理无限大的序列,但不希望采用传统的生成再存储的方式,因为这样极度占用内存与存储空间,因此可以看做是个懒人函数,需要的时候再调用,未调用时可以说只存储一个生成的表达式。
如s = (x*x for x in range(10))print(s)
这里执行结果并不是所期望得到的列表,而是 at 0x00000273C3773C78> ,如果需要得到列表,我们需要加上:for i in range(10):print(next(i))
注意这里有next函数,如果执行完毕再继续next,就会抛出Iteration Error。为了规避,可以用for循环如下语句:for i in s:print(i)
生成器关键字 yield, 如斐波那契数列:def fib(max):n,a,b =0,1,1while n < max:yield aa,b =b,a+bn = n+1return 'done'
a = fib(10)for i in a:print(i)
更进一步,尝试生成杨辉三角:(最早期写了个特别锉的代码,鄙视一下自己: https://blog.csdn.net/weixin_44521703/article/details/91043327)def triangles():N = [1]while True:yield NN = [1] + [N[i] + N[i+1] for i in range(len(N) - 1) ]+ [1]
n = 0results = []for t in triangles():print(t)results.append(t)n = n + 1if n == 10:break
生成结果:[1][1, 1][1, 2, 1][1, 3, 3, 1][1, 4, 6, 4, 1][1, 5, 10, 10, 5, 1][1, 6, 15, 20, 15, 6, 1][1, 7, 21, 35, 35, 21, 7, 1][1, 8, 28, 56, 70, 56, 28, 8, 1][1, 9, 36, 84, 126, 126, 84, 36, 9, 1]