Invertir claves y valores del diccionario original.

Por ejemplo, llamo a esta función pasando un diccionario como parámetro:

>>> inv_map({'a':1, 'b':2, 'c':3, 'd':2}) {1: ['a'], 2: ['b', 'd'], 3: ['c']} >>> inv_map({'a':3, 'b':3, 'c':3}) {3: ['a', 'c', 'b']} >>> inv_map({'a':2, 'b':1, 'c':2, 'd':1}) {1: ['b', 'd'], 2: ['a', 'c']} 

Si

 map = { 'a': 1, 'b':2 } 

Solo puedo invertir este mapa para obtener:

 inv_map = { 1: 'a', 2: 'b' } 

usando este

 dict((v,k) for k, v in map.iteritems()) 

¿Alguien sabe cómo hacer eso por mi caso?

    Puedes usar un defaultdict con lista:

     >>> from collections import defaultdict >>> m = {'a': 2, 'b': 1, 'c': 2, 'd': 1} >>> dd = defaultdict(list) >>> for k, v in m.iteritems(): ... dd[v].append(k) ... >>> dict(dd) {1: ['b', 'd'], 2: ['a', 'c']} 

    Si no le importa si tiene un dict o un fallo por defecto, puede omitir el último paso y usar el por defecto directamente.

    Probablemente puedes usar defaultdict o setdefault aquí.

     def invertDictionary(orig_dict): result = {} # or change to defaultdict(list) for k, v in orig_dict.iteritems(): result.setdefault(v, []).append(k) 

    EDITAR en python 2.7:

     from itertools import groupby def inv_map(d): return {k : [i[0] for i in list(v)] for k, v in groupby(d.items(),lambda x:x[1])} print inv_map({'a':1, 'b':2, 'c':3, 'd':2}) print inv_map({'a':3, 'b':3, 'c':3}) print inv_map({'a':2, 'b':1, 'c':2, 'd':1}) 

    Salida:

     {1: ['a'], 2: ['b', 'd'], 3: ['c']} {3: ['a', 'c', 'b']} {1: ['b', 'd'], 2: ['a', 'c']}