平级数据到树形结构转换的技巧

2024-02-11 21:32:06

前言

在实际开发中，我们经常会遇到这样的场景：从数据库中查询出的数据是平级结构，但前端页面需要以树形结构来展示。为了满足这一需求，我们需要将平级数据转换成树形结构。本文将详细介绍如何实现平级数据到树形结构的转换，并提供具体步骤和示例代码，帮助您轻松实现数据转换，提升开发效率。

什么是树形结构？

树形结构是一种层次结构，它由一个根节点和多个子节点组成。根节点是树结构的起点，子节点是根节点的子元素。子节点可以进一步拥有自己的子节点，以此类推。树形结构广泛应用于各种领域，例如文件系统、组织结构图、家谱等。

如何将平级数据转换成树形结构？

将平级数据转换成树形结构的过程主要分为以下几个步骤：

确定根节点 ：首先，我们需要确定树形结构的根节点。根节点是树结构的起点，它没有父节点。在平级数据中，根节点通常是具有最高层级或最父级ID的数据项。
建立父子关系 ：确定了根节点后，我们需要建立父子关系。父子关系是指一个节点是另一个节点的父节点，另一个节点是该节点的子节点。在平级数据中，父子关系通常是通过pid和id字段来体现的。pid字段表示父节点的id，id字段表示当前节点的id。
构建树形结构 ：建立了父子关系后，我们就可以构建树形结构了。我们可以使用递归或迭代的方式来构建树形结构。递归的方式是从根节点开始，逐层向下递归，将每个节点的子节点添加到该节点中。迭代的方式是从所有节点中找到根节点，然后逐层向下迭代，将每个节点的子节点添加到该节点中。

示例代码

def convert_flat_data_to_tree(data):
  """
  将平级数据转换成树形结构

  Args:
    data: 平级数据列表

  Returns:
    树形结构字典
  """

  # 确定根节点
  root_node = [item for item in data if item["pid"] == 0][0]

  # 建立父子关系
  for item in data:
    if item["id"] != root_node["id"]:
      for parent_item in data:
        if parent_item["id"] == item["pid"]:
          if "children" not in parent_item:
            parent_item["children"] = []
          parent_item["children"].append(item)

  # 构建树形结构
  tree = root_node

  return tree


if __name__ == "__main__":
  # 测试数据
  data = [
    {"id": 1, "pid": 0, "name": "根节点"},
    {"id": 2, "pid": 1, "name": "子节点1"},
    {"id": 3, "pid": 1, "name": "子节点2"},
    {"id": 4, "pid": 2, "name": "孙节点1"},
    {"id": 5, "pid": 2, "name": "孙节点2"},
    {"id": 6, "pid": 3, "name": "孙节点3"},
  ]

  # 将平级数据转换成树形结构
  tree = convert_flat_data_to_tree(data)

  # 打印树形结构
  print(tree)