Python：矩阵相乘

题目描述

小张最近刚刚学习了矩阵乘法，但是他计算的速度太慢，于是他希望你能帮他写一个矩阵乘法的运算器。

输入描述

输入的第一行包含三个正整数 N,M,K，表示一个 NM的矩阵乘以一个MK的矩阵。接下来N行，每行M个整数，表示第一个矩阵。再接下来的M行，每行K个整数，表示第二个矩阵。

0 < N,M,K ≤ 100, 0 ≤  矩阵中的每个数 ≤ 1000。

输出描述

输出有 N 行，每行 K 个整数，表示矩阵乘法的结果。

输入输出样例

输入

2 1 3
1
2
1 2 3

输出

1 2 3
2 4 6

 参考代码：

n,m,k=map(int,input().split())
a=[]     #存放n*m矩阵
b=[]     #存放m*k矩阵
c=[[0]*k for i in range(n)]
for i in range(n):
  a.append(list(map(int,input().split())))
for j in range(m):
  b.append(list(map(int,input().split())))
for i in range(n):
  for j in range(m):
    for l in range(k):
      c[i][l]+=a[i][j]*b[j][l]  #矩阵乘法c=a*b
for i in range(n):
  for j in range(k):   
    print(c[i][j],end=" ")
  print()  #换行