首页 > 编程 > Python > 正文

python实现矩阵乘法的方法

2020-01-04 18:06:29
字体:
来源:转载
供稿:网友

这篇文章主要介绍了python实现矩阵乘法的方法,实例分析了Python矩阵操作的相关技巧,需要的朋友可以参考下

本文实例讲述了python实现矩阵乘法的方法。分享给大家供大家参考。具体实现方法如下:

 

 
  1. def matrixMul(A, B): 
  2. res = [[0] * len(B[0]) for i in range(len(A))] 
  3. for i in range(len(A)): 
  4. for j in range(len(B[0])): 
  5. for k in range(len(B)): 
  6. res[i][j] += A[i][k] * B[k][j] 
  7. return res 
  8. def matrixMul2(A, B): 
  9. return [[sum(a * b for a, b in zip(a, b)) for b in zip(*B)] for a in A] 
  10. a = [[1,2], [3,4], [5,6], [7,8]] 
  11. b = [[1,2,3,4], [5,6,7,8]] 
  12. print matrixMul(a,b) 
  13. print matrixMul(b,a) 
  14. print "-"*90 
  15. print matrixMul2(a,b) 
  16. print matrixMul2(b,a) 
  17. print "-"*90 
  18. from numpy import dot 
  19. print map(list,dot(a,b)) 
  20. print map(list,dot(b,a)) 
  21.  
  22. #Out: 
  23. #[[11, 14, 17, 20], [23, 30, 37, 44], [35, 46, 57, 68], [47, 62, 77, 92]] 
  24. #[[50, 60], [114, 140]] 
  25. #------------------------------------------------------------------------ 
  26. #[[11, 14, 17, 20], [23, 30, 37, 44], [35, 46, 57, 68], [47, 62, 77, 92]] 
  27. #[[50, 60], [114, 140]] 
  28. #------------------------------------------------------------------------ 
  29. #[[11, 14, 17, 20], [23, 30, 37, 44], [35, 46, 57, 68], [47, 62, 77, 92]] 
  30. #[[50, 60], [114, 140]] 

希望本文所述对大家的Python程序设计有所帮助。

发表评论 共有条评论
用户名: 密码:
验证码: 匿名发表