python3两个三阶矩阵相乘_矩阵相乘的实现-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.]]