Python对角线和

有时候我们需要计算矩阵中的对角线和，那么在Python中如何实现呢？

下面是一段Python代码，可以计算矩阵的对角线和：

def diagonal_sum(matrix):
    """
    计算矩阵的对角线和
    """
    n = len(matrix)
    diagonal_sum = 0
    for i in range(n):
        diagonal_sum += matrix[i][i]
    return diagonal_sum

以上代码中，我们定义了一个函数diagonal_sum()，这个函数接受一个矩阵作为参数，并返回矩阵的对角线和。

在函数内部，我们首先获取矩阵的大小$n$，然后遍历矩阵的主对角线（即行号和列号相等的那些元素），将它们的值相加，最后返回这个和。

下面是一个例子，演示如何使用diagonal_sum()函数来计算一个3x3的矩阵的对角线和：

matrix = [
    [1, 2, 3],
    [4, 5, 6],
    [7, 8, 9]
]
print(diagonal_sum(matrix)) # Output: 15

以上代码中，我们定义了一个3x3的矩阵，并将其作为参数传递给diagonal_sum()函数。运行后，会输出15，这是这个矩阵的主对角线上的元素之和。