python的矩阵加法和乘法

本来以为python的矩阵用list表示出来应该很简单可以搞。。其实发现有大学问。

这里贴出我写的特别不pythonic的矩阵加法，作为反例。

def add(a, b):        
     rows = len(a[0])  
     cols = len(a)     
     c = []            
     for i in range(rows):
         temp = []
         for j in range(cols):
             temp.append(a[i][j] + b[i][j])
         c.append(temp)
     return c

然后搜索了一下资料，果断有个很棒的，不过不知道有没有更棒的。

矩阵加法
def madd(M1, M2):
    if isinstance(M1, (tuple, list)) and isinstance(M2, (tuple, list)):
        return [[m+n for m,n in zip(i,j)] for i, j in zip(M1,M2)]

矩阵乘法
def multi(M1, M2):
    if isinstance(M1, (float, int)) and isinstance(M2, (tuple, list)):
        return [[M1*i for i in j] for j in M2] 
    if isinstance(M1, (tuple, list)) and isinstance(M2, (tuple, list)):
        return [[sum(map(lambda x: x[0]*x[1], zip(i,j)))
                 for j in zip(*M2)] for i in M1] 

代码片段来源于：http://www.hustyx.com/python/