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,1
    while n < max:
        yield a
        a,b =b,a+b
        n = n+1
    return '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 N
        N = [1] + [N[i] + N[i+1] for i in range(len(N) - 1) ]+ [1]

n = 0
results = []
for t in triangles():
    print(t)
    results.append(t)
    n = n + 1
    if 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年8月3日)

今天看到一个很好的思路:

比如上一行是[1,2,1],下一行可以这么构造,造两个列表,在其前面与后面分别加个0,即做成[0,1,2,1]与[1,2,1,0],两个列表主元素相加就是[1,3,3,1]

这个思路实际也很有技巧的,实现算法如下:

 

def generate(numRows):
    r = [[1]]
    for i in range(1, numRows):
        r.append(list(map(lambda x, y: x + y, [0] + r[-1], r[-1] + [0])))
    return r[:numRows]


a = generate(10)
for i in a:
    print(i)

输出结果是:

[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]

 

  • 1
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 2
    评论
评论 2
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值