[level 1] 행렬의 덧셈

코딩테스트 (with 프로그래머스)

문제

image

python 풀이

  1. 하나하나 계산
def  solution(arr1, arr2):
	answer = []
	for  i  in  range(len(arr1)):
		tmp=[]
		for  j  in  range(len(arr1[i])):
		tmp.append(arr1[i][j]+arr2[i][j])
		answer.append(tmp)
return  answer
  1. numpy 이용
    def  solution(arr1, arr2):
     arr1=np.array(arr1)
     arr2=np.array(arr2)
     answer=arr1+arr2
     return  answer.tolist()
    

댓글남기기