当python遇上杨辉三角——生成器与迭代器

小草AI

发布于 2019-07-08 19:17:28

7220

发布于 2019-07-08 19:17:28

文章被收录于专栏：机器学习与python集中营

本文通过一个很简单的例子，即杨辉三角，来说明python的生成器与迭代器语法，熟悉它的运行流程以及注意事项。

生成器：generator，主要是处理无限大的序列，但不希望采用传统的生成再存储的方式，因为这样极度占用内存与存储空间，因此可以看做是个懒人函数，需要的时候再调用，未调用时可以说只存储一个生成的表达式。

如

s = (x*x for x in range(10))print(s)

这里执行结果并不是所期望得到的列表，而是 <generator object <genexpr> 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]

本文参与腾讯云自媒体同步曝光计划，分享自微信公众号。

原始发表：2019-07-02，如有侵权请联系 cloudcommunity@tencent.com 删除

javascript