[연습문제] 행렬의 덧셈 with Python

문제

 

내가 작성한 코드

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

 

코드 설명

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

> answer 변수에서 arr1 길이 만큼의 리스트안의 리스트 변수를 만들어 줍니다.

> 그리고 이중 for문을 사용해서 처음에는 arr1의 길이만큼 반복문을 진행하고, 두 번째 반복문에는 arr1 안에 리스트 값을 반복시켜줍니다.

> 그리고 answer[i].append(arr1[i][j] + arr2[i][j] ) 코드를 작성해서 arr1과 arr2가 같은 크기와 같은 위치의 값을 더하면 되므로 인덱스 값은 동일하게 작성을 해줍니다.

> 그러면 answer 변수에 결과값을 구할 수 있게됩니다.

 

참고할만한 코드

def solution(arr1, arr2):
    answer = [ [c+d for c, d in zip(a, b)] for a, b in zip(arr1, arr2)]
    return answer

 

import numpy as np

def solution(arr1, arr2):
    return (np.array(arr1) + np.array(arr2)).tolist()