python：嵌套字典中的動態鍵

Question

我正在使用 django shell 嘗試創建這樣一個字典：

{'SECTION ONE': 
  {'Category One': [<Price: price element one>],
  'Category Two': [<Price: price element one>,
                  <Price: price element two>]},
 'SECTION TWO': 
  {'Category One': [<Price: price element one>,
                   <Price: price element two>]}}

但是這段代碼： dict[section][category] = [x] 改變“二”中的“價格元素一”，如下面的結果。

dict = dict()
for x in price.objects.all():
   if section not in dict:
       dict[section] = {
       category: [x]
   }
   else:
        dict[section][category] = [x]
        dict[section][category].append(x)




    {'SECTION ONE': 
      {'Category One': [<Price: price element two>],
      'Category Two': [<Price: price element two>,
                      <Price: price element two>]},
     'SECTION TWO': 
      {'Category One': [<Price: price element two>,
                       <Price: price element two>]}}

你怎么能保留所有的元素？

Answer 1

如果類別尚未在mydict[section中定義，您應該只構建一個新列表，因此：

mydict = {}
for x in price.objects.all():
    if section not in mydict:
         mydict[section] = { category: [x] }
    elif category not in mydict[section]:
         mydict[section][category] = [x]
    else:
         mydict[section][category].append(x)

另一種選擇是使用defaultdict ：

from collections import defaultdict

mydict = defaultdict(lambda: defaultdict(list))
for x in price.objects.all():
    mydict[section][category].append(x)

mydict = {k: dict(v) for k, v in mydict.items()}

---

注意：請不要命名變量dict ，它會覆蓋對內置dict function [Python-doc]的引用。 使用例如mydict 。