求一个3×3矩阵对角线元素之和(注意:两条对角线交叉点处的元素只计算一次)Python
时间: 2024-11-05 08:33:59 浏览: 19
Python二维数组实现求出3*3矩阵对角线元素的和示例
在Python中,你可以使用列表推导式以及简单的数学操作来求解3x3矩阵的对角线元素之和。假设我们有一个3x3的二维列表代表矩阵,如下:
```python
matrix = [
[1, 2, 3],
[4, 5, 6],
[7, 8, 9]
]
```
你可以通过下面的代码来计算对角线元素之和:
```python
# 获取主对角线(从左上到右下)
main_diagonal_sum = sum(matrix[i][i] for i in range(len(matrix)))
# 如果有次对角线(从右上到左下),需要额外处理
if len(matrix) > 1: # 防止索引越界
secondary_diagonal_sum = sum(matrix[j][len(matrix)-1-j] for j in range(len(matrix)))
diagonal_sum = main_diagonal_sum + secondary_diagonal_sum
else:
secondary_diagonal_sum = matrix[0][0] # 对于1x1矩阵,主、次对角线相同
diagonal_sum = main_diagonal_sum
diagonal_sum
```
运行上述代码,你会得到矩阵的对角线元素之和。如果有多组数据,可以将`matrix`替换为你实际的数据。
阅读全文