NUMPY067 题解：矩阵乘积的对角线

题目

计算两个矩阵乘积的对角线元素。

解题思路

1. 读取两个矩阵
2. 计算矩阵乘积 $\texttt{np.dot(A, B)}$
3. 获取对角线 $\texttt{np.diag}$

代码

import numpy as np

n = int(input())
A = np.array([list(map(int, input().split())) for _ in range(n)])
B = np.array([list(map(int, input().split())) for _ in range(n)])
result = np.diag(np.dot(A, B))
print(result)

代码详解

• $\texttt{np.dot(A, B)}$ —— 计算矩阵乘法 $A \times B$
• $\texttt{np.diag}$ —— 获取矩阵的对角线元素

运行示例

输入：

2
1 2
3 4
5 6
7 8

分析：

• $A = \begin{bmatrix} 1 & 2 \\ 3 & 4 \end{bmatrix}$
• $B = \begin{bmatrix} 5 & 6 \\ 7 & 8 \end{bmatrix}$
• $A \times B = \begin{bmatrix} 19 & 22 \\ 43 & 50 \end{bmatrix}$
• 对角线元素：$[19, 50]$

输出：

[23 34]

核心知识点

1. $\texttt{np.dot}$ —— 矩阵乘法
2. $\texttt{np.diag}$ —— 获取对角线元素

NUMPY067 题解：矩阵乘积的对角线

题目

计算两个矩阵乘积的对角线元素。

解题思路

1. 读取两个矩阵
2. 计算矩阵乘积 $\texttt{np.dot(A, B)}$
3. 获取对角线 $\texttt{np.diag}$

代码

import numpy as np

n = int(input())
A = np.array([list(map(int, input().split())) for _ in range(n)])
B = np.array([list(map(int, input().split())) for _ in range(n)])
result = np.diag(np.dot(A, B))
print(result)

代码详解

• $\texttt{np.dot(A, B)}$ —— 计算矩阵乘法 $A \times B$
• $\texttt{np.diag}$ —— 获取矩阵的对角线元素