Level 1. 행렬의 덧셈

46. 행렬의 덧셈

코딩테스트 연습 > 연습문제 > 행렬의 덧셈
https://programmers.co.kr/learn/courses/30/lessons/12950

---

문제 설명

Input value =

• arr1,arr2 = 행과 열의 크기가 같은 두 리스트

Output value =

• return 행렬 합

---

제한 사항

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

---

문제 구상

#00
arr1 = [[1,2],[2,3]]
arr2 = [[3,4],[5,6]]

#01-1
for i in range(len(arr1)):
    for j in range(len(arr1[0])):
        arr1[i][j]+=arr2[i][j]
#01-2
[[a+b for a,b in zip(a1,a2)] for a1,a2 in zip(arr1,arr2)]

---

문제 풀이

이중 for문

def solution(arr1,arr2):
    return [[a+b for a,b in zip(a1,a2)] for a1,a2 in zip(arr1,arr2)]

# Code test
arr1,arr2 = [[1,2],[2,3]],[[3,4],[5,6]]
solution(arr1,arr2)

한줄 프로그래밍

def solution(arr1,arr2):
    return [[a+b for a,b in zip(a1,a2)] for a1,a2 in zip(arr1,arr2)]

# Code test
arr1,arr2 = [[1,2],[2,3]],[[3,4],[5,6]]
solution(arr1,arr2)