如何用python输出下列打印方阵？

Question

输入n(0

Answer 1

importnumpyasnp
defnum(i=[0]):
i[0]+=1
returni[0]
shu_ru=int(input("shuru"))
a=np.zeros((shu_ru,shu_ru))
k=1
all=[]
whileklst=[]
foriinrange(k):
forjinrange(k):
ifi+j==k-1andilst.append((i,j))
ifk%2==0:
lst.reverse()
k+=1
all.append(lst_p)
foriinall:
forjini:
a[j]=num()
当a=3，输出
Out[1]:
array([[1.,3.,4.],
[2.,5.,8.],
[6.,7.,9.]])
当a=4，输出
Out[2]:
array([[1.,3.,4.,10.],
[2.,5.,9.,11.],
[6.,8.,12.,15.],
[7.,13.,14.,16.]])
当a=1，输出
Out[2]:array([[1.]])
代码写的比较直白，没有优化，大概就是从下标跟你的输入值的关系着手分析。