[프로그래머스/Programmers] 행렬의 덧셈 - Python

Link

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

 

Solution

해당 문제는 필자의 풀이인 zip()를 사용하는 방법 보다는 numpy를 사용하여 해결하는것이 필자는 더 바람직하다고 생각한다.


하지만 zip()도 자주 쓰이므로 이번 기회에 익혀두기 바란다.

 

def solution(arr1, arr2):
    answer = []

    for i, j in zip(arr1, arr2):
        t = []

        for m, n in zip(i, j):
            t.append(m + n)

        answer.append(t)

    return answer

🔑 key point: zip(), numpy

 

Other Solution

import numpy as np

def solution(arr1, arr2):
    arr1 = np.array(arr1)
    arr2 = np.array(arr2)

    answer = arr1 + arr2

    return answer.tolist()