Computer >> 컴퓨터 >  >> 프로그램 작성 >> Python

Python을 사용하여 두 개의 행렬을 추가하는 방법은 무엇입니까?

<시간/>

파이썬에서 2개의 행렬을 추가하는 가장 쉬운 방법은 행렬을 반복하고 요소를 하나씩 추가하는 것입니다. 예를 들어,

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)

이것은 출력을 줄 것입니다:

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

이를 지원하는 numpy 모듈을 사용할 수도 있습니다.

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

이것은 출력을 줄 것입니다:

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