創建或追加到字典中的列表 - 可以縮短嗎？

Question

可以使用itertools和set來縮短這個Python代碼並仍然可讀嗎？

result = {} 
for widget_type, app in widgets: 
    if widget_type not in result: 
     result[widget_type] = [] 
    result[widget_type].append(app) 

我覺得這只是：

widget_types = zip(*widgets)[0] 
dict([k, [v for w, v in widgets if w == k]) for k in set(widget_types)]) 

Answer 1

可以使用defaultdict(list)。

from collections import defaultdict 

result = defaultdict(list) 
for widget_type, app in widgets: 
    result[widget_type].append(app) 

Answer 2

到defaultdict另一種方法是使用標準字典的setdefault方法：

result = {} 
for widget_type, app in widgets: 
    result.setdefault(widget_type, []).append(app) 

這依賴於一個事實，即列表是可變的，那麼什麼是從setdefault返回相同列表作爲一個在字典中，因此你可以追加到它。

Answer 3

可能會有點慢，但工作

result = {} 
for widget_type, app in widgets: 
    result[widget_type] = result.get(widget_type, []) + [app]