用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

for x in yhTriangle(10):
    print(x)

 

對於每一行，list 的第一個元素和最后一個元素是不變的。如果用L = [] 表示的話， L[0], L[n],是 不變的，

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

所以用這個列表生成式表示 中間的那一部分

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

完整代碼貼出來，大家看看，歡迎提意見啊哈