Python 无限级分类树状结构生成算法的实现


Posted in Python onJanuary 21, 2021

后端研发的同学对无限级分类肯定映像深刻,当初花了不少时间吧?

无限级分类树状结构的应用场景很多,例如后端研发需要把用户相关权限读取出来并生成树状结构,前端研发拿到权限树之后可以按照结构展示用户有权限访问的栏目;再例如网页上的栏目分级:

Python 无限级分类树状结构生成算法的实现

作者在初次接触树状结构生成需求的时候,也是挠头,后来找到了一个代码少且清晰易懂的生成算法:递归。

首先,确保数据库中存储的类别信息如下:

[
 {"id": 1, "name": '电器', "parent": 0},
 {"id": 2, "name": '水果', "parent": 0},
 {"id": 3, "name": '家用电器', "parent": 1},
 {"id": 4, "name": '电吹风', "parent": 3},
 {"id": 5, "name": '电风扇', "parent": 3},
 {"id": 6, "name": '台灯', "parent": 3},
 {"id": 7, "name": '商用电器', "parent": 1},
 {"id": 8, "name": '大型电热锅', "parent": 7},
]

字段 parent 记录的是此条目的父编号,例如电吹风的父编号是 3,即电吹风属于家用电器,而家用电器的父编号是 1,即家用电器属于电器类产品。电吹风条目跟电器条目并无直接的标识进行关联,但需要用树状结构来表明 电器 <- 家用电器 <- 电吹风 的关系。

通过 parent 寻找父编号,并建立关联关系的操作实际上是循环往复的,直到找完所有的结点,这跟递归算法非常契合,很轻松便能写出对应的递归代码:

def generate_tree(source, parent):
 tree = []
 for item in source:
 if item["parent"] == parent:
 item["child"] = generate_tree(source, item["id"])
 tree.append(item)
 return tree

只需要将数据库中存储的信息传递给 generate_tree 函数即可。这段递归代码在往复循环的过程中通过 parent 来寻找子结点,找到子结点后将其添加到树中。完整代码如下:

import json
def generate_tree(source, parent):
 tree = []
 for item in source:
 if item["parent"] == parent:
 item["child"] = generate_tree(source, item["id"])
 tree.append(item)
 return tree
if __name__ == '__main__':
 permission_source = [
 {"id": 1, "name": '电器', "parent": 0},
 {"id": 2, "name": '水果', "parent": 0},
 {"id": 3, "name": '家用电器', "parent": 1},
 {"id": 4, "name": '电吹风', "parent": 2},
 {"id": 5, "name": '电风扇', "parent": 3},
 {"id": 6, "name": '台灯', "parent": 3},
 {"id": 7, "name": '商用电器', "parent": 1},
 {"id": 8, "name": '大型电热锅', "parent": 7},
 ]
 permission_tree = generate_tree(permission_source, 0)
 print(json.dumps(permission_tree, ensure_ascii=False))

你试试运行一下,看看结构是否符合预期。

使用缓存优化算法

递归算法中有很多重复的计算,这些计算不仅占用额外资源,还会降低函数执行效率,因此需要对递归进行优化。这里选用缓存优化法提升函数执行效率。

基本思路是每次找到结点关系后将此条目的编号添加到一个列表中缓存起来,代表此条目已找到结点关系。当往复循环执行函数时再次遇到此条目可以跳过。代码改动很简单,增加一个缓存列表和控制流语句即可:

def generate_tree(source, parent, cache=[]):
 tree = []
 for item in source:
 if item["id"] in cache:
 continue
 if item["parent"] == parent:
 cache.append(item["id"])
 item["child"] = generate_tree(source, item["id"], cache)
 tree.append(item)
 return tree

至此,无限级分类树状结构生成算法完成。你学会了吗?

到此这篇关于Python 无限级分类树状结构生成算法的实现的文章就介绍到这了,更多相关Python 无限级分类树状结构内容请搜索三水点靠木以前的文章或继续浏览下面的相关文章希望大家以后多多支持三水点靠木!

Python 相关文章推荐
改进Django中的表单的简单方法
Jul 17 Python
Python使用urllib2模块抓取HTML页面资源的实例分享
May 03 Python
浅谈python for循环的巧妙运用(迭代、列表生成式)
Sep 26 Python
python中字符串比较使用is、==和cmp()总结
Mar 18 Python
使用python Telnet远程登录执行程序的方法
Jan 26 Python
用Pelican搭建一个极简静态博客系统过程解析
Aug 22 Python
windows中安装Python3.8.0的实现方法
Nov 19 Python
python使用turtle库绘制奥运五环
Feb 24 Python
Pandas将列表(List)转换为数据框(Dataframe)
Apr 24 Python
在Python3.74+PyCharm2020.1 x64中安装使用Kivy的详细教程
Aug 07 Python
Python常用数字处理基本操作汇总
Sep 10 Python
python实现自动打卡的示例代码
Oct 10 Python
python 制作网站筛选工具(附源码)
Jan 21 #Python
python使用scapy模块实现ping扫描的过程详解
Jan 21 #Python
Python 中如何使用 virtualenv 管理虚拟环境
Jan 21 #Python
python使用scapy模块实现ARP扫描的过程
Jan 21 #Python
Python3利用scapy局域网实现自动多线程arp扫描功能
Jan 21 #Python
Pandas直接读取sql脚本的方法
Jan 21 #Python
python asyncio 协程库的使用
Jan 21 #Python
You might like
PHP 网络开发详解之远程文件包含漏洞
2010/04/25 PHP
深入理解PHP内核(一)
2015/11/10 PHP
PHP面向对象程序设计中的self、static、parent关键字用法分析
2019/08/14 PHP
javawscript 三级菜单的实现原理
2009/07/01 Javascript
extjs 学习笔记 四 带分页的grid
2009/10/20 Javascript
JQuery Study Notes 学习笔记(一)
2010/08/04 Javascript
网站404页面3秒后跳到首页的实例代码
2013/08/16 Javascript
js克隆对象、数组的常用方法介绍
2013/09/26 Javascript
一个仿糯米弹框效果demo
2014/07/22 Javascript
JS使用ajax方法获取指定url的head信息中指定字段值的方法
2015/03/24 Javascript
浅谈jQuery中setInterval()方法
2015/07/07 Javascript
字符串反转_JavaScript
2016/04/28 Javascript
JavaScript利用Date实现简单的倒计时实例
2017/01/12 Javascript
利用Angular+Angular-Ui实现分页(代码加简单)
2017/03/10 Javascript
vue中计算属性(computed)、methods和watched之间的区别
2017/07/27 Javascript
详解操作虚拟dom模拟react视图渲染
2018/07/25 Javascript
vue 动态绑定背景图片的方法
2018/08/10 Javascript
Three.js中矩阵和向量的使用教程
2019/03/19 Javascript
[00:12]DAC SOLO赛卫冕冠军 VG.Paparazi灬展现SOLO技巧
2018/04/06 DOTA
用Python实现QQ游戏大家来找茬辅助工具
2014/09/14 Python
跟老齐学Python之玩转字符串(3)
2014/09/14 Python
python结合API实现即时天气信息
2016/01/19 Python
Pandas 对Dataframe结构排序的实现方法
2018/04/10 Python
六行python代码的爱心曲线详解
2019/05/17 Python
numpy数组广播的机制
2019/07/12 Python
Python原始套接字编程实例解析
2020/01/29 Python
40行Python代码实现天气预报和每日鸡汤推送功能
2020/02/27 Python
win10安装python3.6的常见问题
2020/07/01 Python
浅谈优化Django ORM中的性能问题
2020/07/09 Python
用OpenCV进行年龄和性别检测的实现示例
2021/01/29 Python
纯CSS3编写的的精美动画进度条(无flash/无图像/无脚本/附源码)
2013/01/07 HTML / CSS
四风问题自查报告剖析材料
2014/02/08 职场文书
保安岗位职责
2014/02/21 职场文书
导游词之蜀山胜景瓦屋山
2019/11/29 职场文书
纯CSS实现酷炫的霓虹灯效果
2021/04/13 HTML / CSS
Golang 正则匹配效率详解
2021/04/25 Golang