Computer >> Máy Tính >  >> Lập trình >> Python

Làm thế nào để thêm hai ma trận bằng Python?

Cách dễ nhất để thêm 2 ma trận trong python là lặp qua chúng và thêm từng phần tử một. Ví dụ,

X = [[1,2,3], [4,5,6], [7,8,9]]
Y = [[9,8,7], [6,5,4], [3,2,1]]

result = [[0,0,0], [0,0,0], [0,0,0]]

for i in range(len(X)):
   for j in range(len(X[0])):
      result[i][j] = X[i][j] + Y[i][j]

for r in result:
   print(r)

Điều này sẽ cho kết quả:

[10, 10, 10]
[10, 10, 10]
[10, 10, 10]

Bạn cũng có thể sử dụng mô-đun numpy, có hỗ trợ cho việc này.

import numpy as np
X = np.matrix([[1, 2], [3, 4]])
Y = np.matrix([[2, 2], [2, 2]])
result = X + Y
print(result)

Điều này sẽ cho kết quả:

matrix([[3, 4], [5, 6]])