Python中使用defaultdict一行代码构建树形结构
本文介绍如何利用Python内置的collections.defaultdict,通过一行递归定义实现树形数据结构的创建。无需预先赋值,通过键访问即可自动创建子节点。文章展示了如何将其转换为标准字典、迭代处理以及动态添加新节点,提供简洁高效的树形数据操作方案。
树形结构定义与实现
定义
借助Python标准库中的,可以非常便捷地构建一个树形数据结构。
def tree(): return defaultdict(tree)
示例:
采用JSON格式输出。
JSON格式输出与转换
users = tree()
users['harold']['username'] = 'bell'
users['handler']['username'] = 'master'
调用函数,以JSON字符串形式打印结果,输出内容如下所示。
{'harold': {'username': 'bell'}, 'handler': {'username': 'master'}}
无需预先进行赋值操作。
树形结构的迭代与节点添加
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字典对象。
def dicts(t): return {k: dicts(t[k]) for k in t}
使用函数输出转换后的结构。
迭代处理
该树形结构支持便捷的迭代遍历。由于只需简单引用接口,节点便会自动创建。
举例而言,若需解析一组新动物数据并将其加入上述结构中,只需调用相关函数即可。
add(taxonomy, 'Animalia,Chordata,Mammalia,Cetacea,Balaenopteridae,Balaenoptera,blue whale'.split(','))
实现该功能十分简单。
def add(t, keys):
for key in keys:
t = t[key]
以上即为利用Python一行代码实现树形结构的方法,供读者参考。
免责声明:本文内容仅供参考,不代表本站观点。转载或引用请注明出处。









