Python3中使用NetworkX为数据集节点（0-39）命名的方法咨询

原代码存在的问题 #

• Python3已移除xrange，改用range，原代码使用xrange会触发NameError。
• 需求是重命名现有40个节点，但代码错误添加了5000个新节点，完全偏离目标。
• 修改的是图G，但绘图用的是独立对象F，修改不会生效。

解决方案 #

使用NetworkX的relabel_nodes方法批量将原有节点重命名为0-39的整数标记，步骤如下：

1. 读取.net文件并转换为目标图类型（如有向图）。
2. 获取现有节点列表，创建“原节点→新整数标记”的映射字典。
3. 执行重命名操作，确保修改的是绘图用的图对象。
4. 绘图时开启标签显示，验证效果。

修正后的代码 #

import networkx as nx
import matplotlib.pyplot as plt

# 读取pajek格式的.net文件
G = nx.read_pajek('E:/Data_sets/football.net')
# 转换为有向图（根据需求选择是否保留）
F = nx.DiGraph(G)

# 获取所有现有节点的列表
original_nodes = list(F.nodes())
# 创建节点映射：原节点对应0到39的连续整数
node_mapping = {node: idx for idx, node in enumerate(original_nodes)}
# 重命名节点，copy=False表示直接修改原图对象
nx.relabel_nodes(F, node_mapping, copy=False)

# 输出图信息，验证节点数量和标记
print(nx.info(F))

# 绘制图形并显示节点标签
nx.draw(F, with_labels=True)
plt.show()

关键说明 #

• enumerate(original_nodes)自动生成0到39的索引，确保每个原有节点对应唯一的整数标记。
• relabel_nodes的copy=False参数避免创建新图，直接修改目标对象F。
• nx.draw添加with_labels=True才能在可视化图中显示0-39的节点标签。

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