如何编写一个Python函数，用于计算任意大小矩阵主对角线和副对角线元素的总和？

当你需要计算一个任意大小矩阵的主对角线和副对角线元素之和时，编写一个函数可以让你轻松处理不同大小的矩阵。这里提供的函数将帮助你完成这个任务，并且你将能够通过提供的示例来验证你的代码。首先，我们来看一下如何定义这样一个函数。

参考资源链接：[Python二维数组实现求出3*3矩阵对角线元素的和示例](https://wenku.csdn.net/doc/6412b57abe7fbd1778d434b3?spm=1055.2569.3001.10343)

函数定义可以使用嵌套循环来遍历矩阵的行和列，同时通过行号和列号的比较来确定当前元素是否位于对角线上。以下是一个具体实现的例子：

def sum_diagonals(matrix):
    # 获取矩阵的大小
    n = len(matrix)
    
    # 计算主对角线的元素和
    primary_sum = sum(matrix[i][i] for i in range(n))
    
    # 计算副对角线的元素和，从右上到左下的副对角线
    secondary_sum = sum(matrix[i][n-i-1] for i in range(n))
    
    return primary_sum + secondary_sum

# 示例使用3x3矩阵
matrix_3x3 = [
    [1, 2, 3],
    [4, 5, 6],
    [7, 8, 9]
]

# 示例使用4x4矩阵
matrix_4x4 = [
    [1, 2, 3, 4],
    [5, 6, 7, 8],
    [9, 10, 11, 12],
    [13, 14, 15, 16]
]

print(

参考资源链接：[Python二维数组实现求出3*3矩阵对角线元素的和示例](https://wenku.csdn.net/doc/6412b57abe7fbd1778d434b3?spm=1055.2569.3001.10343)