矩阵相乘: 输入两个整数类型的矩阵mat1(m行d列)和mat2(d行n列),返回矩阵相乘后的结果mat1*mat2(m行n列)。矩阵均用二维列表进行表示。
def func3(m1, m2):
m = len(m1)
n = len(m2[0])
m3 = [[0 for i in range(m)] for j in range(n)]
for i in range(m):
for j in range(n):
for k in range(len(m2)):
m3[i][j] += m1[i][k] * m2[k][j]
return m3
if __name__ == '__main__':
print(func3([[1, 2]], [[1], [2]]))
print(func3([[1, 2], [1, 3]], [[1, 1], [1, 0]]))
第三方库: numpy