如何在Python中創建雙重字典？

Question

我想使用setdefault()函數在python的字典中創建字典。 我正在嘗試使用休假字典列出姓名和出生日期。

names = {'Will': 'january', 'Mary': 'february', 'George': 'march', 'Steven': 'april', 'Peter': 'may'}
dates = {'Will': '7/01', 'George': '21/03', 'Steven': '14/03', 'Mary': '2/02'}

我試圖使用set來實現這一點：

res_dict = dict()
for v, k in names.items():
    for v1, k1 in dates.items():
        res_dict.setdefault(v, {}).append(k)
        res_dict.setdefault(v1, {}).append(k1)
return res_dict

但這給我一個錯誤。

結果應為：

res_dict = {'Will': {'january': '7/01'}, 'Mary' : {'february': '2/02'} ,'George': {'march': '21/03'}, 'Steven': {'april': '14/03'}, 'Peter': {'may': ''}}

如何使用setdefault()獲得所需的結果？

Answer 1

您可以嘗試以下方法：

In [17]: results = {}

In [18]: for k, v in names.iteritems():
    results[k] = {v: dates.setdefault(k, '')}
   ....:
   ....:

In [20]: results
Out[20]: 
{'George': {'march': '21/02'},
 'Mary': {'february': '2/02'},
 'Peter': {'may': ''},
 'Steven': {'april': '14/03'},
 'Will': {'january': '7/01'}}

至於您對添加month和day評論，可以類似地添加它們：

In [28]: for k, v in names.iteritems():
    results[k] = {'month': v, 'day': dates.setdefault(k, '')}
   ....:
   ....:

In [30]: results
Out[30]:
{'George': {'day': '21/02', 'month': 'march'},
 'Mary': {'day': '2/02', 'month': 'february'},
 'Peter': {'day': '', 'month': 'may'},
 'Steven': {'day': '14/03', 'month': 'april'},
 'Will': {'day': '7/01', 'month': 'january'}}

如果您想在不存在值的情況下完全省略day ：

In [8]: results = {}

In [9]: for k, v in names.iteritems():
   ...:     results[k] = {'month': v}
   ...:     if dates.has_key(k):
   ...:         results[k]['day'] = dates[k]
   ...:
   ...:

In [10]: results
Out[10]:
{'George': {'day': '21/03', 'month': 'march'},
 'Mary': {'day': '2/02', 'month': 'february'},
 'Peter': {'month': 'may'},
 'Steven': {'day': '14/03', 'month': 'april'},
 'Will': {'day': '7/01', 'month': 'january'}}

在您只知道日期但不知道月份的奇怪情況下，使用defaultdict遍歷一組鍵（如@KayZhu建議）可能是最簡單的解決方案：

In [1]: from collections import defaultdict

In [2]: names = {'Will': 'january', 'Mary': 'february', 'George': 'march', 'Steven': 'april', 'Peter': 'may'}

In [3]: dates = {'Will': '7/01', 'George': '21/03', 'Steven': '14/03', 'Mary': '2/02', 'Marat': '27/03'}

In [4]: results = defaultdict(dict)

In [5]: for name in set(names.keys() + dates.keys()):
   ...:     if name in names:
   ...:         results[name]['month'] = names[name]
   ...:     if name in dates:
   ...:         results[name]['day'] = dates[name]
   ...:
   ...:

In [6]: for k, v in results.iteritems():
   ...:     print k, v
   ...:
   ...:
George {'day': '21/03', 'month': 'march'}
Will {'day': '7/01', 'month': 'january'}
Marat {'day': '27/03'}
Steven {'day': '14/03', 'month': 'april'}
Peter {'month': 'may'}
Mary {'day': '2/02', 'month': 'february'}

Answer 2

一個簡單的單線：

In [38]: names = {'Will': 'january', 'Mary': 'february', 'George': 'march', 'Steven': 'april', 'Peter': 'may'}

In [39]: dates = {'Will': '7/01', 'George': '21/03', 'Steven': '14/03', 'Mary': '2/02'}

In [40]: dict((name,{names[name]:dates.get(name,'')}) for name in names)

out[40]: 
    {'George': {'march': '21/03'},
     'Mary': {'february': '2/02'},
     'Peter': {'may': ''},
     'Steven': {'april': '14/03'},
     'Will': {'january': '7/01'}}

Answer 3

首先需要從names和dates獲取超集鍵：

>>> for k in set(names.keys() + dates.keys()):
...     res_dict[k] = {names.setdefault(k, ''): dates.setdefault(k, None)}
...     
... 
>>> res_dict
{'Will': {'january': '7/01'}, 'Steven': {'april': '14/03'}, 'Peter': {'may': None},
 'Mary': {'february': '2/02'}, 'George': {'march': '21/03'}}

否則，您會錯過鍵值是dates而不是names 。