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

내 코드

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

    for i in range(len(arr1)):
        l=[]
        for k in range(len(arr1[i])):
            l.append(arr1[i][k]+arr2[i][k])
        answer.append(l) 
        
    return answer

arr1과 arr2의 길이가 같다는 점을 사용하여 풀었다. 

 

다른사람 코드

def sumMatrix(A,B):
    answer = [[c + d for c, d in zip(a, b)] for a, b in zip(A,B)]
    return answer

아.. zip.,,저번에 정리했었는데 까먹고있었다... 

먼저 zip으로 a,b를 가져와 list a와 b에서 또 zip을 사용하여 원소를 추출해 더한다.