如何将图边列表转换为D3.js可视化所需的嵌套层级字典？

这问题我之前做D3可视化时碰到过，刚好能匹配你需要的flare.json格式，给你一个清晰的实现思路和代码：

核心思路 #

我们需要把（子节点，父节点）的边列表转换成嵌套结构，关键是用字典快速映射每个节点的对象引用，避免重复创建，同时高效构建父子关系，最后定位根节点（没有父节点的节点）。

Python实现代码 #

def edges_to_hierarchy(edges):
    # 建立节点映射表，存储每个节点的对象，方便快速查找和修改
    node_map = {}
    # 收集所有子节点，用来后续定位根节点
    all_children = set()
    
    for child_name, parent_name in edges:
        all_children.add(child_name)
        
        # 如果父节点还没创建，先初始化节点对象
        if parent_name not in node_map:
            node_map[parent_name] = {"name": parent_name, "children": []}
        # 如果子节点还没创建，同样初始化
        if child_name not in node_map:
            node_map[child_name] = {"name": child_name, "children": []}
        
        # 将子节点对象添加到父节点的children数组中
        node_map[parent_name]["children"].append(node_map[child_name])
    
    # 找到根节点：所有节点中，没有出现在子节点集合里的那个就是根
    root_name = next(name for name in node_map if name not in all_children)
    return node_map[root_name]

# 测试你的输入
input_edges = [ ("a", "b"), ("c", "a"), ("d", "a"), ("e", "a"), ("f", "a"), ("g", "a"), ("h", "g") ]
hierarchy = edges_to_hierarchy(input_edges)

# 格式化输出，和你期望的结构完全一致
import json
print(json.dumps(hierarchy, indent=2))

运行结果 #

输出的结构和你需求的完全匹配：

{
  "name": "b",
  "children": [
    {
      "name": "a",
      "children": [
        {
          "name": "c",
          "children": []
        },
        {
          "name": "d",
          "children": []
        },
        {
          "name": "e",
          "children": []
        },
        {
          "name": "f",
          "children": []
        },
        {
          "name": "g",
          "children": [
            {
              "name": "h",
              "children": []
            }
          ]
        }
      ]
    }
  ]
}

（注：如果不需要空的children数组，可以在最后遍历结构时移除，不过D3的层级布局对空数组也兼容，不影响可视化）

注意事项 #

• 如果你的边列表可能存在重复边，可以在添加子节点前加个判断，避免重复添加：

  if node_map[child_name] not in node_map[parent_name]["children"]:
      node_map[parent_name]["children"].append(node_map[child_name])
  

• 如果存在多个根节点（即多个节点没有父节点），可以修改根节点查找逻辑，返回根节点列表：

  root_names = [name for name in node_map if name not in all_children]
  return [node_map[name] for name in root_names]
  

内容的提问来源于stack exchange，提问作者Maugrim