python生成器实现杨辉三角

时间:2023-03-08 21:45:08
 def triangels():
"""
杨辉三角
"""
lst = [1]
n_count = 2 # 下一行列表长度
while True:
yield lst
lst_n = list(range(0 ,n_count))
lst = [1] + [lst[i-1]+lst[i] for i,v in enumerate(lst_n) if i!=0 and i!=n_count-1] + [1]
n_count += 1

调用:

  n = 0
while n < 10:
next(x)
n += 1

实现效果:

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