按照廖老师提供的思路写的

def triangles():    
    r =[1]
    while True:
        yield r
        r1=r+[0]
        r2=[0]+r
        r=[r1[i]+r2[i] for i in range(min(len(r1),len(r2)))]

n=0
h=int(input('请输入杨辉三角的行数：'))
for i in triangles():
    print(i)
    n=n+1
    if n==h:
        break