본문 바로가기
코딩테스트 연습

프로그래머스_행렬의덧셈_파이썬

by 부자독개 2020. 12. 23.
반응형
def solution(arr1, arr2):
    for i in range(len(arr2)):
        for j in range(len(arr2[0])):
            arr1[i][j] += arr2[i][j]
    return arr1

처음에 풀었을때 위에처럼 풀었음.

 

def solution(arr1, arr2):
    
    return [[arr1[i][j] + arr2[i][j] for j in range(len(arr2[0]))] for i in range(len(arr2))]

파이썬 답게 한줄로 요약? 했을때

 

def solution(arr1, arr2):
    
    return [[ v1+v2 for v1,v2 in zip(a1,a2)] for a1,a2 in zip(arr1,arr2)]

다른사람의 풀이에서 내장함수 zip을 이용한걸 보고 이해해볼겸 짜봤다.

반응형

댓글