求n乘n矩阵对角线之和

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

对角线就是行号和列号相同的单元，以及行号和列号相加等于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)

arr = [[1,2,3],[4,5,6],[7,8,9]]
sum = 0
for i in range(3):
    for j in range(3):
        if (i == j): sum += arr[i][j]
print(sum)

您好，我是有问必答小助手，您的问题已经有小伙伴帮您解答，感谢您对有问必答的支持与关注！
PS：问答VIP年卡 【限时加赠：IT技术图书免费领】，了解详情>>> https://vip.csdn.net/askvip?utm_source=1146287632