web-dev-qa-db-fra.com

Équivalent Python de Zip pour les dictionnaires

Si j'ai ces deux listes:

la = [1, 2, 3]
lb = [4, 5, 6]

Je peux les parcourir comme suit:

for i in range(min(len(la), len(lb))):
    print la[i], lb[i]

Ou plus pythoniquement

for a, b in Zip(la, lb):
    print a, b

Et si j'ai deux dictionnaires?

da = {'a': 1, 'b': 2, 'c': 3}
db = {'a': 4, 'b': 5, 'c': 6}

Encore une fois, je peux itérer manuellement:

for key in set(da.keys()) & set(db.keys()):
    print key, da[key], db[key]

Existe-t-il une méthode intégrée qui me permet d’itérer comme suit?

for key, value_a, value_b in common_entries(da, db):
    print key, value_a, value_b
pythondictionaryiterator
27
Eric

Aucune fonction ou méthode intégrée ne peut le faire. Cependant, vous pouvez facilement définir le vôtre.

def common_entries(*dcts):
    for i in set(dcts[0]).intersection(*dcts[1:]):
        yield (i,) + Tuple(d[i] for d in dcts)

Ceci repose sur la "méthode manuelle" que vous fournissez, mais, comme Zip, peut être utilisé pour un nombre quelconque de dictionnaires.

>>> da = {'a': 1, 'b': 2, 'c': 3}
>>> db = {'a': 4, 'b': 5, 'c': 6}
>>> list(common_entries(da, db))
[('c', 3, 6), ('b', 2, 5), ('a', 1, 4)]

Lorsqu'un seul dictionnaire est fourni en tant qu'argument, il renvoie essentiellement dct.items().

>>> list(common_entries(da))
[('c', 3), ('b', 2), ('a', 1)]
27
Volatility

Vous voudrez peut-être faire une intersection, en utilisant le type Python Set.

da = {'a': 1, 'b': 2, 'c': 3, 'e': 7}
db = {'a': 4, 'b': 5, 'c': 6, 'd': 9}

dc = set(da) & set(db)

for i in dc:
  print i,da[i],db[i]

À votre santé,

K.

9
Koreth

Les vues de dictionnaire sont déjà définies comme dans Python 3. Vous pouvez supprimer set():

for key in da.keys() & db.keys():
    print(key, da[key], db[key])

En Python 2:

for key in da.viewkeys() & db.viewkeys():
    print key, da[key], db[key]
3
pylang

Dans le cas où quelqu'un recherche une solution généralisée:

import operator
from functools import reduce


def Zip_mappings(*mappings):
    keys_sets = map(set, mappings)
    common_keys = reduce(set.intersection, keys_sets)
    for key in common_keys:
        yield (key,) + Tuple(map(operator.itemgetter(key), mappings))

ou si vous souhaitez séparer la clé des valeurs et utiliser une syntaxe telle que

for key, (values, ...) in Zip_mappings(...):
    ...

on peut remplacer la dernière ligne par

yield key, Tuple(map(operator.itemgetter(key), mappings))

Des tests

from collections import Counter


counter = Counter('abra')
other_counter = Counter('kadabra')
last_counter = Counter('abbreviation')
for (character,
     frequency, other_frequency, last_frequency) in Zip_mappings(counter,
                                                                 other_counter,
                                                                 last_counter):
    print('character "{}" has next frequencies: {}, {}, {}'
          .format(character,
                  frequency,
                  other_frequency,
                  last_frequency))

nous donne

character "a" has next frequencies: 2, 3, 2
character "r" has next frequencies: 1, 1, 1
character "b" has next frequencies: 1, 1, 2

(testé sur Python 2.7.12 & Python 3.5.2)

1
Azat Ibrakov