PROGRAMMERS Python Lv1. 행렬의 덧셈

최대 1 분 소요

이번 문제는 행렬의 덧셈이다.

나의 코드는 다음과 같다.

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

이번 문제는 처음에 쉽게 잘 풀지 못해 다른 사람의 코드를 공부하려고 한다.

우선 이 코드는 for문과 zip의 활용이 핵심이다.

우선 arr1, arr2에서 각각의 원소를 묶고, 묶은 원소들 안의 값들을

또 다시 묶어주면서 덧셈을 할 수 있다. 또한 리스트를 [[]] 이중으로 표현해줘야

2차원 배열이 유지된다.

댓글남기기