NUMPY086 题解：计算矩阵的迹

题目

计算矩阵的迹（对角线元素之和）。

解题思路

1. 读取矩阵
2. 使用 $\texttt{np.trace}$ 计算迹

代码

import numpy as np

n = int(input())
Z = np.array([list(map(int, input().split())) for _ in range(n)])
result = np.trace(Z)
print(result)

代码详解

迹 = 主对角线元素之和：

$$\text{tr}(A) = \sum_{i=1}^{n} a_{ii}$ ## 运行示例 **输入：** ``` 3 1 2 3 4 5 6 7 8 9 ``` **分析：** - 对角线元素：$1, 5, 9$ - 迹 $= 1 + 5 + 9 = 15$ **输出：** ``` 15 ``` ## 核心知识点 1. $\texttt{np.trace}$ —— 计算矩阵的迹$$