Python - 내장함수 (zip)

1) 내장 함수

 

(1) zip (iterable, iterable, ...)

• zip() 함수는 여러 개의 순회 가능한 iterable 객체를 인자로 받고, 각 객체가 담고 있는 원소를 튜플의 형태로 차례로 접근할 수 있는 반복자 (iterator)를 반환한다.

 

[예시]    
numbers = [1, 2, 3]
letters = [“A”, “B”, “C”]    

for pair in zip(numbers, letters):
    print(pair)

[결과]
>>> (1, 'A')
>>> (2, 'B')
>>> (3, 'C')

→ numbers, letters 리스트의 각 요소를 꺼내어 튜플의 형태로 반환

[병렬 처리]
for number, upper, lower in zip("12345", "ABCDE", "abcde"):
    print(number, upper, lower)

→ 이처럼 여러 그룹의 데이터를 루프를 한 번만 돌면서 병렬로 처리할 수 있으며, zip() 함수는 가변인자를 받기 때문에 2개 이상의 인자를 넘길 수 있다.

→ 튜퓰의 원소를 3개의 변수에 할당하기 위해서 for 문에서 인자 풀기 (UnPacking)을 하였으며 이와 비슷한 경우로는 enumerate() 함수이다 


[Unzip]
numbers = (1, 2, 3)
letters = ("A", "B", "C")
pairs = list(zip(numbers, letters))
print(pairs)
>>> [(1, 'A'), (2, 'B'), (3, 'C')]

numbers, letters = zip(*pairs)

→ zip() 함수로 엮어 놓은 데이터를 다시 해체 (Unzip)하고 싶을 때도 위처럼 zip() 함수를 이용하여 할 수 있다.

[사전 변환]
keys = [1, 2, 3]
values = [“A”, “B”, “C”]

dict(zip(keys, values))                    // {1: 'A', 2: 'B', 3: 'C'}

→ dict() 함수에 키와 값으로 이루어진 튜플을 넘기면 사전이 생성되는 원리를 이용한 것이다!

[주의 사항]
numbers = ["1", "2", "3"]
letters = ["A"]
list(zip(numbers, letters))                // [('1', 'A')]

→ zip() 함수로 넘기는 인자의 길이가 다를 때는 주의해야 하며, 가장 짧은 인자를 기준으로 데이터가 엮이고 나머지는 버려지기 때문이다.