代码:
#! /usr/bin/env python
#coding=utf-8
import numpy as np
def MatrixMultiply(a, b):
n11,n12=np.shape(a)
n21,n22=np.shape(b)
n=max(n11,n22)
c=np.zeros([n11,n22],dtype=int)
for i in range(0, n11):
for j in range(0, n22):
#c[i][j]=0
for k in range(0, n21):
c[i][j]=c[i][j]+a[i][k]*b[k][j]
return c
if __name__ == '__main__':
a=[[1,1],[2,3]]
b=[[1,0,0,1],[2,1,2,4]]
c=MatrixMultiply(a, b)
print(c)
注意:
- 矩阵相乘的维度要求