(Python)행렬의 덧셈(프로그래머스 Lv.1)/zip()활용

행렬의 덧셈(프로그래머스 Lv.1)

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

프로그래머스

 

내 코드

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

베스트 코드

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

이참에 zip()의 활용법을 정리해보았다.

https://star-ccomputer-go.tistory.com/57

Zip() 사용법