1. 행렬의 덧셈
URL : https://programmers.co.kr/learn/courses/30/lessons/12950
1.1 문제 설명
- 행렬의 덧셈은 행과 열의 크기가 같은 두 행렬의 같은 행, 같은 열의 값을 서로 더한 결과가 됩니다. 2개의 행렬 arr1과 arr2를 입력받아, 행렬 덧셈의 결과를 반환하는 함수, solution을 완성해주세요.
1.2 제한 조건
- 행렬 arr1, arr2의 행과 열의 길이는 500을 넘지 않습니다.
1.3 Solution
1
2
arr1 = [[1,2],[2,3]]
arr2 = [[3,4],[5,6]]
1
2
3
4
5
6
import operator
answer = []
for i in range(0,len(arr1)):
answer.append(list(map(operator.add, arr1[i], arr2[i])))
answer
1
[[4, 6], [7, 9]]
- operator와 map을 사용하여 행렬의 덧샘을 만들었음
1
2
3
4
5
6
import operator
def solution(arr1, arr2):
answer = []
for i in range(0,len(arr1)):
answer.append(list(map(operator.add, arr1[i], arr2[i])))
return answer
- 위의 코드를 함수로 작성