题解 | #矩阵乘法#

矩阵乘法

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

简洁明了的代码。

def fun():
    # 计算i行j列
    def get_i_j_value(i, j):
        ls_a, ls_b = a[i], [b_row[j] for b_row in b]
        return sum([ls_a[ret_i] * ls_b[ret_i] for ret_i in range(len(ls_a))])
	  
    # 取得矩阵1和2的行列
    row1, col1, col2 = int(input().strip()), int(input().strip()), int(input().strip())
    # 推导式得到矩阵1和2
    a = [list(map(int, input().strip().split(" "))) for _ in range(row1)]
    b = [list(map(int, input().strip().split(" "))) for _ in range(col1)]
	
    # 遍历结果的第i,j位,求值
    res = [[get_i_j_value(i, j) for j in range(col2)] for i in range(row1)]
	
    # 打印结果
    for res_row in res:
        print(" ".join(list(map(str, res_row))))


fun()

全部评论

相关推荐

把球:这个听过,你加了就会发现是字节的hr
点赞 评论 收藏
分享
11-06 10:58
已编辑
门头沟学院 嵌入式工程师
双非25想找富婆不想打工:哦,这该死的伦敦腔,我敢打赌,你简直是个天才,如果我有offer的话,我一定用offer狠狠的打在你的脸上
点赞 评论 收藏
分享
不愿透露姓名的神秘牛友
11-27 10:48
点赞 评论 收藏
分享
评论
点赞
收藏
分享
牛客网
牛客企业服务