Networkx节点边标注问题：字典转换报错及代码修正求助

错误原因 #

最常见的触发场景有两种：

1. 标签字典不完整：部分节点未设置label属性，导致nx.get_node_attributes返回的字典缺少部分节点的键值对，而nx.draw要求labels参数必须覆盖所有要显示标签的节点。
2. 参数格式错误：labels参数必须是{节点对象: 标签文本}的字典，若你传入的字典键不是图中存在的节点（比如节点是整数你用了字符串当键），也会触发ValueError。

另外，如果你用了$v_i$这种LaTeX格式标签但没开启matplotlib的LaTeX支持，虽然不会直接报ValueError，但标签会显示异常。

修正后的代码示例 #

import networkx as nx
import matplotlib.pyplot as plt

# 开启LaTeX渲染，确保$v_i$格式标签正常显示
plt.rcParams['text.usetex'] = True

# 自定义节点数量
node_count = 5
G = nx.Graph()

# 批量添加节点并设置label属性，确保无遗漏
for idx in range(1, node_count + 1):
    G.add_node(idx, label=f"$v_{idx}$")

# 正确生成节点标签字典
node_labels = nx.get_node_attributes(G, 'label')

# 批量添加边并设置label属性
for idx in range(1, node_count):
    G.add_edge(idx, idx + 1, label=f"$e_{idx}$")

# 绘制图形
pos = nx.spring_layout(G)
# 先绘制节点和边，不显示默认标签
nx.draw(G, pos, with_labels=False, node_size=500)
# 用自定义标签绘制节点标注
nx.draw_networkx_labels(G, pos, labels=node_labels, font_size=14)
# 绘制边标注
nx.draw_networkx_edge_labels(G, pos, edge_labels=nx.get_edge_attributes(G, 'label'))

plt.show()

关键注意点 #

• 确保所有节点都设置了label属性，避免循环时遗漏（比如从0开始计数但节点是从1开始的）。
• nx.draw_networkx_labels比直接在nx.draw中传labels参数更灵活，适合自定义标注场景。
• 必须开启text.usetex才能正确渲染LaTeX格式的上下标标签，否则$v_i$会被原样输出。

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