用python实现3*3矩阵对角线元素之和，引申实现N*N阶

1、输入一个3阶矩阵，求对角线元素之和

2、输入：

请输入一个三阶矩阵，3*3格式：

1 2 3

4 5 6

7 8 9

输出：

对角线元素之和是：15

3、代码实现：

list3 = []
print("请输入一个三阶矩阵，3*3格式：")
for i in range(3):
    list3.append(input().split())
print("对角线元素之和是：", end="")
print(int(list3[0][0]) + int(list3[1][1]) + int(list3[2][2]))

4、引申求N*N阶矩阵对角线元素之和：

number = int(input("请输入矩阵的阶数，如3表示三阶："))
listN = []
print("请输入一个N阶矩阵，N*N格式：")
for i in range(number):
    listN.append(input().split())

sum = 0
for i in range(number):
    sum += int(listN[i][i])
print("对角线元素之和是：%d" % sum, end="")

输出：

请输入矩阵的阶数，如3表示三阶：4

请输入一个N阶矩阵，N*N格式：

1 2 3 4

5 6 7 8

9 10 11 12

13 14 15 16

对角线元素之和是：34