[编程入门]矩阵对角线求和-题解(Python代码)

初学者，求轻喷

ls = [input().split(), input().split(), input().split()]
sum_1 = 0
sum_2 = 0

for i in range(3):
   for j in range(3):
       if i == j:
           sum_1 = sum_1 + int(ls[i][j])
for i in range(3):
   for j in range(3):
       if i + j == 2:
           sum_2 = sum_2 + int(ls[i][j])

print(sum_1, sum_2)