python矩阵乘法函数实现

#矩阵乘法
import numpy as np
a=np.array([[0,1,2],[3,4,5],[6,7,8],[9,10,11]],float)
b=np.array([[0,1],[2,3],[4,5]],float)
def fun(a,b):
m,n,o=len(a),len(b[0]),len(b)
#len(b[0])：b的列数
#len(b) ：b的行数
c=np.array(range(m*n),float)
c=c.reshape(m,n)
for i in range(m):
for j in range(n):
c[i,j]=0
for k in range(o):
#c[i,j]=a[i,k]*b[k,j]+a[i,1]*b[1,j]+a[i,2]*b[2,j]+a[i,3]*b[3,j]
c[i,j]=a[i,k]*b[k,j]+c[i,j]
return c

c=fun(a,b)
print©

矩阵行数：len(b)
矩阵列数：len(b[0])