python-矩阵相乘

解题思路:

注意事项:

参考代码:

def matrix_mul(A, B):  
      
    return [[sum(a*b for a,b in zip(row,col)) for col in zip(*B)] for row in A]  
      
if __name__ == '__main__':  
    n1,m1 = map(int,input().strip().split())  
    A = [[int(j) for j in input().strip().split()] for i in range(n1)]  
    n2,m2 = map(int,input().strip().split())  
    B = [[int(j) for j in input().strip().split()] for i in range(n2)]  
    C = matrix_mul(A,B)  
  
    for i in range(n1):  
        for j in range(m2):  
            print(C[i][j],end=' ')  
        print()