2012-01-19 27 views
6

Tengo un tipo de lógica detallada que me gustaría compactar con algunas comprensiones.Comprensión de listas y dict combinados

Básicamente, tengo un objeto dict que estoy leyendo desde el que tengo 16 valores que me preocupan. Estoy recibiendo las llaves que quiero con la siguiente comprensión:

["I%d" % (i,) for i in range(16)] 

El diccionario fuente de tipo de parece a esto:

{ "I0": [0,1,5,2], "I1": [1,3,5,2], "I2": [5,9,10,1], ... } 

me gustaría mapa esencialmente este diccionario a ser algo así como esto:

[ 
    { "I0": 0, "I1": 1, "I2": 5, ... } 
    { "I0": 1, "I1": 3, "I2": 9, ... } 
    ... 
] 

Cómo puedo trazar me cosas con lista y comprensiones del diccionario para transformar mi dict fuente en mi lista de destinos de los diccionarios?

Respuesta

7

Esta es una solución totalmente funcional que se puede aplicar en tamaño arbitrario.

d = { "I0": [0,1,5,2], "I1": [1,3,5,2], "I2": [5,9,10,1]} 
map(dict, zip(*map(lambda (k, v): map(lambda vv: (k, vv), v), d.iteritems()))) 

para elaborar: (estoy usando ipython y el subrayado _ significa la salida anterior)

In [1]: d = {'I0': [0, 1, 5, 2], 'I1': [1, 3, 5, 2], 'I2': [5, 9, 10, 1]} 

In [2]: map(lambda (k, v): map(lambda vv: (k, vv), v), _.iteritems()) 
Out[2]: 
[[('I1', 1), ('I1', 3), ('I1', 5), ('I1', 2)], 
[('I0', 0), ('I0', 1), ('I0', 5), ('I0', 2)], 
[('I2', 5), ('I2', 9), ('I2', 10), ('I2', 1)]] 

In [3]: zip(*_) 
Out[3]: 
[(('I1', 1), ('I0', 0), ('I2', 5)), 
(('I1', 3), ('I0', 1), ('I2', 9)), 
(('I1', 5), ('I0', 5), ('I2', 10)), 
(('I1', 2), ('I0', 2), ('I2', 1))] 

In [4]: map(dict, _) 
Out[4]: 
[{'I0': 0, 'I1': 1, 'I2': 5}, 
{'I0': 1, 'I1': 3, 'I2': 9}, 
{'I0': 5, 'I1': 5, 'I2': 10}, 
{'I0': 2, 'I1': 2, 'I2': 1}] 
6

Cómo lo resolvería:

En primer lugar, me gustaría tener para cada tecla una lista que contiene tuplas, donde el primer elemento sería la clave y el segundo sería uno de los valores de la lista:

>>> [ [ (k, i) for i in l] for k, l in d.items() ] 
[[('I1', 1), ('I1', 3), ('I1', 5), ('I1', 2)], 
    [('I0', 0), ('I0', 1), ('I0', 5), ('I0', 2)], 
    [('I2', 5), ('I2', 9), ('I2', 10), ('I2', 1)]] 

entonces yo transversales esta lista, la creación de una lista de tuplas que contienen cada tecla correspondiente, utilizando la función zip:

>>> list(zip(*[ [ (k, i) for i in l] for k, l in d.items() ])) 
[(('I1', 1), ('I0', 0), ('I2', 5)), 
(('I1', 3), ('I0', 1), ('I2', 9)), 
(('I1', 5), ('I0', 5), ('I2', 10)), 
    (('I1', 2), ('I0', 2), ('I2', 1))] 

Estos sublistas pueden pasado como parámetro al constructor dict:

>>> [dict(lp) for lp in zip(*[ [ (k, i) for i in l] for k, l in d.items() ])] 
[{'I0': 0, 'I1': 1, 'I2': 5}, 
{'I0': 1, 'I1': 3, 'I2': 9}, 
{'I0': 5, 'I1': 5, 'I2': 10}, 
{'I0': 2, 'I1': 2, 'I2': 1}] 

en la práctica, sin embargo, yo nunca recomendaría uno hacer tal cosa en una sola línea:

>>> pairs = [ [ (k, i) for i in l] for k, l in d.items() ] 
>>> transversed = zip(*pairs) 
>>> ds = [dict(t) for t in transversed] 
>>> pprint(ds) 
[{'I0': 0, 'I1': 1, 'I2': 5}, 
{'I0': 1, 'I1': 3, 'I2': 9}, 
{'I0': 5, 'I1': 5, 'I2': 10}, 
{'I0': 2, 'I1': 2, 'I2': 1}] 

en realidad, wo Por ejemplo, di que publiqué esta respuesta principalmente para sugerirle que divida su solución en más de una línea.

+0

+1 para la elaboración paso a paso. Nuestros enfoques son los mismos, excepto que hice un uso extenso de 'map' :) – qiao

+0

@qiao sí, adoptamos el mismo enfoque de una manera más abstracta. Solo tengo cierta preferencia por las listas de comprensión, pero su solución es excelente. – brandizzi

0

hay solución a corto y sencillo:

keys = data.keys() 
values = data.values() 
transformed = [dict(zip(keys, t)) for t in zip(*values)] 

La clave aquí se transpone matriz de valores, que se realiza con zip(*values), entonces solo reconstruir predice.

Cuestiones relacionadas