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

[문제]

[해결]

def solution(arr1, arr2):
    result = []
    for x, y in zip(arr1, arr2):
        sum = []
        for a, b in zip(x,y):
            sum.append(a+b)
        result.append(sum)
    return result

[정리 및 새롭게 알게 된 점]

먼저 zip함수를 통해 모든 행과 열이 들어있는 리스트를 반복문을 돌리고 각 행열 원소를 추출하였다. 그런 다음, 리스트를 담는 새로운 변수 sum을 만들고, 다시한번 zip함수를 사용하여 각 원소의 행과 열을 추출하는 반복문을 돌렸다. 그 후, 추출된 행과 열의 원소를 각각 더하고 sum변수에 append 시켜 새로운 행열원소 합계를 만든다음, 다시한번 최종 result 변수에 sum변수를 넣어주어 문제를 해결하였다.