我一直在寻找一段时间来在python中的graphviz中指定图形中的节点位置。我在dot中找到了rank command的子图,这就是我要找的东西,但我找不到在python中将graphgraph中的子图和等级组合起来的方法。我也试图强制节点位置,但也没有工作。我创建了一个我想要实现的简单示例。
这是我的代码:
from graphviz import Digraph
top_nodes = ['a', 'b', 'c']
other_nodes = ['d', 'e', 'f', 'g', 'm', 'n']
g = Digraph('test', format='png')
for n in top_nodes:
g.node(str(n), color='red')
for n in other_nodes:
g.node(str(n))
g.edge('a', 'd')
g.edge('d', 'g')
g.edge('g', 'm')
g.edge('m', 'n')
g.edge('b', 'e')
g.edge('b', 'f')
g.edge('e', 'n')
g.edge('c', 'f')
g.view()
这是输出:
我希望红色节点(“源”)位于同一级别的图形顶部,只要保留分层布局,其余节点的位置就不那么重要了。
答案 0 :(得分:4)
我遇到了同样的问题,发现使用子图解决了这个问题。
from graphviz import Digraph
top_nodes = ['a', 'b', 'c']
other_nodes = ['d', 'e', 'f', 'g', 'm', 'n']
g = Digraph('test', format='png')
s = Digraph('subgraph')
s.graph_attr.update(rank='min')
for n in top_nodes:
s.node(str(n), color='red')
for n in other_nodes:
g.node(str(n))
g.edge('a', 'd')
g.edge('d', 'g')
g.edge('g', 'm')
g.edge('m', 'n')
g.edge('b', 'e')
g.edge('b', 'f')
g.edge('e', 'n')
g.edge('c', 'f')
g.subgraph(s)
g.view()
答案 1 :(得分:2)
从代码开始,使用子图 context 就足够了,只需添加
with g.subgraph() as s:
s.attr(rank = 'same')
for n in top_nodes: s.node(n)
似乎比显式声明一个子图并将其添加到原始图要容易。
顺便说一下,此解决方案生成的@DOT源与@RaHo答案中的ine完全相同。
digraph test {
a [color=red]
b [color=red]
c [color=red]
d
e
f
g
m
n
a -> d
d -> g
g -> m
m -> n
b -> e
b -> f
e -> n
c -> f
{
rank=same
a
b
c
}
}
答案 2 :(得分:1)