如何让PyGraphviz集群互相指向并实现正确渲染？

问题描述 #

我曾查阅过相关帖子，但按照Graphviz规范实现后渲染效果不符：本该从集群边缘出发指向节点的箭头，箭身在边缘处消失，仅节点处显示箭头。

我的NetworkX代码 #

G = nx.DiGraph(name='G')
G.add_nodes_from(['a', 'b', 'c', 'd', 'e', 'f', 'g', 'h'])
A = to_agraph(G, compound=True)

cluster0 = A.add_subgraph(['a', 'b', 'c', 'd'], name='cluster0')
cluster1 = A.add_subgraph(['e', 'f', 'g'], name='cluster1')

cluster0.add_edges_from([('a', 'b'), ('a', 'c'), ('b', 'd'), ('c', 'd')])
cluster1.add_edges_from([('e', 'g'), ('e', 'f')])

A.add_edges_from([('c', 'e'), ('b', 'f'), ('c', 'g'), ('d', 'h')])#, ('d', 'e'), ])
A.get_edge('c', 'e').attr.update({'ltail': 'cluster0'})

生成的Graphviz代码 #

strict digraph G {
    graph [compound=True,
        name=G
    ];
    subgraph cluster1 {
        e -> f;
        e -> g;
    }
    subgraph cluster0 {
        a -> b;
        a -> c;
        b -> d;
        c -> d;
    }
    b -> f;
    c -> e  [ltail=cluster0];
    c -> g;
    d -> h;
}

渲染代码 #

A.layout(prog='dot')
png_str = A.draw(format='png', prog='dot', args=f"-Gdpi={DPI}")
sio = BytesIO(png_str)
img = mpimg.imread(sio)
plt.imshow(img)
plt.axis('off')
plt.show()

渲染结果 #

• 加上ltail=cluster0属性：箭头异常（箭身在集群边缘处消失，仅目标节点显示箭头）
• 注释掉ltail=cluster0属性：箭头正常（完整显示从c到e的箭头）

错误原因及修正方法 #

核心错误是给已有节点间的边添加ltail属性——ltail/lhead是用来表示集群与节点/集群之间的连接，而非集群内节点和外部节点的连接。当你给c->e这条实际存在的节点间边加上ltail=cluster0，Graphviz会混淆两种连接逻辑，导致箭头渲染异常。

正确做法是创建一条虚拟的集群到节点的边，而非复用已有节点间的边：

修正后的NetworkX代码 #

G = nx.DiGraph(name='G')
G.add_nodes_from(['a', 'b', 'c', 'd', 'e', 'f', 'g', 'h'])
A = to_agraph(G, compound=True)

cluster0 = A.add_subgraph(['a', 'b', 'c', 'd'], name='cluster0')
cluster1 = A.add_subgraph(['e', 'f', 'g'], name='cluster1')

cluster0.add_edges_from([('a', 'b'), ('a', 'c'), ('b', 'd'), ('c', 'd')])
cluster1.add_edges_from([('e', 'g'), ('e', 'f')])

# 添加常规节点间边
A.add_edges_from([('b', 'f'), ('c', 'g'), ('d', 'h')])
# 创建集群到节点的关联边，指定ltail为集群名称
A.add_edge('cluster0', 'e', ltail='cluster0')
# 可选：如果不需要显示c到e的节点边，可设置为隐藏
# A.get_edge('c', 'e').attr['style'] = 'invis'

对应生成的Graphviz代码 #

strict digraph G {
    graph [compound=True,
        name=G
    ];
    subgraph cluster1 {
        e -> f;
        e -> g;
    }
    subgraph cluster0 {
        a -> b;
        a -> c;
        b -> d;
        c -> d;
    }
    b -> f;
    c -> g;
    d -> h;
    cluster0 -> e [ltail=cluster0];
}

这样渲染后，箭头会从cluster0的边缘出发，正常指向e节点。

嵌套集群内部边缘指向外部的实现示例 #

如果要实现嵌套集群（外层集群包含子集群）的边缘指向外部节点，只需在定义边时指定ltail为子集群名称即可：

嵌套集群示例代码 #

G = nx.DiGraph(name='G')
G.add_nodes_from(['a1', 'a2', 'b', 'c', 'd', 'e'])
A = to_agraph(G, compound=True)

# 外层集群
cluster0 = A.add_subgraph(['a1', 'a2', 'b', 'c'], name='cluster0')
# 嵌套子集群
cluster0_1 = A.add_subgraph(['a1', 'a2'], name='cluster0_1', parent=cluster0)
cluster1 = A.add_subgraph(['d', 'e'], name='cluster1')

# 添加内部边
cluster0_1.add_edges_from([('a1', 'a2')])
cluster0.add_edges_from([('b', 'c')])
cluster1.add_edges_from([('d', 'e')])

# 从嵌套子集群边缘指向外部节点
A.add_edge('cluster0_1', 'd', ltail='cluster0_1')
# 从外层集群边缘指向外部节点
A.add_edge('cluster0', 'e', ltail='cluster0')

对应Graphviz代码 #

strict digraph G {
    graph [compound=True,
        name=G
    ];
    subgraph cluster1 {
        d -> e;
    }
    subgraph cluster0 {
        b -> c;
        subgraph cluster0_1 {
            a1 -> a2;
        }
    }
    cluster0_1 -> d [ltail=cluster0_1];
    cluster0 -> e [ltail=cluster0];
}

渲染后可以看到：箭头从嵌套子集群cluster0_1的边缘指向d，从外层集群cluster0的边缘指向e，效果正常。

内容的提问来源于stack exchange，提问作者user23239455