22/01/22 행렬의 합 (lev_1)

1. 행렬의 합

문제설명 :
행렬의 덧셈은 행과 열의 크기가 같은 두 행렬의 같은 행, 같은 열의 값을 서로 더한 결과가 됩니다. 2개의 행렬 arr1과 arr2를 입력받아, 행렬 덧셈의 결과를 반환하는 함수, solution을 완성해주세요.

제한사항 :
❗️ 행렬 arr1, arr2의 행과 열의 길이는 500을 넘지 않습니다.

입출력 예시 :

arr1		arr2		return
ex1)
[[1,2],[2,3]]	[[3,4],[5,6]]	[[4,6],[7,9]]
ex2)
[[1],[2]]	[[3],[4]]	[[4],[6]]

풀이 1

arr_1 = [[1,3],[5,5],[2,7]]
arr_2 = [[1,3],[4,1],[10,12]]

def sol(arr1, arr2):

    answer = arr1                           # 합을 담을 list matrix 사이즈를 매개 변수로 받는 list matrix중 하나로 초기화

    for i in range(len(arr1)):              # list 를 index 0부터 끝까지 반복
        for i2 in range(len(arr1[i])):      # list i번째 index 인자(리스트)의 내부 index 숫자 만큼 반복
            sum0 = arr1[i][i2] + arr2[i][i2]# arr1 인자와 arr2 인자의 합
            answer[i][i2] = sum0            # 위의 두 인자 합을 answer list 의 해당 위치에 할당
    return answer                           # answer matrix 반환
result = sol(arr_1,arr_2)

print(result)

좋은 웹페이지 즐겨찾기