Инвертировать ключи и значения исходного словаря
например, я вызываю эту функцию, передавая словарь в качестве параметра:
>>> 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']}
если
map = { 'a': 1, 'b':2 }
Я могу только переверните эту карту, чтобы получить:
inv_map = { 1: 'a', 2: 'b' }
С помощью этого
dict((v,k) for k, v in map.iteritems())
кто-нибудь знает, как это сделать в моем случае?
3 ответов
вы можете использовать defaultdict со списком:
>>> 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']}
Если вам все равно, есть ли у вас dict или defaultdict, вы можете опустить последний шаг и использовать defaultdict напрямую.
вы, вероятно, можете использовать defaultdict
или setdefault
здесь.
def invertDictionary(orig_dict):
result = {} # or change to defaultdict(list)
for k, v in orig_dict.iteritems():
result.setdefault(v, []).append(k)
редактировать в 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})
выход:
{1: ['a'], 2: ['b', 'd'], 3: ['c']}
{3: ['a', 'c', 'b']}
{1: ['b', 'd'], 2: ['a', 'c']}