python生成树

本文主要是介绍python生成树，对大家解决编程问题具有一定的参考价值，需要的程序猿们随着小编来一起学习吧！

def convert2tree(data):
    result = []
    obj = {}
    for x in data:
        obj[x.get("id", "")] = x
    for x in data:
        parent_id = x.get("parent_id")
        if parent_id:
            parent = obj.get(parent_id, {})
            if parent:
                if not parent.get("children"):
                    parent["children"] = []
                if not x.get("children"):
                    x["children"] = []
                parent["children"].append(x)
            else:
                if not x.get("children"):
                    x["children"] = []
                result.append(x)
        else:
            if not x.get("children"):
                x["children"] = []
            result.append(x)
    return result


resources = [{"id": 7, "name": "test1", "parent_id": 2},
             {"id": 2, "name": "test2", "parent_id": None},
             {"id": "9", "name": "test3", "parent_id": 2},
             {"id": 12, "name": "test4", "parent_id": 7},
             {"id": 8, "name": "test5", "parent_id": 2}]

print(convert2tree(resources))

这篇关于python生成树的文章就介绍到这儿，希望我们推荐的文章对大家有所帮助，也希望大家多多支持为之网！