Python - intersezione tra una lista e le chiavi di un dizionario

Question

Ho una lista che assomiglia a questo:

l1 = ['200:200', '90:728'] 

Ho un dizionario che assomiglia a questo:

d1 = {'200:200':{'foo':'bar'},'300:300':{'foo':'bar'}} 

ho bisogno di ottenere filtrare il dictioary dove sono presenti solo le chiavi in ​​l1. Il dict dovrebbe essere simile a questo:

result = {'200:200':{'foo':'bar'}} 

In sostanza un incrocio di una lista e le chiavi di una dict mentre tornava la sottosezione del dict.

Come faccio a farlo in modo efficiente quando il tempo è un problema per un set di grandi dimensioni?

Grazie

Answer 1

È possibile utilizzare il seguente codice:

keys = set(l1).intersection(set(d1.keys())) 
result = {k:d1[k] for k in keys} 

EDIT: Come commentatori suggeriscono è possibile sostituire la prima linea con, in Python 2.x:

keys = set(l1).intersection(d1) 

E in Python 3.x:

keys = d1.keys() & l1 

Answer 2

È possibile utilizzare un elenco di comprensione nel costruttore dict:

result = dict([(k,d1[k]) for k in l1 if k in d1]) 

Se siete preoccupati per la rimozione di chiavi duplicate, fare l1 in un set prima:

result = dict([(k,d1[k]) for k in set(l1) if k in d1]) 

Answer 3

In 3.x, questo può essere semplice come:

>>> {k: d1[k] for k in (d1.keys() & l1)} 
{'200:200': {'foo': 'bar'}} 

Sotto 2.7, è possibile utilizzare dict.viewkeys() per ricreare questa funzionalità:

>>> {k: d1[k] for k in (d1.viewkeys() & l1)} 
{'200:200': {'foo': 'bar'}} 

Sotto le vecchie versioni di 2.x, è un po 'più dettagliato:

>>> {k: d1[k] for k in (set(d1).intersection(l1))} 
{'200:200': {'foo': 'bar'}} 

Answer 4

Non sei sicuro di ogni performance soluzione, ma lo farei:

{k: v for k, v in d1.items() if k in l1} 

Answer 5

Definire efficiente. Comunque ecco cosa vorrei fare. Se fosse troppo lento probabilmente lo sposterei su Cython.

s1 = set(l1) 
s2 = set(d1.keys()) 
s3 = s1 & s2 
# now you can access d1 using only keys in s3, or construct a new dict if you like 
d2 = dict([(k,d1[k]) for k in s3]) 

Answer 6

Se l'allocazione della memoria e la deallocazione stanno rendendo questo processo troppo lungo, itertools in soccorso.

import itertools 
result = {dict_key:d1[dict_key] for dict_key in itertools.ifilter(lambda list_item: list_item in d1, l1) } 

Ciò non inutilmente allocare memoria per collezione nuova, e l1 potrebbe facilmente essere un iteratore invece di una lista.