Question

我在图中有多个独立的树。我想分别提取它们。我在用pydot作图。我要分别父1图和父2图。在我的用例中，树将随机生长（不是一一对应）。

graph = pydot.Dot(graph_type="digraph")

parent_node_1 = pydot.Node(name='Parent_1', style='filled', fillcolor='yellow')
parent_node_2 = pydot.Node(name='Parent_2', style='filled', fillcolor='yellow')

child_node_1 = pydot.Node(name='Child 1', style='filled', fillcolor='yellow')
child_node_2 = pydot.Node(name='Child 2', style='filled', fillcolor='yellow')

e1 = pydot.Edge('Parent_1', 'Child 1')
e2 = pydot.Edge('Parent_2', 'Child 2')

graph.add_node(parent_node_1)
graph.add_node(parent_node_2)

graph.add_node(child_node_1)
graph.add_node(child_node_2)

graph.add_edge(e1)
graph.add_edge(e2)

graph.write_png('dummy_graph.png')

Output of program

Answer 1

这在pydot中是一个麻烦的问题，因为您需要能够以一种简单的方式遍历图形。有可能，但我不建议这样做。下面的代码段是一个适用于您的测试用例的简单代码。我敢打赌，某处有一些错误，请谨慎使用。

def get_out_edges_and_children(g, node):
    out_edges = {e for e in g.get_edges() if e.get_source() == node.get_name()}
    children_names = {e.get_destination() for e in out_edges}
    children = [n for n in graph.get_nodes() if n.get_name() in children_names]
    return out_edges, children


all_node_names = {n.get_name() for n in graph.get_nodes()}
all_children_names = {e.get_destination() for e in graph.get_edges()}
all_roots = all_node_names - all_children_names #roots are children to noone

trees = []
for r in all_roots:
    nodes_to_process= graph.get_node(r)
    t = pydot.Dot()
    t.add_node(nodes_to_process[0])

    i = 0
    while i < len(nodes_to_process):
        current_node=nodes_to_process[i]
        edges,children = get_out_edges_and_children(graph,current_node)
        for c in children: t.add_node(c)
        for e in edges: t.add_edge(e)
        nodes_to_process += children
        i += 1

    trees.append(t)

请改用其他图形库以获得更强大的解决方案，例如networkx。它也可以import pydot objects，因此过渡应该很流畅！

在图形或森林中提取不同的树

1 个答案: