如何在pydot中使用带箭头的线条

Question

我使用pydot和这些命令绘制了一个二叉搜索树

graph = pydot.Dot(graph_type='graph')
visit(n5, BFS)
graph.write_png('example1_graph.png')

其中，函数视图遍历树并调用一个简单的例程来绘制边：

def draw(parent_name, child_name):
    # color for lines = red
    edge = pydot.Edge(parent_name, child_name, color="#ff0000")
    graph.add_edge(edge) 

但连接节点的线条是简单的直线。有没有办法将简单线条改为有向箭头？

Answer 1

要使用箭头连接节点，请使用有向图类型。更改代码的第一行，连接节点的行将是箭头而不是行。

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