행렬의 덧셈(x)

Kylie·2022년 11월 14일
0

프로그래머스 Lv.1

목록 보기
39/69

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

arr1 = [[1,2],[2,3]]
arr2 = [[3,4],[5,6]]
return = [[4,6],[7,9]]

arr1 = [[1],[2]]
arr2 = [[3],[4]]
return = [[4],[6]]

내 코드

def solution(arr1, arr2):
    answer = [[c+d for c, d in zip(a, b)] for a, b in zip(arr1, arr2)]
    return answer

다른 풀이

def sumMatrix(A,B):
    return [list(map(sum, zip(*x))) for x in zip(A, B)]
profile
딥린이

0개의 댓글