프로그래머스) 행렬의 덧셈 (파이썬)

문제 설명

 

행렬의 덧셈은 행과 열의 크기가 같은 두 행렬의 같은 행, 같은 열의 값을 서로 더한 결과가 됩니다. 2개의 행렬 arr1과 arr2를 입력받아, 행렬 덧셈의 결과를 반환하는 함수, solution을 완성해주세요.

 

제한 사항

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

 

입출력 예

[[1,2],[2,3]]	[[3,4],[5,6]]	[[4,6],[7,9]]
[[1],[2]]	[[3],[4]]	[[4],[6]]

 

나의 풀이

 

def solution(arr1,arr2):
    x = []
    for i,k in zip(arr1,arr2):
        w = []
        for a,b in zip(i,k):
            w.append(a+b)
        x.append(w)
    return x

 

이번 문제는 처음 for문을 에서 zip함수를 사용하여 i,k에 담아내고, 두번째 for문에서는 i,k를 a,b로 담아내었다.

 

그리고 각각의 변수에 담아낸 후 리턴하였다.