如何在Graphviz中按指定节点的BFS距离层级绘制双向图

要实现以AA为根节点、按BFS距离自动分层的垂直图，完全不用手动写rank=same，有两种可行方案：

方案1：用Python脚本自动生成带层级的Graphviz代码（精确分层） #

这是最准确的方法，先通过BFS计算每个节点到AA的层级，再自动给Graphviz代码添加分层规则：

1. 编写BFS计算脚本 #

把下面的代码存为bfs_rank.py：

import sys
from collections import deque

# 解析Graphviz代码，构建邻接表
def parse_graphviz(graph_code):
    adj = {}
    lines = [line.strip() for line in graph_code.split('\n') 
             if line.strip() and not line.startswith('strict graph') and not line.startswith('}')]
    for line in lines:
        u, vs = line.rstrip(';').split('--')
        u = u.strip()
        for v in vs.strip().strip('{}').split():
            v = v.strip()
            adj.setdefault(u, set()).add(v)
            adj.setdefault(v, set()).add(u)
    return adj

# BFS计算每个节点的层级（距离AA的步数）
def bfs_levels(adj, root):
    levels, visited = {}, set()
    q = deque([(root, 0)])
    visited.add(root)
    while q:
        node, level = q.popleft()
        levels[node] = level
        for neighbor in adj[node]:
            if neighbor not in visited:
                visited.add(neighbor)
                q.append((neighbor, level + 1))
    # 处理未连通节点，放到最后一层
    max_level = max(levels.values()) if levels else 0
    for node in adj:
        if node not in levels:
            levels[node] = max_level + 1
    return levels

# 生成带层级的Graphviz代码
def generate_ranked_graph(graph_code, levels):
    level_groups = {}
    for node, level in levels.items():
        level_groups.setdefault(level, []).append(node)
    
    output = []
    for line in graph_code.split('\n'):
        output.append(line)
        if line.strip() == 'strict graph {':
            # 明确设置垂直布局
            output.append('    rankdir=TB;')
            # 自动添加各层级的rank=same规则
            for level in sorted(level_groups.keys()):
                nodes = '; '.join(level_groups[level])
                output.append(f'    {{rank=same; {nodes};}}')
    return '\n'.join(output)

if __name__ == '__main__':
    graph_code = sys.stdin.read()
    adj = parse_graphviz(graph_code)
    levels = bfs_levels(adj, 'AA')
    print(generate_ranked_graph(graph_code, levels))

2. 运行脚本生成新的Graphviz文件 #

把你的原始Graphviz代码保存为original.dot，然后执行：

python bfs_rank.py < original.dot > ranked.dot

生成的ranked.dot会自动包含所有层级的分组，比如第0层是AA，第1层是JW、HX、MA、KO，完全不用手动写。

3. 渲染成图 #

用dot命令生成垂直图：

dot -Tpng ranked.dot -o final_graph.png

方案2：用Graphviz的neato布局（近似效果） #

如果不想写脚本，可以试试用neato的参数调整，让节点近似按距离AA的远近排列：
在你的原始Graphviz代码开头添加这些配置：

strict graph {
    rankdir=TB; // 垂直布局
    root=AA;    // 指定根节点
    distortion=-4; // 负数让节点向根聚集，可调整数值优化
    overlap=false;
    splines=true;
    // 你的原始边定义...
}

然后用neato渲染：

neato -Tpng original.dot -o approx_graph.png

这种方法是近似效果，层级划分的精度不如脚本方案，但胜在不用写代码。

注意事项 #

• 脚本方案是完全精确按BFS层级排序，完美匹配你的需求。
• 如果图中有和AA不连通的节点，脚本会自动把它们放到最后一层。
• rankdir=TB是Graphviz默认的垂直布局（从上到下），可以明确写出来确保效果。

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