코딩테스트 연습 스터디 진행중 입니다. ✍✍✍
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
메모
2차원 list는 ndarray로 변형할 수 있다.
연산 후 다시 tolist()를 활용하여 list로 값 그대로 변형할 수 있다.
감사합니다.🍩🍩🍩