矩阵相乘的实现-python

• 第一个数组的列和第二个数组的行一样的时候才能相乘
• 相乘原则:第一个数组的行乘以第二个数组的列然后相加放入对应位置.

import numpy as np

def matrix_multi(m1,m2):
	# 首先建立一个值都是0的矩阵,矩阵形状是矩阵1的行数和矩阵2的列数组成
    results = np.zeros((m1.shape[0],m2.shape[1]))
    # 判断矩阵1的列和矩阵2的行数是否相同,如果不相同,则两个矩阵无法相乘,就直接返回
    if m1.shape[1] != m2.shape[0]:
        return
    # 首先遍历矩阵1的行
    for i in range(m1.shape[0]):
    	# 这是遍历矩阵2的列
        for j in range(m2.shape[1]):
            sum = 0
            # 这里遍历矩阵1的列 和遍历矩阵2的行都可以,因为他们是相同的
            for k in range(m2.shape[0]):
            	# 把对应位置相乘并相加后得到的值放入指定位置
                sum += (m1[i][k] * m2[k][j])
                results[i][j] = sum
    return results
    
m1 = np.array([[1,2],[3,4],[4,5]])  #   3x2
m2 = np.array([[2],[5]])            # 2x1

print(matrix_multi(m1,m2))
# 结果是: 3行 1列
#	 [[12.]
	  [26.]
 	  [33.]]