python给定一个n×n的方阵,本题要求计算该矩阵主、副对角线上的所有元素之和。主对角线为从矩阵的左上角至右下角的连线,副对角线为从矩阵的右上角至左下角的连线。。 输入格式: 输入第一行给出正整数n(1<n≤10)
时间: 2023-11-26 17:02:39 浏览: 199
python 实现矩阵对角线之和
5星 · 资源好评率100%
好的,您的问题是如何用 Python 计算一个矩阵的主、副对角线上的元素之和,是吗?
可以使用 NumPy 库中的 `trace` 函数计算主对角线上的元素之和,使用 `trace` 函数的可选参数 `offset` 来计算副对角线上的元素之和。具体代码如下:
```python
import numpy as np
# 读取矩阵大小
n = int(input())
# 读取矩阵
matrix = []
for i in range(n):
row = list(map(int, input().split()))
matrix.append(row)
# 将矩阵转换为 NumPy 数组
matrix = np.array(matrix)
# 计算主对角线和副对角线上的元素之和
main_sum = np.trace(matrix)
anti_sum = np.trace(matrix[::-1], offset=n-1)
# 输出结果
print(main_sum, anti_sum)
```
这个程序会先读取输入中的矩阵大小和矩阵元素,然后使用 `numpy.array` 函数将矩阵转换为 NumPy 数组。接着,使用 `numpy.trace` 函数计算主对角线上的元素之和,并使用 `numpy.trace` 函数和可选参数 `offset` 计算副对角线上的元素之和。最后,输出结果。
希望这个代码对您有帮助!
阅读全文