python遍历列表结构为树的数据为树状结构

谁践踏了优雅 2023-07-09 02:14 16阅读 0赞

### python 遍历列表结构为树的数据为树状结构 ###

> 作者：lizhonglin  
> github: https://github.com/Leezhonglin/  
> blog: https://leezhonglin.github.io/

我们要把下面的数据解析层树状结构。我这里写了一个函数，方便自己使用。

例如：

test_data = [
        { 'id': 1, 'title': 'GGG', 'parent_id': 0},
        { 'id': 2, 'title': 'AAA', 'parent_id': 0},
        { 'id': 3, 'title': 'BBB', 'parent_id': 1},
        { 'id': 4, 'title': 'CCC', 'parent_id': 1},
        { 'id': 5, 'title': 'DDD', 'parent_id': 2},
        { 'id': 6, 'title': 'EEE', 'parent_id': 3},
        { 'id': 7, 'title': 'FFF', 'parent_id': 4},
    ]

#### 函数 ####

def tree(data, root, root_field, node_field):
        """ 解析list数据为树结构 :param data: 被解析的数据 :param root: 根节点值 :param root_field: 根节点字段 :param node_field: 节点字段 :return: list """
        l = []
        for i in data:
            if i.get(root_field) == root:
                l.append(i)
        for i in data:
            node = i.get(node_field)
            children = []
            for j in data:
                parent = j.get(root_field)
                if node == parent:
                    children.append(j)
            i['children'] = children
        return l

#### 实际测试结果 ####

print(tree(test_data, 0, 'parent_id', 'id'))

#### 运行完的结果 ####

[{ 
    	'id': 1,
    	'title': 'GGG',
    	'parent_id': 0,
    	'children': [{ 
    		'id': 3,
    		'title': 'BBB',
    		'parent_id': 1,
    		'children': [{ 
    			'id': 6,
    			'title': 'EEE',
    			'parent_id': 3,
    			'children': []
    		}]
    	}, { 
    		'id': 4,
    		'title': 'CCC',
    		'parent_id': 1,
    		'children': [{ 
    			'id': 7,
    			'title': 'FFF',
    			'parent_id': 4,
    			'children': []
    		}]
    	}]
    }, { 
    	'id': 2,
    	'title': 'AAA',
    	'parent_id': 0,
    	'children': [{ 
    		'id': 5,
    		'title': 'DDD',
    		'parent_id': 2,
    		'children': []
    	}]
    }]