Posts 행렬의 덧셈 [Python]
Post
Cancel

행렬의 덧셈 [Python]

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
  • 위의 코드를 함수로 작성
This post is licensed under CC BY 4.0 by the author.