python求矩阵主对角线元素之和

在Python中，如果你有一个二维列表表示的矩阵（也称为数组），你可以通过遍历矩阵并仅选择主对角线上的元素来计算它们的和。这里是一个简单的例子：

def sum_diagonal(matrix):
    # 确保矩阵是方阵，即行数等于列数
    assert len(matrix) == len(matrix[0]), "Matrix must be square"
    
    # 初始化主对角线元素之和
    diagonal_sum = 0
    
    # 遍历矩阵
    for i in range(len(matrix)):
        diagonal_sum += matrix[i][i]  # 只添加对角线元素
        
    return diagonal_sum

# 示例
matrix = [[1, 2, 3],
          [4, 5, 6],
          [7, 8, 9]]
print(sum_diagonal(matrix))  # 输出：45

这个函数首先检查矩阵是否是方形的，然后通过双层循环（外层遍历行，内层遍历列），将每个对应位置的元素加到总和上，只对主对角线上的元素有效。