首页 > 试题广场 >

计算矩阵的积

[编程题]计算矩阵的积
  • 热度指数:599 时间限制:C/C++ 1秒,其他语言2秒 空间限制:C/C++ 256M,其他语言512M
  • 算法知识视频讲解
将两个矩阵相乘,即 𝐶 = 𝐴 ⋅ 𝐵 

输入描述:
第1行输入一个矩阵A,第2行输入一个矩阵B。


输出描述:
返回𝐴 ⋅ 𝐵 。
如果矩阵形状不一致,则返回 -1。
示例1

输入

[[2,2],[2,4]]
[[2,1],[3,4]]

输出

[[10, 10], [16, 18]]

备注:
1.Python3对应的输入、输出已给出,您只用实现核心功能函数即可。
2.支持numpy、scipy、pandas、scikit-learn库。
def matrixmul(a,b):
    # 补全代码
    import numpy as np
    a = np.array(a)
    b = np.array(b)
    if a.shape[1] != b.shape[0]:
        return -1
    return np.dot(a,b).tolist()

发表于 2025-04-21 22:39:33 回复(0)