Как использовать линии со стрелкой в ​​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')