728x90
1. 행렬의 덧셈
문제
행렬의 덧셈은 행과 열의 크기가 같은 두 행렬의 같은 행, 같은 열의 값을 서로 더한 결과가 됩니다. 2개의 행렬 arr1과 arr2를 입력받아, 행렬 덧셈의 결과를 반환하는 함수, solution을 완성해주세요.
내 풀이
import numpy as np
def solution(arr1, arr2):
arr1_ = np.array(arr1)
arr2_ = np.array(arr2)
answer = arr1_ + arr2_
return answer.tolist()
tolist(): 넘파이 배열을 리스트 형식으로 바꿔주는 메서드
다른 사람 풀이
def solution(arr1, arr2):
answer = []
for i in range(len(arr1)):
row = []
for j in range(len(arr1[i])):
row.append(arr1[i][j] + arr2[i][j])
answer.append(row)
return answer
728x90
'PYTHON > 프로그래머스 코딩테스트 연습' 카테고리의 다른 글
[프로그래머스] 파이썬 코딩테스트 연습 LV1(최대공약수와 최소공배수) (0) | 2024.01.16 |
---|---|
[프로그래머스] 파이썬 코딩테스트 연습 LV1,2(직사각형 별찍기, 최솟값 만들기) (1) | 2024.01.14 |
[프로그래머스] 파이썬 코딩테스트 연습 LV1 (문자열 다루기 기본) (2) | 2024.01.11 |
[프로그래머스] 파이썬 코딩테스트 연습 LV1(부족한 금액 계산하기) (0) | 2024.01.11 |
[프로그래머스] 파이썬 코딩테스트 연습 LV1 문자열 내림차순으로 배치하기 (0) | 2024.01.07 |