使用defaultdict的Python單行樹。如何減少所需參數的數量？

Question

我使用這個gist's defaultdict單行樹。

def tree(): return defaultdict(tree) 

當前，您必須爲每個要添加的節點提供單獨的[]。

即：

users = tree() 
users['harold']['username']['hrldcpr'] 
users['handler']['username']['matthandlersux'] 

---

我的問題是，我如何能夠壓平輸入，這樣我可以提供一個列表來達到同樣的效果？

即：

users = tree() 
users['harold', 'username', 'hrldcpr'] 
users['handler', 'username', 'matthandlersux'] 

感謝您的幫助！

Answer 1

通過提供list和tree作爲參數，您可以簡單地定義一個函數，如insert來創建該節點。

def insert(tree, List): 
    for node in List: 
     tree = tree[node] 

users = tree() 
insert(users, ['harold', 'username', 'hrldcpr']) 

將創建一個結構{'harold' : {'username' : {'hrldcp' : {} } } }