Python JSON 到 anytree 树

Question

我有 JSON 数据，看起来像

{'Tree': [
  {"From": "1",
  "To": "2"},
  {"From": "2",
  "To": "3"}
]}

基本上是节点之间所有现有引用的数组。 我希望能够使用此数据从选定的根节点开始绘制树。

我发现使用 anytree 我必须将一个节点链接到父节点，但理想情况下我只想将它链接到名称并让它自己组合树结构。

# I want to do this
child[0] = Node(data[0]["To"], parent=data[0]["From"])

# But I think I need to assign a node, not its name
child[1] = Node(data[1]["To"], parent=child[?])

关于如何做到这一点的建议？

Answer 1

分两步执行此操作：

1. 将输入格式转换为邻接列表：由节点键作为键的字典，以及包含连接节点键的关联列表。

2. 选择一个根节点，然后使用邻接表创建任意树。

下面是处理这两个任务的两个函数：

from anytree import Node, RenderTree
from collections import defaultdict 

def makegraph(edges):
    adj = defaultdict(list)
    for edge in edges:
        a, b = edge.values()
        adj[a].append(b)
        adj[b].append(a)
    return adj

def maketree(adj, nodekey):
    visited = set()
    
    def dfs(nodekey, parent):
        if nodekey not in visited:
            visited.add(nodekey)
            node = Node(nodekey, parent)
            for childkey in adj[nodekey]:
                dfs(childkey, node)
            return node
                
    return dfs(nodekey, None)

下面是如何使用它：

edges = [
    {"From": "1", "To": "2"},
    {"From": "2", "To": "3"}
]

adj = makegraph(edges)
# Choose "1" as root:
root = maketree(adj, '1')
print(RenderTree(root))
# Now let's choose "2" as root:
root = maketree(adj, '2')
print(RenderTree(root))