Mappa due elenchi in un unico elenco di dizionari
-
04-07-2019 - |
Domanda
Immagina di avere queste liste di Python:
keys = ['name', 'age']
values = ['Monty', 42, 'Matt', 28, 'Frank', 33]
Esiste un modo diretto o almeno semplice per produrre il seguente elenco di dizionari?
[
{'name': 'Monty', 'age': 42},
{'name': 'Matt', 'age': 28},
{'name': 'Frank', 'age': 33}
]
Soluzione
Ecco la zip zip
def mapper(keys, values):
n = len(keys)
return [dict(zip(keys, values[i:i + n]))
for i in range(0, len(values), n)]
Altri suggerimenti
Non è carino, ma ecco una riga che utilizza una comprensione dell'elenco, zip e stepping:
[dict(zip(keys, a)) for a in zip(values[::2], values[1::2])]
Modo stupido, ma uno che mi viene subito in 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
fa quasi quello che vuoi; sfortunatamente, invece di scorrere l'elenco più breve, si rompe. Forse c'è una funzione correlata che cicla?
$ python
>>> keys = ['name', 'age']
>>> values = ['Monty', 42, 'Matt', 28, 'Frank', 33]
>>> dict(zip(keys, values))
{'age': 42, 'name': 'Monty'}
/ EDIT: Oh, vuoi una lista di dict . I seguenti lavori (grazie anche a Peter):
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]))
Nella risposta di Konrad Rudolph
zip fa quasi quello che vuoi; sfortunatamente, invece di scorrere l'elenco più breve, si rompe. Forse c'è una funzione correlata che cicla?
Ecco un modo:
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))]
Non lo chiamerò Pythonic (penso che sia troppo intelligente), ma potrebbe essere quello che stai cercando.
Non c'è alcun vantaggio nel ciclare l'elenco keys
usando itertools
.cycle ()
, perché ogni attraversamento di chiavi
corrisponde alla creazione di un dizionario.
MODIFICA: Ecco un altro modo:
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))]
Questo è molto più pitonico: esiste una funzione di utilità leggibile con uno scopo chiaro e il resto del codice scorre naturalmente da esso.
Ecco il mio approccio semplice. Sembra essere vicino all'idea che @Cheery aveva tranne che distruggo l'elenco di input.
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
Ancora un altro tentativo, forse più stupido del primo:
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'}]
Ma sta a te decidere se è più stupido.
[dict(zip(keys,values[n:n+len(keys)])) for n in xrange(0,len(values),len(keys)) ]
UG-LEEE. Odio vedere un codice simile a quello. Ma sembra giusto.
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 )
Ancora un po 'brutto, ma i nomi aiutano a dargli un senso.