使用NetworkX绘制DFA有向多重图时如何避免边重叠？

我之前做DFA可视化的时候也踩过这个坑，NetworkX默认的spring布局对有向多重图的支持确实不太理想，给你几个实用的调整方案，能有效解决边重叠和布局混乱的问题：

1. 改用Graphviz驱动的布局算法 #

spring_layout是基于力导向的，对有向图的层次结构不友好，推荐用Graphviz的dot布局，它专门为有向图优化，能生成更规整的层次结构。不过需要先安装pygraphviz库（依赖Graphviz本体，得先装Graphviz软件再装Python包）。

修改布局部分的代码：

# 替换原来的spring_layout
pos = nx.nx_agraph.graphviz_layout(G, prog='dot')

prog='dot'会让节点按逻辑层级排列，非常适合DFA这类状态机的可视化。

2. 给重叠边添加弧形偏移 #

如果还是有少量边重叠，可以通过设置边的连接样式，让重叠的边以不同弧度的曲线展示，避免直线重叠。需要单独绘制边并指定connectionstyle参数：

# 先绘制节点
nx.draw(G, pos, with_labels=True, arrows=True, node_size=1500, font_size=12)
# 绘制边，设置弧形连接
nx.draw_networkx_edges(G, pos, connectionstyle="arc3,rad=0.2", arrows=True)
# 绘制边标签
nx.draw_networkx_edge_labels(G, pos, edge_labels=nx.get_edge_attributes(G, "label"), font_size=10)

rad=0.2控制弧形的弯曲程度，你可以根据图的大小调整这个值，比如边多的话用0.3或0.4。

3. 整合后的完整plot方法 #

把上面的优化点整合到你的代码里，最终的plot函数应该是这样：

def plot(self): 
    G = nx.MultiDiGraph() 
    nodes = self.collect_nodes() 
    for n in nodes: 
        G.add_node(n.id) 
    for n in nodes: 
        for naar, label in n.get_edges(): 
            G.add_edge(n.id, naar, label=label) 
    
    # 使用Graphviz的dot布局
    pos = nx.nx_agraph.graphviz_layout(G, prog='dot') 
    
    # 分步绘制，精细控制样式
    nx.draw(G, pos, with_labels=True, arrows=True, node_size=1500, font_size=12, node_color='lightblue')
    nx.draw_networkx_edges(G, pos, connectionstyle="arc3,rad=0.2", arrows=True, edge_color='gray')
    nx.draw_networkx_edge_labels(G, pos, edge_labels=nx.get_edge_attributes(G, "label"), font_size=10, label_pos=0.3)
    
    plt.show()

额外小技巧 #

• 如果你的DFA有起始状态或接受状态，可以给这些节点设置不同的颜色或形状，比如接受状态用双圈：在add_node的时候加上shape='doublecircle'属性，然后绘制节点时指定node_shape参数。
• 调整label_pos=0.3可以让边标签更靠近起点或终点，避免和边重叠。

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