【Python】生成器和杨辉三角

来源:互联网 发布:新零售软件 编辑:程序博客网 时间:2024/06/05 14:19

生成器

在python中,一边循环一边计算的机制,称为生成器(generator)。generator保存的是算法,它也是可迭代对象。
创建generator:
方法一:将一个列表生成式的[]改成()。
方法二:将print(b)改为yield b。
执行过程:
generator和函数的执行流程不一样。函数是顺序执行,遇到return语句或者最后一行函数语句就返回。而变成generator的函数,在每次调用next()的时候执行,遇到yield语句返回,再次执行时从上次返回的yield语句处继续执行。利用for循环来迭代打印。

练习:杨辉三角

把每一行看做一个list,试写一个generator,不断输出下一行的list:

# 期待输出:# [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]n = 0for t in triangles():    print(t)    n = n + 1    if n == 10:        break

以下代码是”@其实我姓谢“同学的。
代码如下:

def triangles():    L = [1]         while True:        yield L        L.append(0)        L = [L[i-1]+L[i] for i in range(len(L))]n=0a = input("请输入杨辉三角的行数:")for t in triangles():    print(t)    n = n+1    if n == int(a):        break

【分析】

L = [1]每一行第一个元素都是1。进入循环。打印L。对与N行来说,只要知道N-1行,就可以得出N行的元素,因为每一行的元素都由上一行的元素的左右元素相加得到的,然而N行的最后一个元素由N-1的最后一个元素再加0得到的,因为N-1少一个元素,所以就是添加0的作用。此行是关键,range(len(L))得到的使0到x-1的数,len(L)计算该L有几个元素,L[i-1]+L[i],使得相邻的两个元素相加,以次打印出来。
原创粉丝点击