Python倒詞典

Question

我目前正在編寫一個函數，它需要一個具有不可變值的字典並返回一個反轉字典。到目前爲止，我的代碼越來越極其簡單的試驗權，但它仍然有一些扭結制定出

def dict_invert(d): 

    inv_map = {v: k for k, v in d.items()} 

    return inv_map 

    list1 = [1,2,3,4,5,6,7,8] 
    list2 = {[1]:3245,[2]:4356,[3]:6578} 
    d = {['a']:[],['b']:[]} 
    d['a'].append(list1) 
    d['b'].append(list2) 

如何解決我的代碼，以便它通過測試案例？

我唯一的想法是將列表2改爲[1:32, 2:43, 3:54, 4:65];然而，我仍然有"[]"在正確的位置有問題。我不知道該怎麼做。

Answer 1

訣竅是要認識到多個鍵可以具有相同的值，所以當反轉時，您必須確保您的值映射到鍵列表。

from collections import defaultdict 

def dict_invert(d): 
    inv_map = defaultdict(list) 
    for k, v in d.items(): 
    inv_map[v].append(k) 
    return inv_map 

編輯：

只是增加了一點更有幫助信息...

的defaultdict(list)使得字典的默認值= list()當通過[]或get訪問（通常時它會分別增加KeyError或返回None）。

有了這一defaultdict，你可以使用一些邏輯來組鍵一起......這裏有一個例子來說明（從上面我的意見）

原字典：K0 - > V0，K1 - > V0，K2 - > V0

應該反轉爲：V0 - > [K0，K1，K2]

編輯2：

你的測試，似乎迫使你到使用正常的字典，在這種情況下......

def dict_invert(d): 
    inv_map = {} 
    for k, v in d.items(): 
     if v not in inv_map: 
      inv_map[v] = [] 
     inv_map[v].append(k) 
    return inv_map