[Python] 연습문제 : 행렬의 덧셈

코딩테스트 연습 - 행렬의 덧셈

 

💡문제💡

---

행과 열의 크기가 같은 두 행렬의 같은 행, 같은 열의 값을 서로 더한 결과를 return 하시오.

 

[제한 조건]

• 행렬 arr1, arr2의 행과 열의 길이는 500을 넘지 않습니다.

 

코드 및 풀이👩🏻‍💻

---

[SOLUTION METHOD]

 

1. arr1, arr2 각각을 array로 만듦

2. 같은 행, 같은 열끼리 더한 후 answer에 저장 (이때, int형으로 변환 후 계산해야 오류가 나지 않음)

3. result 리스트에 저장 후 return

 

 

[CODE]

'''

import numpy as np

def solution(arr1, arr2):
    result = []
    A = np.array(arr1)
    B = np.array(arr2)

    for i in range(len(arr1)):
        answer = []
        for j in range(len(A[0])):
            answer.append(int(A[i][j]+B[i][j]))
        result.append(answer)

    return result

'''