python 核心編程 第六章 習題16 矩陣加法和乘法

很簡單，但是寫的有點複雜感覺，有沒有朋友有簡單一些的方法，還請不吝賜教～

#!/usr/bin/python def mar_add(list1,list2):m = len(list1)n = len(list1[0])if m!= len(list2):print "error ! two list must have same dimson"return if n!= len(list2[0]):print "error ! two list must have same dimson"return new_list = []for i in range(m):new_list.append([])for j in range(n):new_list[i].append(list1[i][j] + list2[i][j])return new_listdef mar_multiply(list1,list2):m = len(list1)k = len(list1[0])n = len(list2[0])if k != len(list2):print "error ! dimension error !"return new_list = []for i in range(m):new_list.append([])for j in range(n):new_list[i].append(0)for x in range(k):new_list[i][j] += list1[i][x] * list2[x][j]return new_listif __name__ == "__main__":list1 = [[1,2],[0,1],[2,3],[5,6]]list2 = [[0,1,2],[1,1,0]]print mar_multiply(list1,list2)