遞歸字典創建python

Question

有沒有什麼辦法來動態創建丟失的鍵如果我想要在子字典中設置一個變量。

基本上我想創建任何丟失的鍵並設置我的值。

self.portdict[switchname][str(neighbor['name'])]['local']['ports'] = [] 

目前，我這樣做，但它的凌亂：

if not switchname in self.portdict: 
    self.portdict[switchname] = {} 
if not str(neighbor['name']) in self.portdict[switchname]: 
    self.portdict[switchname][str(neighbor['name'])] = {} 
if not 'local' in self.portdict[switchname][str(neighbor['name'])]: 
    self.portdict[switchname][str(neighbor['name'])]['local'] = {} 
if not 'ports' in self.portdict[switchname][str(neighbor['name'])]['local']: 
    self.portdict[switchname][str(neighbor['name'])]['local']['ports'] = [] 

有沒有辦法中的一個或兩行而不是這樣做呢？

Answer 1

這是不容易的遞歸的事：

def set_by_path(dct, path, value): 
    ipath = iter(path) 
    p_last = next(ipath) 
    try: 
     while True: 
      p_next = next(ipath) 
      dct = dct.setdefault(p_last, {}) 
      p_last = p_next 
    except StopIteration: 
     dct[p_last] = value 

和測試案例：

d = {} 
set_by_path(d, ['foo', 'bar', 'baz'], 'qux') 
print d # {'foo': {'bar': {'baz': 'qux'}}} 

如果你想擁有它，所以你並不需要的功能，你可以使用下面的defaultdict工廠允許你任意深入地嵌套：

from collections import defaultdict 

defaultdict_factory = lambda : defaultdict(defaultdict_factory) 

d = defaultdict_factory() 
d['foo']['bar']['baz'] = 'qux' 
print d 

Answer 2

使用collections.defaultdict

self.portdict = defaultdict(lambda: defaultdict(lambda: defaultdict(lambda: defaultdict(lambda: [])))) 

Answer 3

有一個近距離觀察collections.defaultdict：

from collections import defaultdict 
foo = defaultdict(dict) 
foo['bar'] = defaultdict(dict) 
foo['bar']['baz'] = defaultdict(dict) 
foo['bar']['baz']['aaa'] = 1 
foo['bor'] = 0 
foo['bir'] = defaultdict(list) 
foo['bir']['biz'].append(1) 
foo['bir']['biz'].append(2) 

print foo 
defaultdict(<type 'dict'>, {'bir': defaultdict(<type 'list'>, {'biz': [1, 2]}), 'bor': 0, 'bar': defaultdict(<type 'dict'>, {'baz': defaultdict(<type 'dict'>, {'aaa': 1})})}) 

Answer 4

我碰到的，在過去類似的問題。我發現defaultdict對我來說是正確的答案 - 但是編寫超長的定義（比如@ o11c的答案或@ Apero的答案）是不好的。以下是我想出了替代：

from collections import defaultdict 
from functools import partial 

def NestedDefaultDict(levels, baseFn): 
    def NDD(lvl): 
     return partial(defaultdict, NDD(lvl-1)) if lvl > 0 else baseFn 
    return defaultdict(NDD(levels-1)) 

這將創建嵌套字典levels字典。所以如果你有levels = 3，那麼你需要3個鍵來訪問底層的值。第二個參數是一個用來創建底層值的函數。類似list或lambda: 0或甚至dict會運作良好。

下面是一個使用「自動」鍵，帶4 levels，並且list作爲默認函數的例子：

>>> x = NestedDefaultDict(4, list) 
>>> x[1][2][3][4].append('hello') 
>>> x 
defaultdict(<functools.partial object at 0x10b5c22b8>, {1: defaultdict(<functools.partial object at 0x10b5c2260>, {2: defaultdict(<functools.partial object at 0x10b5c2208>, {3: defaultdict(<type 'list'>, {4: ['hello']})})})}) 

我認爲這基本上你想要什麼，你的問題的情況下。你4「級別」是開關名，鄰居的名字，本地，&港口 - 和它看起來像你想list在最下層的存儲您的端口。

使用2 levels和lambda: 0作爲默認又如：

>>> y = NestedDefaultDict(2, lambda: 0) 
>>> y['foo']['bar'] += 7 
>>> y['foo']['baz'] += 10 
>>> y['foo']['bar'] += 1 
>>> y 
defaultdict(<functools.partial object at 0x1021f1310>, {'foo': defaultdict(<function <lambda> at 0x1021f3938>, {'baz': 10, 'bar': 8})})