生成器
在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 = 0
for 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=0
a = 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],使得相邻的两个元素相加,以次打印出来。