thông minh hơn "đảo ngược" của một từ điển trong python (acc cho một số giá trị giống nhau)?

Question

def revert_dict(d): 
    rd = {} 
    for key in d: 
     val = d[key] 
     if val in rd: 
      rd[val].append(key) 
     else: 
      rd[val] = [key] 
    return rd 

>>> revert_dict({'srvc3': '1', 'srvc2': '1', 'srvc1': '2'}) 
{'1': ['srvc3', 'srvc2'], '2': ['srvc1']} 

Điều này rõ ràng không phải là trao đổi đơn giản các khóa với giá trị: điều này sẽ ghi đè lên một số giá trị (như khóa mới) mà không phải là những gì tôi sau.

Nếu 2 hoặc nhiều giá trị giống nhau cho các khóa khác nhau, các phím được cho là được nhóm trong danh sách.

Chức năng trên hoạt động, nhưng tôi tự hỏi có cách nào thông minh hơn/nhanh hơn không?

Answer 1

Điều đó có vẻ khá tốt. Bạn có thể đơn giản hóa nó một chút bằng cách sử dụng defaultdict:

import collections 

def revert_dict(d): 
    rd = collections.defaultdict(list) 

    for key, value in d.iteritems(): 
     rd[value].append(key) 

    return rd 

Answer 2

Một cách tiếp cận khác. Không chắc chắn nếu điều này là nhanh hơn (tôi nghi ngờ rằng).

from itertools import groupby 

old_dict = {'srvc3': '1', 'srvc2': '1', 'srvc1': '2'} 
funcval = d.__getitem__ 
new_dict = dict((val, list(keys)) for val, keys in \ 
       groupby(sorted(d.iterkeys(), key=funcval), funcval)) 

# new_dict: 
# {'1': ['srvc3', 'srvc2'], '2': ['srvc1']} 

mã ban đầu của bạn chắc chắn không phải là xấu (có thể đọc được) mặc dù tôi có lẽ đã viết nó như thế này (ưu tiên chủ yếu là cá nhân, thực sự):

def revert_dict(d): 
    rd = {} 
    for key, val in d.items(): 
     try: 
      rd[val].append(key) 
     except KeyError: 
      rd[val] = [key] 
    return rd 

Answer 3

Có lẽ không phải là hiệu quả, nhưng:

ks, vs = old_dict.items() 
new_dict = dict((v, [k for k in ks if old_dict[k] == v]) for v in set(vs)) 

Answer 4

def invertDictionary(d): 
    rd = {} 
    for x,y in d.iteritems(): 
      if y in rd.keys(): 
        rd[y].append(x) 
      else: 
        rd[y] = [x] 
    return rd 

Tôi thấy điều này dễ đọc nhất với bản thân mình. Bất kỳ nhược điểm nào?

Answer 5

def revert_dict(d): 
    rd = {} 
    for key,val in d.iteritems(): 
     rd[val] = rd.get(val,[]) + [key] 
    return rd 


print revert_dict({'srvc3': '1', 'srvc2': '1', 'srvc1': '2', 
      'srvc4': '8', 'srvc5': '2', 'srvc6': '2', 
      'srvc7': '77', 'srvc8': '1', 'srvc9': '2', 
      'srvc10': '3', 'srvc11': '1'}) 

kết quả

{'1': ['srvc11', 'srvc3', 'srvc2', 'srvc8'], '77': ['srvc7'], '3': ['srvc10'], '2': ['srvc6', 'srvc5', 'srvc1', 'srvc9'], '8': ['srvc4']} 

thêm:

def revert_dict(d): 
    rd = {} 
    for key,val in d.iteritems(): 
     rd[val] = rd[val]+ [key] if val in rd else [key] 
    return rd