我正在使用python中的anytree包构建树,这些树旨在表示从树叶到树根的流。我有以下代码。
from anytree import Node
from anytree.exporter import DotExporter
A = Node('A')
B = Node('B', parent = A)
C = Node('C', parent = A)
DotExporter(A).to_picture('example.png')
它会生成下面的图像。
我想修改这个图像,使箭头指向相反的方向。我知道在graphviz中,将[dir=back]
添加到边缘定义行将会得到我想要的结果。通过运行以下代码:
for line in DotExporter(A):
print(line)
我得到了输出:
digraph tree {
"A";
"B";
"C";
"A" -> "B";
"A" -> "C";
}
但是,我如何修改anytree接口的DotExporter
输出,以将[dir=back]
添加到边缘定义行并反转箭头的方向?
发布于 2020-04-25 03:56:30
要修改DotExporter
为Dot
定义的边属性,需要更改DotExporter
的edgeattrfunc
属性。
from anytree import Node
from anytree.exporter import DotExporter
A = Node('A')
B = Node('B', parent = A)
C = Node('C', parent = A)
DotExporter(A, edgeattrfunc = lambda node, child: "dir=back").to_picture('example.png')
查看输出,您现在得到:
for line in DotExporter(A, edgeattrfunc = lambda node, child: "dir=back"):
print(line)
digraph tree {
"A";
"B";
"C";
"A" -> "B" [dir=back];
"A" -> "C" [dir=back];
}
这将生成以下图像。
https://stackoverflow.com/questions/61416377
复制相似问题