python实现矩阵相乘

面试时，面试官让现现场用python写两个矩阵相乘的程序，不可以直接调api。十分钟以后程序写了出来，但是自己看着都别扭，首先代码冗余，这很不python，其次用的时间有点长啊，这种程序花了十分钟，估计面试官也挺无语。回来赶紧把面试时的代码精简了一下，留着备用或者给网上其他小伙伴用。

def matrixMul(A, B):
    if len(A[0]) == len(B):
        res = [[0] * len(B[0]) for i in range(len(A))]
        for i in range(len(A)):
            for j in range(len(B[0])):
                for k in range(len(B)):
                    res[i][j] += A[i][k] * B[k][j]
        return res
    return ('输入矩阵有误！')