如何从pydotplus或分层图绘制引擎提取节点位置？

核心思路 #

pydotplus调用Graphviz引擎（如dot）渲染图像时，会先计算节点布局坐标，但默认不会将坐标存入节点对象。我们可以通过获取布局后的DOT源码来提取这些坐标，无需依赖pygraphviz。

步骤1：获取带布局信息的DOT源码 #

调用create_dot()方法生成包含节点pos属性的DOT文本，这是Graphviz布局引擎计算后的结果：

# 假设你的pydotplus图对象为pdp_graph
dot_source = pdp_graph.create_dot().decode('utf-8')

步骤2：解析DOT源码提取坐标 #

用正则表达式匹配节点行中的pos属性，提取节点ID和对应的x、y坐标：

import re

node_coords = {}
# 正则匹配节点定义行，兼容带引号/不带引号的节点ID
pos_regex = re.compile(r'^\s*"?(\w+)"?\s+\[.*pos="([\d.]+),([\d.]+)!".*\];$', re.MULTILINE)

for match in pos_regex.finditer(dot_source):
    node_id = match.group(1)
    x = float(match.group(2))
    y = float(match.group(3))
    node_coords[node_id] = (x, y)

注：坐标单位为点（point），默认对应96dpi下的像素（1pt=1px），如果需要适配其他dpi可自行换算。

步骤3：映射到你的节点数据 #

如果你的DataFrame节点ID与pydotplus中的节点ID一致，直接通过字典映射即可：

# 假设节点DataFrame为nodes_df，含"node_id"列
nodes_df['x_coord'] = nodes_df['node_id'].apply(lambda id: node_coords[id][0] if id in node_coords else None)
nodes_df['y_coord'] = nodes_df['node_id'].apply(lambda id: node_coords[id][1] if id in node_coords else None)

注意事项 #

• 确保布局引擎一致：生成DOT和PNG时使用相同的引擎（默认是dot，对应分层布局），如果需要其他布局（如力导向），需指定prog参数，例如pdp_graph.create_dot(prog='neato')。
• 过滤无关节点：pydotplus可能自动生成一些辅助节点（如边缘的虚拟节点），这些节点的ID通常带有特殊字符，可根据实际情况过滤。

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