Python求3*3矩阵对角线元素之和

3*3矩阵对角线元素之和

1   2   3
4   5   6
7   8   9

array = [[1.0, 2.0, 3.0], [4.0, 5.0, 6.0], [7.0, 8.0, 9.0]]

a = []
sum = 0.0
for i in range(3):
    a.append([])
    for j in range(3):
        a[i].append(float(input('请输入第%s行第%s列的元素：' % (i + 1, j + 1))))
    print('数组：',a)
for i in range(3):
    sum += a[i][i]
print('数组之和：',sum)

打印结果：
请输入第1行第1列的元素：1
请输入第1行第2列的元素：2
请输入第1行第3列的元素：3
数组： [[1.0, 2.0, 3.0]]

请输入第2行第1列的元素：4
请输入第2行第2列的元素：5
请输入第2行第3列的元素：6
数组： [[1.0, 2.0, 3.0], [4.0, 5.0, 6.0]]

请输入第3行第1列的元素：7
请输入第3行第2列的元素：8
请输入第3行第3列的元素：9
数组： [[1.0, 2.0, 3.0], [4.0, 5.0, 6.0], [7.0, 8.0, 9.0]]

数组之和： 15.0