Computer >> บทช่วยสอนคอมพิวเตอร์ >  >> การเขียนโปรแกรม >> Python

โปรแกรม Python คูณเมทริกซ์สองตัว


คูณเมทริกซ์สองตัวขนาด n×น ด้วย triple nested loops

อัลกอริทึม

  1. รับเมทริกซ์ A, B
  2. สร้าง result n×n เติม 0
  3. for i, j, k: result[i][j] += A[i][k] * B[k][j]

โค้ดตัวอย่าง

# Input A, B (omitted)
result = [[0]*n for _ in range(n)]
for i in range(n):
    for j in range(n):
        for k in range(n):
            result[i][j] += A[i][k] * B[k][j]

for r in result:
    print(r)

ผลลัพธ์ตัวอย่าง

A = [[2,1,4],[2,1,2],[3,4,3]]
B = [[1,2,3],[4,5,6],[7,8,9]]
Result:
[34, 41, 48]
[20, 25, 30]
[40, 50, 60]