20180517函數用於計算矩陣相乘

最近學習Python，於是寫下了如下程序：

import numpy as np

def MatrixMultip(a, b):
    m = a.shape[0]
    n = a.shape[1]
    
    n1 = b.shape[0]
    h = b.shape[1]
    
    if(n!=n1):
        print("The matrix can't multiply!\n")
        return 0
    else:
        c = np.random.rand(m, h)
        for i in range(m):
            for j in range(h):
                c[i][j] = 0
                for k in range(n):
                     c[i][j] += a[i][k]*b[k][j]
                        
        return c

測試用例如下：

m = np.random.rand(4, 5)
print(m)
n = np.random.rand(5, 3)
print(n)
c = MatrixMultip(m, n)
print(c)

運行結果如下：

[[0.05759342 0.91642957 0.76487463 0.98983764 0.3132928 ]
 [0.45148381 0.98424459 0.48097279 0.25893481 0.39565063]
 [0.37260015 0.99184222 0.90811488 0.56404501 0.50999819]
 [0.42517741 0.75780188 0.92259223 0.97579056 0.71583077]]
[[0.1350007  0.81260325 0.57126273]
 [0.13598473 0.70808614 0.46339762]
 [0.56981689 0.82154439 0.56143592]
 [0.11517964 0.89892064 0.75890407]
 [0.52431086 0.15669915 0.62043604]]
[[0.84650603 2.26296833 1.83257031]
 [0.70612723 1.75370762 1.42603047]
 [1.03499997 2.33809048 1.92679552]
 [1.17386643 2.62936837 2.29668665]]

當然還測試了矩陣不匹配的情況，在此就不列出了。

此段代碼主要是用到了循環，而且還考慮了numpy定義數組的方法。