用python生成器实现杨辉三角

先看杨辉三角的形态:函数

1
        1   1
      1   2   1
    1   3   3   1
  1   4   6   4   1
1   5   10  10  5   1


上学的时候大可能是用c语言的两层for循环在实现,如今咱们尝试用生成器来实现。

先说思路:咱们把每一次层看做一个list, 经过一个for循环,经过迭代,每次生成一个list,而生成器就在每一行生成list中起做用,咱们先定义一个函数,做用是生成每一行的list
1 for x in yhTriangle(10):
2     print(x)

 

对于每一行,list 的第一个元素和最后一个元素是不变的。若是用L = [] 表示的话, L[0], L[n],是 不变的,spa

[l[i] + l[i + 1] for i in range(len(l) - 1)]

因此用这个列表生成式表示 中间的那一部分code

1 def yhTriangle(n):
2     l, index = [1], 0
3     while index < n:
4         yield l
5         l = [1] + [l[i] + l[i + 1] for i in range(len(l) - 1)] + [1]
6         index += 1

完整代码贴出来,你们看看,欢迎提意见啊哈blog

相关文章
相关标签/搜索