행렬의 덧셈_프로그래머스_파이썬

• 사전지식
  • 2차원 리스트
  • tolist
• 2차원 리스트

arr = [[1,2],[2,4]]

arr[1]

[2, 4]

arr1[1][0]

2

• tolist : 행렬은 리스트로 변환시켜준다.

import numpy as np
a = [[1,2], [2,3]]
b = [[3,4],[5,6]]

a = np.array(a)
b = np.array(b)
c= a+b

c

array([[4, 6],
       [7, 9]])

c.tolist()

[[4, 6], [7, 9]]

---

• 나의 풀이

def solution(arr1, arr2):
    arr3 = [[0]*len(arr1[0]) for _ in range(len(arr1))]
    for i in range(len(arr1)):
        for j in range(len(arr1[0])):
            arr3[i][j] = arr1[i][j] + arr2[i][j]
    return arr3

• 다른 사람 풀이

import numpy as np
def sumMatrix(A,B):
    A=np.array(A)
    B=np.array(B)
    answer=A+B
    return answer.tolist()

# 아래는 테스트로 출력해 보기 위한 코드입니다.
print(sumMatrix([[1,2], [2,3]], [[3,4],[5,6]]))

[[4, 6], [7, 9]]