Função built-in Python zip()

Da documentação do Python 3

Itera sobre vários iteráveis em paralelo, produzindo tuplas com um item de cada um.

Introdução

A função zip() em Python é uma função built-in que aceita dois ou mais iteráveis (como listas, tuplas ou strings) e os agrega em um único iterador de tuplas. Cada tupla contém elementos de todos os iteráveis de entrada na mesma posição.

Exemplos

furniture = ['table', 'chair', 'rack', 'shelf']
price = [100, 50, 80, 40]

for item, amount in zip(furniture, price):
    print(f'The {item} costs ${amount}')
The table costs $100
The chair costs $50
The rack costs $80
The shelf costs $40

Outros Casos de Uso

A função zip em Python mescla múltiplos iteráveis em tuplas.

# Combinando três listas
list1 = [1, 2, 3]
list2 = ['a', 'b', 'c']
list3 = [True, False, True]

zipped = zip(list1, list2, list3)
print(list(zipped))
[(1, 'a', True), (2, 'b', False), (3, 'c', True)]

Descompactando (Unzipping)

# Descompactando um objeto zipado
zipped = [(1, 'a'), (2, 'b'), (3, 'c')]
list1, list2 = zip(*zipped)
print(list1)
print(list2)
(1, 2, 3)
('a', 'b', 'c')

Mais Exemplos

Zipando com Comprimentos Diferentes

zip para de criar tuplas quando o iterável mais curto se esgota.

numbers = [1, 2, 3]
letters = ['a', 'b']

for num, letter in zip(numbers, letters):
    print(f'{num} -> {letter}')
1 -> a
2 -> b

Usando zip com Dicionários

Você pode usar zip para combinar chaves e valores de duas listas em um dicionário.

keys = ['name', 'age', 'city']
values = ['Alice', 25, 'New York']

my_dict = dict(zip(keys, values))
print(my_dict)
{'name': 'Alice', 'age': 25, 'city': 'New York'}

Usando zip com List Comprehensions

Você pode usar zip em list comprehensions para um código mais conciso.

list1 = [1, 2, 3]
list2 = [4, 5, 6]

summed = [x + y for x, y in zip(list1, list2)]
print(summed)
[5, 7, 9]