def s(a,sum=0):
"""返回矩阵a的对角线元数之和"""
for i in range(len(a)):
sum+=a[i][i]
return sum
def insert_list(a,n):
"""给空列表a插入n阶数组"""
for i in range(n):
a.append([])
for j in range(n):
a[i].append(int(input("请输入%d行%d列的元素"%(i+1,j+1))))
a = []
n = int(input("您想添加一个几阶数组:"))
insert_list(a,n)
print('添加的数组为:',a)
print('数组对角线之和:',s(a))