Fuente: Mi publicación de blog (mejor formato)
Ejemplo
numbers = [1,2,3]
letters = 'abcd'
zip(numbers, letters)
# [(1, 'a'), (2, 'b'), (3, 'c')]
Entrada
Cero o más iterables [1] (ej. Lista, cadena, tupla, diccionario)
Salida (lista)
Primera tupla = (elemento_1 de números, elemento_1 de letras)
2da tupla = (números e_2, letras e_2)
...
n-ésima tupla = (e_n números, e_n letras)
- Lista de n tuplas: n es la longitud del argumento más corto (entrada)
- len (números) == 3 <len (letras) == 4 → corto = 3 → devuelve 3 tuplas
- Longitud de cada tupla = # de argumentos (la tupla toma un elemento de cada arg)
- args = (números, letras); len (args) == 2 → tupla con 2 elementos
ith tuple = (element_i arg1, element_i arg2 ..., element_i arg n)
Casos de borde
1) Cadena vacía: len (str) = 0 = sin tuplas
2) Cadena simple: len (str) == 2 tuplas con len (args) == 1 elemento (s)
zip()
# []
zip('')
# []
zip('hi')
# [('h',), ('i',)]
¡Zip en acción!
1. Cree un diccionario [2] a partir de dos listas
keys = ["drink","band","food"]
values = ["La Croix", "Daft Punk", "Sushi"]
my_favorite = dict( zip(keys, values) )
my_favorite["drink"]
# 'La Croix'
my_faves = dict()
for i in range(len(keys)):
my_faves[keys[i]] = values[i]
zip es una solución elegante, clara y concisa
2. Imprimir columnas en una tabla
"*" [3] se llama "desempaque": f(*[arg1,arg2,arg3]) == f(arg1, arg2, arg3)
student_grades = [
[ 'Morty' , 1 , "B" ],
[ 'Rick' , 4 , "A" ],
[ 'Jerry' , 3 , "M" ],
[ 'Kramer' , 0 , "F" ],
]
row_1 = student_grades[0]
print row_1
# ['Morty', 1, 'B']
columns = zip(*student_grades)
names = columns[0]
print names
# ('Morty', 'Rick', 'Jerry', 'Kramer')
Crédito adicional: descomprimir
zip(*args) se llama "descomprimir" porque tiene el efecto inverso de zip
numbers = (1,2,3)
letters = ('a','b','c')
zipped = zip(numbers, letters)
print zipped
# [(1, 'a'), (2, 'b'), (3, 'c')]
unzipped = zip(*zipped)
print unzipped
# [(1, 2, 3), ('a', 'b', 'c')]
unzipped: tuple_1 = e1 de cada tupla comprimida. tupla_2 = e2 de cadazipped
Notas al pie
- Un objeto capaz de devolver a sus miembros uno a la vez (ej. Lista [1,2,3], cadena 'Me gusta codin', tupla (1,2,3), diccionario {'a': 1, 'b' : 2})
- {clave1: valor1, clave2: valor2 ...}
- "Desembalaje" (*)
* Código:
# foo - function, returns sum of two arguments
def foo(x,y):
return x + y
print foo(3,4)
# 7
numbers = [1,2]
print foo(numbers)
# TypeError: foo() takes exactly 2 arguments (1 given)
print foo(*numbers)
# 3
*tomó numbers(1 arg) y "desempaquetó" sus '2 elementos en 2 argumentos
ziphacer? ¿Por qué deberías tener tres listas? ¿Cómo se verían?