如何用Python编写一个通用的树状结构转换器解决多级分类问题?

冬涛姑娘_4213

冬涛姑娘_4213

2026-07-15

989人浏览

原创

直接递归构建树漏根节点的主因是硬编码判断parent_id值(如==0),而根节点实际可能为none/null,导致顶层分类被跳过;正确做法是取id列与parent_id列的差集确定真实根节点。

如何用python编写一个通用的树状结构转换器解决多级分类问题?

为什么直接递归构建树会漏掉根节点

常见错误是只从数据库查出所有分类,然后按 parent_id 逐层找子节点,但忽略了一个关键事实:根节点的 parent_id 通常是 None、0 或 NULL,而不同业务系统约定不一致。如果硬编码判断 parent_id == 0,遇到用 None 表示根的系统就会跳过全部顶层分类。

实操建议:

  • 先扫描一遍原始数据,收集所有出现过的 parent_id 值,再取差集得到真正的根——即那些在 id 列中存在、但在 parent_id 列中不存在的 id
  • 用字典预建所有节点引用:node_map = {item['id']: {**item, 'children': []} for item in items},避免每次递归都重新查数据
  • 遍历一次完成挂载:对每个非根节点,用 node_map.get(item['parent_id']) 安全获取父节点并追加到其 children 列表

如何让转换器兼容 SQLAlchemy、Django ORM 和纯字典数据

核心不是适配 ORM,而是统一输入结构。SQLAlchemy 的 Row、Django 的 Model 实例、甚至 JSON 字典,只要能提取出 id 和 parent_id 字段,就能进同一套逻辑。

实操建议:

  • 定义一个提取函数参数 id_key='id' 和 parent_key='parent_id',默认值覆盖大多数场景,调用时可显式传入 id_key='category_id' 适配定制字段名
  • 对任意对象,用 getattr(obj, id_key, obj.get(id_key) if hasattr(obj, 'get') else None) 统一取值,兼顾属性访问和字典访问
  • 避免在转换器内部做 isinstance(obj, Model) 类型判断——类型检查交给使用者做,转换器只处理“能取到两个键的任意结构”

扁平数据转树后,怎么安全支持前端无限滚动展开

前端常需要点击某节点才加载其子节点(懒加载),但树状结构已一次性生成,直接返回整棵树会造成冗余传输。更麻烦的是,某些节点可能有几百个子项,但用户只点开其中一两个。

Shadows Python Sensei
Shadows Python Sensei

Python 最佳实践助手——代码规范、设计模式、性能优化、测试与类型注解。适用于编写或审查 Python 代码。

下载

实操建议:

  • 转换器输出时保留原始扁平列表的索引映射,例如 tree_node['flat_index'] = i,方便后续按需反查
  • 给每个节点加一个 has_children: bool 字段,值来自 any(child['parent_id'] == node_id for child in items),而不是看 children 是否为空列表(后者在懒加载下恒为 [])
  • 不主动过滤或截断数据——是否分页、是否折叠、是否限制层级,这些属于消费方逻辑,转换器只保证结构可推导、字段可扩展

JSON 序列化时报错 “Object of type set is not JSON serializable” 怎么办

这通常是因为节点里混入了 set、datetime、自定义类等非标准类型,尤其在从 ORM 取数据时,某些字段(如 PostgreSQL 的 ARRAY 或 JSONB)可能被映射成 set 或 UUID。

实操建议:

  • 转换器不负责序列化,只返回纯净的嵌套字典+列表结构;把 json.dumps() 留给上层,并提示使用者传入 default=str 或自定义 JSONEncoder
  • 若必须内置序列化,用 jsonable_encoder(FastAPI)或 dataclasses.asdict()(对 dataclass)比手动写 if isinstance(x, datetime) 更可靠
  • 最简兜底:在构建 node_map 时就做类型清洗,例如 'created_at': item.get('created_at').isoformat() if hasattr(item.get('created_at'), 'isoformat') else item.get('created_at')

真正难的不是建树,而是定义清楚“谁来决定哪个字段算 ID、哪个算父级、空值怎么表达”。一旦协议模糊,后面所有优化都会打滑。

Python免费学习笔记(深入):立即使用
在学习笔记中,你将探索 Python 的核心概念和高级技巧!

相关专题

更多
python打包成可执行文件
python打包成可执行文件

本专题为大家带来python打包成可执行文件相关的文章,大家可以免费的下载体验。

2023.07.20

1591

4

python能做什么
python能做什么

python能做的有:可用于开发基于控制台的应用程序、多媒体部分开发、用于开发基于Web的应用程序、使用python处理数据、系统编程等等。本专题为大家提供python相关的各种文章、以及下载和课程。

2023.07.25

3804

7

format在python中的用法
format在python中的用法

Python中的format是一种字符串格式化方法,用于将变量或值插入到字符串中的占位符位置。通过format方法,我们可以动态地构建字符串,使其包含不同值。php中文网给大家带来了相关的教程以及文章,欢迎大家前来阅读学习。

2023.07.31

1589

3

python教程
python教程

Python已成为一门网红语言,即使是在非编程开发者当中,也掀起了一股学习的热潮。本专题为大家带来python教程的相关文章,大家可以免费体验学习。

2023.08.03

21877

23

python环境变量的配置
python环境变量的配置

Python是一种流行的编程语言,被广泛用于软件开发、数据分析和科学计算等领域。在安装Python之后,我们需要配置环境变量,以便在任何位置都能够访问Python的可执行文件。php中文网给大家带来了相关的教程以及文章,欢迎大家前来学习阅读。

2023.08.04

2687

5

python eval
python eval

eval函数是Python中一个非常强大的函数,它可以将字符串作为Python代码进行执行,实现动态编程的效果。然而,由于其潜在的安全风险和性能问题,需要谨慎使用。php中文网给大家带来了相关的教程以及文章,欢迎大家前来学习阅读。

2023.08.04

2747

5

scratch和python区别
scratch和python区别

scratch和python的区别:1、scratch是一种专为初学者设计的图形化编程语言,python是一种文本编程语言;2、scratch使用的是基于积木的编程语法,python采用更加传统的文本编程语法等等。本专题为大家提供scratch和python相关的文章、下载、课程内容,供大家免费下载体验。

2023.08.11

1103

5

python合并两个列表
python合并两个列表

Python是一种强大的编程语言,具有许多方便的功能和工具。在Python中,有多种方法可以合并两个列表。php中文网给大家带来了相关的教程以及文章,欢迎大家前来学习阅读。

2023.08.10

596

4

python是前端还是后端
python是前端还是后端

Python属于前端也属于后端,其灵活性和丰富的生态系统使得开发人员能够在不同的领域中灵活运用。本专题为大家提供python相关的文章、下载、课程内容,供大家免费下载体验。

2023.08.11

2123

5

热门下载

更多
网站特效
/
网站源码
/
网站素材
/
前端模板

精品课程

更多
相关推荐
/
热门推荐
/
最新课程