Python矩阵输入和对角线求和

题目描述
求一个3×3矩阵对角线元素之和。
输入 矩阵
输出 主对角线 副对角线 元素和
样例输入
1 2 3
1 1 1
3 2 1

样例输出 3 7

arr=[]
for i in range(1,4):
	arr.append(list(map(int,input().split())))
print(arr[0][0]+arr[1][1]+arr[2][2],arr[0][2]+arr[1][1]+arr[2][0])

这里我们这样进行一个矩阵的输入，好处是我们可以想C语言中的二维数组一样，通过脚标顺利的制定到我们想要找到的元素