帮我编写一段程序实现杨辉三角吧

好，以下是Python编写的杨辉三角程序：

def yanghui_triangle(n):
    # 初始化杨辉三角矩阵
    tri = [[1] * (i + 1) for i in range(n)]
    # 逐行计算杨辉三角
    for i in range(2, n):
        for j in range(1, i):
            tri[i][j] = tri[i-1][j-1] + tri[i-1][j]
    return tri

# 测试
n = 6
tri = yanghui_triangle(n)
for i in range(n):
    print(' ' * (n-i), end='')
    for j in range(i+1):
        print('{:^4d}'.format(tri[i][j]), end=' ')
    print()

运行结果如下：

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

在这个程序中，我们使用一个二维数组 `tri` 存储杨辉三角的每个元素。首先初始化 `tri` 为一个 `n` 行的矩阵，每行都是从左到右依次为 1，即 `[[1], [1, 1], ... , [1, 1, ..., 1]]`。接着，我们遍历每一行 `i`，并且遍历该行的每个元素 `j`，根据杨辉三角的递推公式 `tri[i][j] = tri[i-1][j-1] + tri[i-1][j]` 计算当前行的元素。最后，我们按照规定的格式输出杨辉三角即可。