python實現矩陣乘法
矩陣相乘需要前面矩陣的行數與後面矩陣的列數相同方可相乘。第一步,先將前面矩陣的每一行分別與後面矩陣的列相乘,作為結果矩陣的行列;第二步算出結果即可。
# 2 3 3 4 # 1 2 3 # 4 5 6 # 1 2 3 4 # 5 6 7 8 # 9 10 11 12 lst1, lst2 = [], [] n1,m1,n2,m2 = map(int,input().split()) for i in range(n1): nums = list(map(int,input().split())) #輸入一行數據 lst1.append(nums) for i in range(n2): nums = list(map(int,input().split())) lst2.append(nums) res = [] for i in range(n1): res.append([]) for j in range(m2): lst4 = [] lst3 = lst1[i] for k in range(n2): lst4.append(lst2[k][j]) res_num = sum(map(lambda x,y:x*y,lst3,lst4)) res[i].append(res_num) print(res) import numpy as np print('numpy:',np.dot(lst1,lst2)) #使用numpy驗證結果
2 3 3 4 1 2 3 4 5 6 1 2 3 4 5 6 7 8 9 10 11 12 [[38, 44, 50, 56], [83, 98, 113, 128]] numpy: [[ 38 44 50 56] [ 83 98 113 128]]
到此這篇關於python實現矩陣乘法的文章就介紹到這瞭,更多相關python矩陣乘法內容請搜索WalkonNet以前的文章或繼續瀏覽下面的相關文章希望大傢以後多多支持WalkonNet!
推薦閱讀:
- Python 如何解決稀疏矩陣運算
- Python中X[:,0]和X[:,1]的用法
- Python NumPy教程之索引詳解
- 實例詳解Python中的numpy.abs和abs函數
- python中數組array和列表list的基本用法及區別解析