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

임정민·2022년 11월 26일
2

알고리즘 문제풀이

목록 보기
2/173
post-thumbnail

코딩테스트 연습 스터디 진행중 입니다. ✍✍✍
Notion : https://www.notion.so/1c911ca6572e4513bd8ed091aa508d67

문제

https://school.programmers.co.kr/learn/courses/30/lessons/12950

풀이

[나의 풀이]

import numpy as np # 행렬문제 -> 강의에서 배운 numpy 행렬 활용

def solution(arr1, arr2):
    
    arr1 = np.array(arr1) # list형의 arr를 np.array형으로 변환
    arr2 = np.array(arr2)
    
    answer = arr1 + arr2
    
#     print(f"answer의 type: {type(answer)}")
#     print(answer)
#     print()
    
#     print(f"answer.tolist()의 type : {type(answer.tolist())}")
#     print(answer.tolist())
#     print()
    
#     print(f"type(answer) == type(answer.tolist()): {type(answer) == type(answer.tolist())}") 
    # list형과 np.array형은 모양은 비슷하지만 다른 클래스 타입
    
    return answer.tolist()

[팀원의 풀이]

def solution(arr1,arr2):
    answer = []
    for i in range(0,len(arr1)):
        ans = []
        for j in range(len(arr1[i])):
            ans.append(arr1[i][j] + arr2[i][j])
        answer.append(ans)
    return answer

메모

  1. ndarray 와 list

2차원 list는 ndarray로 변형할 수 있다.
연산 후 다시 tolist()를 활용하여 list로 값 그대로 변형할 수 있다.

감사합니다.🍩🍩🍩

profile
https://github.com/min731

0개의 댓글