一行 Python 代碼搞定一棵樹

使用 Python 內建的 defaultdict 方法能夠輕鬆定義一個樹的數據結構。

簡單的說樹也能夠是一個字典數據結構

 

 

 

 

 

Python

 

1	def tree(): return defaultdict(tree)

這就是所有，就一行代碼。

若是你繼續下面的代碼，須要先引入

 

 

 

 

 

Python

 

1	from collections import defaultdict

實例

JSON-esque

如今咱們建立一個 JSON-esque 嵌套字典無需顯式的建立子字典：

 

 

 

 

 

Python

 

1 2 3

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

而後可經過 <code>print(json.dumps(users))</code> 來打印 JSON 數據，結果以下：

 

 

 

 

 

 

Python

 

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

無需賦值

 

咱們不須要經過賦值就能夠建立結構：

 

 

 

 

 

 

Python

 

1 2 3 4 5 6 7 8

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']

要打印有好的信息，須要轉成標準的字典對象：

 

 

 

 

 

Python

 

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

如今可經過 pprint(dicts(taxonomy)) 進行打印了:

 

 

 

 

 

 

Python

 

1 2 3 4 5 6 7

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

子結構也被看成是字典對象了，而葉子節點是一個空的字典對象

迭代

能夠使用有趣的方法對樹進行迭代。

例如咱們解析一個動物的列表並添加到以前定義的 taxonomy 中，咱們能夠使用以下代碼：

 

 

 

 

 

 

Python

 

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

簡化實現：

 

 

 

 

 

Python

 

1 2 3

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

咱們仍然無需賦值：

 

 

 

 

 

 

Python

 

1 2 3 4 5 6 7 8

{'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': {}}}}}}

結論

上面說起的這些可能用處不大，只是作了一些有意思的代碼。

若是你喜歡 Python 的話，把這個當成是樂趣來理解。