Python中使用defaultdict一行代码构建树形结构

今天早睡
今天早睡
发布时间:2026-10-13 | 5人浏览过
原创

本文介绍如何利用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一行代码实现树形结构的方法,供读者参考。

相关标签:

本站声明

本文内容仅供参考,不代表本站观点。转载或引用请注明出处。如发现内容问题,请通过联系我们页面反馈。

Python 相关产品和应用

全部下载

专题列表