使用 Python 內(nèi)建的defaultdict 方法可以輕松定義一個樹的數(shù)據(jù)結(jié)構(gòu)。

簡單的說樹也可以是一個字典數(shù)據(jù)結(jié)構(gòu)
 

def tree(): return defaultdict(tree)

這就是全部，就一行代碼。

如果你繼續(xù)下面的代碼，需要先引入
 

from collections import defaultdict

實(shí)例

JSON-esque

現(xiàn)在我們創(chuàng)建一個 JSON-esque 嵌套字典無需顯式的創(chuàng)建子字典：
 

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

然后可通過 <code>print(json.dumps(users))</code> 來打印 JSON 數(shù)據(jù)，結(jié)果如下：
 

{"harold": {"username": "hrldcpr"}, "handler": {"username": "matthandlersux"}}

無需賦值

我們不需要通過賦值就可以創(chuàng)建結(jié)構(gòu)：
 

taxonomy = tree()taxonomy['Animalia']['Chordata']['Mammalia']['Carnivora']['Felidae']['Felis']['cat']taxonomy['Animalia']['Chordata']['Mammalia']['Carnivora']['Felidae']['Panthera']['lion']taxonomy['Animalia']['Chordata']['Mammalia']['Carnivora']['Canidae']['Canis']['dog']taxonomy['Animalia']['Chordata']['Mammalia']['Carnivora']['Canidae']['Canis']['coyote']taxonomy['Plantae']['Solanales']['Solanaceae']['Solanum']['tomato']taxonomy['Plantae']['Solanales']['Solanaceae']['Solanum']['potato']taxonomy['Plantae']['Solanales']['Convolvulaceae']['Ipomoea']['sweet potato']

要打印有好的信息，需要轉(zhuǎn)成標(biāo)準(zhǔn)的字典對象：
 

def dicts(t): return {k: dicts(t[k]) for k in t}

現(xiàn)在可通過 pprint(dicts(taxonomy)) 進(jìn)行打印了:
 

{'Animalia': {'Chordata': {'Mammalia': {'Carnivora': {'Canidae': {'Canis': {'coyote': {},                                      'dog': {}}},                           'Felidae': {'Felis': {'cat': {}},                                 'Panthera': {'lion': {}}}}}}}, 'Plantae': {'Solanales': {'Convolvulaceae': {'Ipomoea': {'sweet potato': {}}},              'Solanaceae': {'Solanum': {'potato': {},                           'tomato': {}}}}}}

子結(jié)構(gòu)也被當(dāng)作是字典對象了，而葉子節(jié)點(diǎn)是一個空的字典對象

迭代

可以使用有趣的方法對樹進(jìn)行迭代。

例如我們解析一個動物的列表并添加到之前定義的 taxonomy 中，我們可以使用如下代碼：
 

add(taxonomy,  'Animalia,Chordata,Mammalia,Cetacea,Balaenopteridae,Balaenoptera,blue whale'.split(','))

簡化實(shí)現(xiàn)：
 

def add(t, keys): for key in keys:  t = t[key]

我們?nèi)匀粺o需賦值：
 

{'Animalia': {'Chordata': {'Mammalia': {'Carnivora': {'Canidae': {'Canis': {'coyote': {},                                      'dog': {}}},                           'Felidae': {'Felis': {'cat': {}},                                 'Panthera': {'lion': {}}}},                    'Cetacea': {'Balaenopteridae': {'Balaenoptera': {'blue whale': {}}}}}}}, 'Plantae': {'Solanales': {'Convolvulaceae': {'Ipomoea': {'sweet potato': {}}},              'Solanaceae': {'Solanum': {'potato': {},                           'tomato': {}}}}}}