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

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

本文介绍如何利用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 相关产品和应用

全部下载
Selenium

Selenium

4.51.0 · Windows

用于浏览器自动化和 Web 测试的 Python 工具。

scikit-learn

scikit-learn

1.9.1 · Windows

常用 Python 机器学习算法与模型评估工具库。

Black

Black

26.10.1 · Windows

强调一致性的 Python 自动代码格式化工具。

Ruff

Ruff

0.17.1 · macOS

由 Rust 编写的高速 Python 代码检查和格式化工具。

pytest

pytest

9.1.1 · Windows

简洁而强大的 Python 自动化测试框架。

FastAPI

FastAPI

0.146.0 · Windows

高性能 Python API 框架,支持类型提示和自动接口文档。

Flask

Flask

3.1.3 · Windows

轻量灵活的 Python Web 应用框架。

Django

Django

6.1.2 · Windows

成熟的 Python Web 框架,适合构建内容站点和业务系统。

Spyder

Spyder

6.1.7 · Windows

面向科学计算与数据分析的 Python 集成开发环境。

SciPy

SciPy

1.18.1 · Windows

面向科学、工程和技术计算的 Python 算法库。