python：实现杨辉三角

给出 n(n≤20)，输出杨辉三角的前 nn行。
什么是杨辉三角，满足下列规律的统称为杨辉三角：
1
1 1
1 2 1
1 3 3 1
1 4 6 4 1
1 5 10 10 5 1
代码：

#5732
import numpy as np
n=int(input())   #输入n
a=np.zeros(shape=[n,n])     #创建0矩阵用于存储杨辉三角计算的数

#print(a)
for i in range(n):
    for j in range(i+1):
        if j==0 or i==j:        #从杨辉三角规律来看，每一行的第一个数与最后一个数都是1，亦即满足当取每一行的第一个元素和行数与该行的列数相等时打印1
            a[i,j]=1
            print(1,end=' ')
        else:
            a[i,j]=a[i-1,j-1]+a[i-1,j]  #否则可以从规律看出下一行的第i，j行的元素等于上一行的第j-1列加第j列的元素之和
            print(int(a[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 

Process finished with exit code 0