求n乘n矩阵对角线之和

Question

求n乘n矩阵对角线之和怎么写，不会哇。求n乘n矩阵对角线之和怎么写，不会哇求n乘n矩阵对角线之和怎么写，不会哇

Answer 1

对角线就是行号和列号相同的单元，以及行号和列号相加等于n的单元

n=int(input())
arr=[[0 for x in range(0,n)] for y in range(0,n)]
sum = 0
for i in range(n):
    for j in range(n):
        arr[i][j] = int(input())
        if i==j or i+j==n-1:
           sum += arr[i][j]
print(arr)
print(sum)