Obtenir un sous-ensemble d'un dictionnaire Python

Question

J'ai un dictionnaire :

{'key1':1, 'key2':2, 'key3':3}

J'ai besoin de passer un sous-ensemble de ce dictionnaire à un code tiers. Il veut seulement un dictionnaire contenant des clés ['key1', 'key2', 'key99'] et s'il obtient une autre clé (par exemple 'key3' ), il explose dans un méchant désordre. Le code en question est hors de mon contrôle et je me retrouve dans une situation où je dois nettoyer mon dictionnaire.

Quelle est la meilleure façon de limiter un dictionnaire à un ensemble de clés ?

Étant donné l'exemple de dictionnaire et les clés autorisées ci-dessus, je veux :

{'key1':1, 'key2':2}

Answer 1

In [38]: adict={'key1':1, 'key2':2, 'key3':3}
In [41]: dict((k,adict[k]) for k in ('key1','key2','key99') if k in adict)
Out[41]: {'key1': 1, 'key2': 2}

En Python3 (ou Python2.7 ou ultérieur), vous pouvez le faire avec une balise dict-compréhension aussi :

>>> {k:adict[k] for k in ('key1','key2','key99') if k in adict}
{'key2': 2, 'key1': 1}

Answer 2

dict(filter(lambda i:i[0] in validkeys, d.iteritems()))

Answer 3

Dans les versions modernes de Python (2.7+,3.0+), utilisez un dictionnaire de compréhension :

d = {'key1':1, 'key2':2, 'key3':3}
included_keys = ['key1', 'key2', 'key99']

{k:v for k,v in d.items() if k in included_keys}

Answer 4

Une autre solution sans si en dict compréhension.

>>> a = {'key1':1, 'key2':2, 'key3':3}
>>> b = {'key1':1, 'key2':2}
>>> { k:a[k] for k in b.keys()}
{'key2': 2, 'key1': 1}

Answer 5

Ma façon de procéder est la suivante .

from operator import itemgetter

def subdict(d, ks):
    return dict(zip(ks, itemgetter(*ks)(d)))

my_dict = {'key1':1, 'key2':2, 'key3':3}

subdict(my_dict, ['key1', 'key3'])

Mise à jour

Je dois cependant admettre que l'implémentation ci-dessus ne gère pas le cas où la longueur de la variable ks est 0 ou 1. Le code suivant gère la situation et ce n'est plus un one-liner.

def subdict(d, ks):
    vals = []
    if len(ks) >= 1:
        vals = itemgetter(*ks)(d)
        if len(ks) == 1:
            vals = [vals]
    return dict(zip(ks, vals))