面试时,面试官让现现场用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 ('输入矩阵有误!')