Python矩阵添加对角线元素

Question

我需要制作一个小程序，它将采用 4x4 矩阵并添加列和行的对角线。

1 2 3 4
1 2 3 4
1 2 3 4
1 2 3 4

像这样会加 1+2+3+4。

抱歉含糊不清，这就是我想出的方法，可以通过每列直接向下添加一个矩阵。我还没有使用过 numpy。

matrix = [1,2,3,4],[5,6,7,8],[9,10,11,12] #Assume a list is given
total = 0
    for column in range(0, len(matrix[0])):
        for row in range(0, len(matrix)):
            total += matrix[row][column]
        print("Sum for column " + str(column) + " is " + str(total))
        total = 0   #Reset total to zero before restarting count

我不是 100% 从哪里开始，所以任何想法都会很棒。我的老师告诉我们抵消作为提示。

Answer 1

对角线元素在同一行列坐标:

matrix = [[1,2,3,4],
          [1,2,3,4],
          [1,2,3,4],
          [1,2,3,4]]
answer = sum(matrix[i][i] for i in range(len(matrix)))