python 求一个3*3矩阵的两条对角线元素之和。
时间: 2024-10-12 19:12:14 浏览: 57
Python二维数组实现求出3*3矩阵对角线元素的和示例
5星 · 资源好评率100%
在Python中,你可以通过创建一个3x3的二维列表(矩阵)来表示这个矩阵,然后计算主对角线(从左上到右下)和副对角线(从左下到右上)的元素之和。下面是一个简单的示例:
```python
def sum_diagonals(matrix):
# 主对角线元素
main_diagonal_sum = sum(matrix[i][i] for i in range(len(matrix)))
# 副对角线元素(如果矩阵是方阵)
if len(matrix) == 3: # 仅当矩阵是3x3时才有副对角线
secondary_diagonal_sum = sum(matrix[i][len(matrix)-1-i] for i in range(len(matrix)))
else:
secondary_diagonal_sum = None
return main_diagonal_sum, secondary_diagonal_sum
# 创建一个3x3矩阵
matrix = [[1, 2, 3],
[4, 5, 6],
[7, 8, 9]]
main_sum, secondary_sum = sum_diagonals(matrix)
print(f"主对角线元素之和: {main_sum}")
if secondary_sum is not None:
print(f"副对角线元素之和: {secondary_sum}")
```
在这个例子中,`sum_diagonals`函数接受一个3x3的列表作为输入,计算并返回两个对角线元素之和。
阅读全文