定义一个3*3的矩阵，求对角线元素之和

定义一个3*3的矩阵是一个二维数组，其中包含9个元素，行数和列数都是3。如果我们要计算这个矩阵的对角线元素之和，我们通常是指主对角线（从左上角到右下角的元素）和副对角线（从左下角到右上角的元素）的和。

例如，对于一个3x3矩阵：

[ a11, a12, a13 ]
[ a21, a22, a23 ]
[ a31, a32, a33 ]

主对角线元素包括 a11, a22, a33，副对角线元素包括 a13, a21, a32。假设矩阵的每个元素都是整数，我们可以像下面这样计算它们的和：

# 定义矩阵
matrix = [
    [a11, a12, a13],
    [a21, a22, a23],
    [a31, a32, a33]
]

# 计算主对角线和副对角线元素之和
main_diagonal_sum = sum(matrix[i][i] for i in range(3))
secondary_diagonal_sum = sum(matrix[i][3-i] for i in range(3))

# 总和
diagonal_sum = main_diagonal_sum + secondary_diagonal_sum