如何在networkx.DiGraph中按自定义坐标字典设置节点位置并在绘图中生效

核心问题根因 #

• NetworkX 所有绘图接口都不会自动读取节点属性中存储的pos字段值，必须通过函数的pos参数显式传入位置字典才会生效，即便坐标已经正确写入节点元数据也不会被绘图流程自动获取。
• 使用nx.draw_shell这类绑定默认布局算法的接口时，如果没有显式传入pos参数，接口会强制调用自带的shell_layout算法计算节点位置，完全跳过已存储的节点坐标，这是渲染结果和自定义位置不符的直接原因。
• 即便切换到基础绘图函数nx.draw，缺少显式pos参数时，函数会自动触发spring_layout默认布局计算，同样不会使用已存储在节点属性中的坐标值。

正确实现流程 #

1. 完成图结构与节点坐标的基础写入（当前实现的这步无逻辑错误）：

   import networkx as nx
   import matplotlib.pyplot as plt
   
   # 初始化有向图
   G = nx.DiGraph()
   # 加载节点列表
   nodes = ['Start', 'S0', 'S1', 'S2', ...]
   G.add_nodes_from(nodes)
   
   # 自定义位置字典
   positions = {'Start': (2, 2), 'S0': (0, 0), ...}
   # 将坐标写入节点属性
   for node, pos in positions.items():
       G.nodes[node]['pos'] = pos
   

2. 渲染时显式传入自定义坐标：

   # 从节点属性中提取已存储的自定义坐标
   custom_pos = nx.get_node_attributes(G, 'pos')
   # 使用基础绘图接口，绑定提取好的坐标
   nx.draw(
       G,
       pos=custom_pos,
       with_labels=True,
       node_size=500
   )
   plt.show()
   

常见踩坑提醒 #

• 不要依赖节点上存储的pos属性自动生效，NetworkX 没有实现自动读取该属性用于绘图的逻辑，必须主动提取坐标字典并传入绘图函数的pos参数。
• 若需要保留自定义坐标的排布效果，优先使用基础nx.draw接口；如果用带内置布局的draw_*接口，必须确认显式传入自定义坐标，否则接口会自动触发内置布局算法覆盖自定义位置。
• 若手中已有原始位置字典，无需强制将坐标存入节点属性，可直接将字典作为pos参数传入绘图函数，存储节点属性只是坐标复用的可选实现，不是渲染自定义坐标的必需步骤。

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