[프로그래머스] Level 1_행렬의 덧셈

유형

연습문제

 

문제

https://programmers.co.kr/learn/courses/30/lessons/12950

코딩테스트 연습 - 행렬의 덧셈

 

나의 문제풀이

- 핵심은 이중리스트 초기화

 - 이중리스트에 한 줄씩 추가하기

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

 

다른 사람의 문제풀이

 1) 새로운 리스트 만들지 않고 기존 리스트 중 하나의 리스트에 값 더하기

def sumMatrix(A,B):
    for i in range(len(A)) :
        for j in range(len(A[0])):
            A[i][j] += B[i][j] 
    return A

 

 2) zip 이용한 코드

• zip : 리스트에서 인덱스가 동일한 요소끼리 묶어주는 것
  https://ooyoung.tistory.com/60

파이썬 zip( ) 함수_ 길이가 같은 리스트 등의 요소를 묶어주는 함수 (Python)

def sumMatrix(A,B):
    answer = [[c + d for c, d in zip(a, b)] for a, b in zip(A,B)]
    return answer

 

3) numpy 이용

import numpy as np

def sumMatrix(A,B):
    A=np.array(A)
    B=np.array(B)
    answer=A+B
    return answer.tolist()