Mapear dos listas en una sola lista de diccionarios
-
04-07-2019 - |
Pregunta
Imagina que tengo estas listas de python:
keys = ['name', 'age']
values = ['Monty', 42, 'Matt', 28, 'Frank', 33]
¿Existe una forma directa o al menos sencilla de generar la siguiente lista de diccionarios?
[
{'name': 'Monty', 'age': 42},
{'name': 'Matt', 'age': 28},
{'name': 'Frank', 'age': 33}
]
Solución
Aquí está la manera de zip
def mapper(keys, values):
n = len(keys)
return [dict(zip(keys, values[i:i + n]))
for i in range(0, len(values), n)]
Otros consejos
No es bonito, pero aquí hay una sola línea que utiliza una lista de comprensión, zip y stepping:
[dict(zip(keys, a)) for a in zip(values[::2], values[1::2])]
Manera muda, pero una que viene inmediatamente a mi mente:
def fields_from_list(keys, values):
iterator = iter(values)
while True:
yield dict((key, iterator.next()) for key in keys)
list(fields_from_list(keys, values)) # to produce a list.
zip
casi hace lo que quieres; desafortunadamente, en lugar de completar la lista más corta, se rompe. Tal vez hay una función relacionada que ciclos?
$ python
>>> keys = ['name', 'age']
>>> values = ['Monty', 42, 'Matt', 28, 'Frank', 33]
>>> dict(zip(keys, values))
{'age': 42, 'name': 'Monty'}
/ EDIT: Oh, quieres una lista de dict . Los siguientes trabajos (gracias a Peter, también):
from itertoos import cycle
keys = ['name', 'age']
values = ['Monty', 42, 'Matt', 28, 'Frank', 33]
x = zip(cycle(keys), values)
map(lambda a: dict(a), zip(x[::2], x[1::2]))
En la respuesta de Konrad Rudolph
zip casi hace lo que quieres; desafortunadamente, en lugar de completar la lista más corta, se rompe. Tal vez hay una función relacionada que ciclos?
Aquí hay una manera:
keys = ['name', 'age']
values = ['Monty', 42, 'Matt', 28, 'Frank', 33]
iter_values = iter(values)
[dict(zip(keys, iter_values)) for _ in range(len(values) // len(keys))]
No lo llamaré Pythonic (creo que es demasiado inteligente), pero podría ser lo que están buscando.
No hay ningún beneficio en el ciclo de la lista de keys
utilizando itertools
.cycle ()
, porque cada recorrido de keys
corresponde a la creación de un diccionario.
EDITAR: Aquí hay otra forma:
def iter_cut(seq, size):
for i in range(len(seq) / size):
yield seq[i*size:(i+1)*size]
keys = ['name', 'age']
values = ['Monty', 42, 'Matt', 28, 'Frank', 33]
[dict(zip(keys, some_values)) for some_values in iter_cut(values, len(keys))]
Esto es mucho más pythonic: hay una función de utilidad legible con un propósito claro, y el resto del código fluye naturalmente de ella.
Aquí está mi enfoque simple. Parece estar cerca de la idea que tenía @Cheery, excepto que destruyo la lista de entrada.
def pack(keys, values):
"""This function destructively creates a list of dictionaries from the input lists."""
retval = []
while values:
d = {}
for x in keys:
d[x] = values.pop(0)
retval.append(d)
return retval
Otro intento, quizás más tonto que el primero:
def split_seq(seq, count):
i = iter(seq)
while True:
yield [i.next() for _ in xrange(count)]
>>> [dict(zip(keys, rec)) for rec in split_seq(values, len(keys))]
[{'age': 42, 'name': 'Monty'},
{'age': 28, 'name': 'Matt'},
{'age': 33, 'name': 'Frank'}]
Pero depende de ti decidir si es más tonto.
[dict(zip(keys,values[n:n+len(keys)])) for n in xrange(0,len(values),len(keys)) ]
UG-LEEE. No me gustaría ver el código que se parece a eso. Pero se ve bien.
def dictizer(keys, values):
steps = xrange(0,len(values),len(keys))
bites = ( values[n:n+len(keys)] for n in steps)
return ( dict(zip(keys,bite)) for bite in bites )
Todavía un poco feo, pero los nombres ayudan a darle sentido.