행렬의 덧셈 - 2차 배열을 한줄 for문으로 표현하는 법

 

해결일	언어	레벨	테스트명
20240624	Python	level 0	행렬의 덧셈

 

링크 : https://school.programmers.co.kr/learn/courses/30/lessons/12950/solution_groups?language=python3&type=all

 

 

 

 

---

 

내 코드

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

    for i in range(len(arr1)) :
        temp = []
        for j1, j2 in zip(arr1[i],arr2[i]):
            temp.append(j1+j2)
        answer.append(temp)
    return answer

 

 

 

개선점: 지금 위에 쓴 걸 간단히 줄이면 아래 코드가 되는 것 같다...

 

베스트 코드

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