#NUMPY067. 矩阵乘积的对角线
矩阵乘积的对角线
描述
给定两个 矩阵,计算它们的乘积矩阵的对角线元素。
格式
输入
第一行是整数 (矩阵大小,)
接下来 行,每 行一个矩阵,每行 个整数。
输出
打印乘积矩阵的对角线元素(数组形式)。
样例
2
1 2
3 4
5 6
7 8
[23 34]
提示
—— 矩阵乘法; —— 获取对角线
给定两个 n×n 矩阵,计算它们的乘积矩阵的对角线元素。
第一行是整数 n(矩阵大小,1≤n≤10)
接下来 2n 行,每 n 行一个矩阵,每行 n 个整数。
打印乘积矩阵的对角线元素(数组形式)。
2
1 2
3 4
5 6
7 8
[23 34]
np.dot —— 矩阵乘法;np.diag —— 获取对角线