[프로그래머스] 행렬의 덧셈 (feat. zip)

 

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

문제 설명 및 test data 

 

문제를 보자마자 생각한건 for i in range(len(A)) 이런 구조를 사용하고싶지 않았다. 풀 수야 있겠지만 주구장창 이중for문으로 풀어대면 발전이 있을까 싶었다. 그래도 일단 내 능력으론 풀 수 있어야 하니..

 

def solution(arr1, arr2):
    answer=[]
    for idx in range(len(arr1)):
        answer.append(matrix_sum(arr1[idx],arr2[idx]))
    
        
    
    return answer


def matrix_sum(a,b):
    return_arr=[]
    for i in range(len(a)):
        temp = a[i]+b[i]
        return_arr.append(temp)
    return return_arr

ㅋㅋ 쪽팔리다 진짜 ㅋㅋㅋ

풀어놓고 나서 고민을 해보다가

def solution(arr1, arr2):
    answer=[]
    for i in range(len(arr1)):
        answer.append(list(map(lambda x,y: x+y, arr1[i],arr2[i])))
        
    return answer

정말 오랜만에 사용해본 map.. 그냥 생각도 잘 안나서 진짜 프리토킹하듯이 코드쓰고 돌렸는데 한번에 됐다...zz

???????????

이래서 "언어"인가?? ㅋㅋ 이제 좀 뿌듯해져서 남들 풀이를 봤는데 옛날부터 간혹 보이던 zip method..가 있었다

 

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

zip이 뭔데 진짜.. 하고 알아보니~

 

(참고 사이트)

https://blog.naver.com/regenesis90/222346953808

[Python] zip() : 2개 이상의 리스트를 상보적으로 묶어 dict를 만들거나, 튜플을 반환하는 함수

 

너무 쉬운 메소드였더라! 앞으로 자주 써야겠다