Python矩阵添加对角元素

我需要做一个小程序，将采取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%从这里开始，所以任何想法都会很好。我的老师告诉我们偏移量作为提示。

对角线元素位于相同的行-列坐标：

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)))

简单使用numpy

>>> import numpy as np
>>> matrix = [[1,2,3,4],
              [1,2,3,4],
              [1,2,3,4],
              [1,2,3,4]]
>>> sum(np.diag(matrix))
10

---

如果你能告诉我们你已经得到了什么以及你的困境，这会有所帮助。事实上，这是一个相当直截了当的问题，让这个问题看起来像是“你能帮我做作业吗？”之类的问题。一般来说，这里的人更容易接受我们认为是在帮助你学习的问题，而不仅仅是给你答案。如果这里的任何答案能够解决你的问题，请确保通过单击左侧的绿色复选标记“接受”最好的答案。