zip 내장 함수에 대하여

이번에는 파이썬의 zip함수에 대해서 간단하게 알아보도록 하겠다:

a = [1,2,3,4]
b= ['a', 'b', 'c', 'd']

ab_list = list(zip(a,b))
...
ab_list = [(1, 'a'), (2, 'b'), (3, 'c'), (4, 'd')]

볼 수 있듯이, zip 함수는 enumerate함수와 비슷하지만, 서로 엮어주는 역할을 하는 것에 다른 것을 알 수 있다. 그렇다면 다음 프로그래머스 예제를 한번 보자:

A = [[1,2], [3,4]]
B = [[2,3], [4,5]]

print([c+d for c,d in zip(a,b)] for a,b in zip(A,B))
>>> [[3,5], [7,9]]

즉, A, B 리스트 안에 있는 내용들을 각각 알맞게 매칭 시키려고 할 때 zip 함수를 사용하면 더 편한 것을 알 수 있다.