row = [1]  # 第一行

    n = 0  # 行数计数器

    while True:

        yield row

        row = [1] + [row[i] + row[i+1] for i in range(len(row)-1)] + [1]  # 生成下一行

        n += 1

        if n == 10:

            break

#交作业

def triangles():

row = [1] # 第一行

n = 0 # 行数计数器

while True:

yield row

row = [1] + [row[i] + row[i+1] for i in range(len(row)-1)] + [1] # 生成下一行

n += 1

if n == max_rows: # 设置停止条件，max_rows 是输入的行数

raise StopIteration

# 获取输入行数

max_rows = int(input("请输入要生成的杨辉三角行数："))

# 使用生成器生成杨辉三角

generator = triangles()

for row in generator:

print(row)

#可以自己设置行数的杨辉三角