题解 | 矩阵和向量的点积

矩阵和向量的点积

https://www.nowcoder.com/practice/a7c9e4eb3b3745cf8d81d6159051937d

def matrix_vector_dot_product(matrix, vector):
    if not matrix or not vector:
        return -1

    if isinstance(vector[0],list):
        return -1
    cols=len(matrix[0])
    if cols!=len(vector):
        return -1
    result=[]
    for row in matrix:
        product=sum(row_i*col_i for row_i,col_i in zip(row,vector))
        result.append(product)
    return result

    

# 主程序
if __name__ == "__main__":
    # 输入矩阵和向量
    matrix_input = input()
    vector_input = input()

    # 处理输入
    import ast
    matrix = ast.literal_eval(matrix_input)
    vector = ast.literal_eval(vector_input)

    # 调用函数计算点积
    output = matrix_vector_dot_product(matrix, vector)
    
    # 输出结果
    print(output)

全部评论

相关推荐

07-11 15:12
门头沟学院 Java
别人在上班,我就在工位上看看视频啥的,这正常吗?
程序员小白条:实习就是摸鱼,只是公司指标,把你进来了,可能那时候客户很多,但等你进来的时候,已经是淡季了,根本没多少需求,或者说根本不适合实习生去完成,因此你就每天干坐着就行,可能1,2个月都没需求
实习生的蛐蛐区
点赞 评论 收藏
分享
不愿透露姓名的神秘牛友
07-11 11:25
点赞 评论 收藏
分享
07-07 12:25
门头沟学院 Java
程序员牛肉:你这个智邮公司做的就是那个乐山市税务系统的服务吗?
点赞 评论 收藏
分享
评论
点赞
收藏
分享

创作者周榜

更多
牛客网
牛客网在线编程
牛客网题解
牛客企业服务