Python3.6.0下实现矩阵乘法
在数学建模的比赛时,有时候不能使用python的第三方库,很多函数只好自己写。
网上有很多矩阵乘法的实现,但没有能在python3.6.0下使用的,所以自己做了点修改:
令
aa = [[1, 3, 5], [2, 4, 6]]
bb = [[1], [2], [1]]
定义函数
def matmul(self, A, B):
return [[sum(a * b for a, b in zip(*a, *b)) for b in zip(transpose(B))] for a in zip(A)]
相比于之前的程序,在最后一个for循环,用*号把a