PROGRAMMERS Python Lv1. 행렬의 덧셈
이번 문제는 행렬의 덧셈이다.
나의 코드는 다음과 같다.
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차원 배열이 유지된다.
댓글남기기