[파이썬] zip, zip_longest 사용법

zip?

Syntax :  zip(*iterators) 
Parameters : Python iterables or containers ( list, string etc ) 
Return Value : Returns a single iterator object, having mapped values from all the 
containers.

 

zip은 2개 이상의 iterables 객체를 한 개의 iterable 객체로 결합시킵니다.

x = [1, 2, 3]
y = ['a', 'b', 'c']

combination = list(zip(x, y))
print(combination)

출력

[(1, 'a'), (2, 'b'), (3, 'c')]

 

일반적을 두 개의 분리된 collections를 한 루프에서 동작 시킬때 활용을 많이 합니다.

names = ['karl', 'john', 'adle']
ages = [30, 20, 25]

for name, age in zip(names, ages):
    print(name, age)

출력

karl 30
john 20
adle 25

 

zip() 함수로 묶을 것을 다시 해체할 수 있습니다.

unpacking 연산자와 zip() 함수를 통해서 가능합니다.

names = ['karl', 'john', 'adle']
ages = [30, 20, 25]

# 묶기
pairs = zip(names, ages)

# unpacking
names, ages = zip(*pairs)
print(names)
print(ages)

출력

('karl', 'john', 'adle')
(30, 20, 25)

 

zip_longest() (길이간 긴 자료형 기준으로 묶기)

zip()으로 묶을 때 길이가 다른 자료형을 묶으면 어떻게 될까요?

짧은 자료형 기준으로 묶이고 나머지는 버려집니다.

# 길이가 다른 자료형
names = ['karl', 'john', 'adle']
ages = [30, 20]

# 길이가 다른 자료형을 묶기
pairs = zip(names, ages)

for name, age in pairs:
    print(name, age)

출력

karl 30
john 20

 

긴 길이의 자료형에 맞춰서 zip()  합하고 짧은 자료형 비어있는 부분은 특정 값으로 채울 수 없을까요?

zip_longest() 함수가 바로 이런 역할을 해낼 수 있습니다.

# zip_longest() syntax
zip_longest( iterable1, iterable2, fillval)

 

아래 예시를 살펴 보시죠

from itertools import zip_longest

# 길이가 다른 자료형
names = ['karl', 'john', 'adle']
ages = [30, 20]

# 길이가 다른 자료형을 묶고, 비어있는 부분은 None으로 채움
pairs = zip_longest(names, ages, fillvalue='-')

for name, age in pairs:
    print(name, age)

출력

karl 30
john 20
adle -