如何为nx.Graph()节点分配pos属性，调用nx.draw()时默认使用且可随图存储？

1. 为已存在节点批量添加pos属性 #

直接操作NetworkX图的节点属性字典即可，你可以先用内置布局算法生成坐标，也可以自定义坐标字典后批量赋值：

import networkx as nx
import numpy as np

# 示例：已有现成的图结构
G = nx.Graph()
G.add_edges_from([(1,2), (2,3), (3,1), (1,4)])

# 方式1：用内置布局算法生成坐标
pos = nx.spring_layout(G, seed=42) # 加seed固定布局
# 批量赋值pos属性，转成列表方便后续JSON序列化
for n, coord in pos.items():
    G.nodes[n]['pos'] = coord.tolist()

# 方式2：自定义坐标手动赋值
# custom_pos = {1: [0,0], 2: [1,0], 3: [0.5, 1], 4: [-0.5, 0.5]}
# for n, coord in custom_pos.items():
#     G.nodes[n]['pos'] = coord

2. 绘图时直接调用pos属性 #

不需要单独维护pos变量，每次绘图直接从节点属性提取坐标即可：

# 提取节点的pos属性生成绘图用的坐标字典
draw_pos = {n: G.nodes[n]['pos'] for n in G.nodes}
nx.draw(G, pos=draw_pos, with_labels=True, node_color='lightblue')

如果需要简化调用，可以自行封装一个一行调用的绘图工具函数，实现默认使用节点pos属性的效果。

3. 带pos属性的图存JSON方案 #

用NetworkX自带的JSON序列化工具即可把图结构和所有节点属性（包括pos）一起存储，无需单独管理位置文件：

import json
from networkx.readwrite import json_graph

# 序列化存储
graph_json = json_graph.node_link_data(G)
with open("graph_with_pos.json", "w", encoding="utf-8") as f:
    json.dump(graph_json, f, indent=2, ensure_ascii=False)

# 读取还原
with open("graph_with_pos.json", "r", encoding="utf-8") as f:
    load_data = json.load(f)
G_new = json_graph.node_link_graph(load_data)
# 还原后的图自带所有pos属性，可直接提取绘图

注意事项 #

• 存储JSON前必须把numpy格式的坐标转成Python原生列表，否则会触发序列化报错
• 手动指定坐标或者用布局算法生成坐标的逻辑互相兼容，两种方式生成的pos属性读取规则完全一致
• 生成布局时添加seed参数可保证每次生成的坐标固定，避免布局随机变化

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