python求矩阵对角线元素和

please help me……真的写不出来,麻烦大家帮我看看如果可以的话教教我思路,真的特别特别感谢大家

img

img

img

def diagsum(n,A):
    '''
    n代表矩阵的行数
    A代表的时一个n*n的整数矩阵
    '''
    if n==1:
        sum=A[0]
    else:
        sum1=0
        sum2=0
        for i in range(n):
            sum1=sum1+A[i][i]
            sum2=sum2+A[i][n-1-i]
        sum=sum1+sum2
        if n%2 != 0:
            sum=sum-A[int((n-1)/2)][int((n-1)/2)]
    return sum

n=3
A=[[1,2,3],[4,5,6],[7,8,9]]
sum=diagsum(n,A)
print(sum)
不知道你这个问题是否已经解决, 如果还没有解决的话:

如果你已经解决了该问题, 非常希望你能够分享一下解决方案, 写成博客, 将相关链接放在评论区, 以帮助更多的人 ^-^