如何在NetworkX绘图中显示边权重且避免图面杂乱

解决方案 #

你遇到的标签杂乱问题主要来自两个基础错误+默认显示参数不合理：

1. 你在初始化空图时就计算了布局pos，此时图中没有任何节点边，生成的坐标是无效的，后续加完元素后标签位置完全错乱
2. 边标签默认无背景、字体偏大，容易和节点、边重叠

修正后可直接运行的代码

import networkx as nx
import matplotlib.pyplot as plt

# 1. 初始化图、加载数据
my_graph = nx.Graph()
edges = nx.read_weighted_edgelist('edges.txt')
nodes = nx.read_adjlist("nodes.txt")
my_graph.add_edges_from(edges.edges(data=True)) # 要带data=True才会读入权重属性
my_graph.add_nodes_from(nodes)

# 2. 加完所有元素后再计算布局
pos = nx.shell_layout(my_graph)
# 如果shell布局太挤，可以换用spring布局调整间距：
# pos = nx.spring_layout(my_graph, k=0.8, seed=42) # seed固定布局避免每次跑都变

# 3. 提前设置画布大小，画布越大越不容易杂乱
plt.figure(figsize=(12, 10))

# 4. 绘制基础图，可绑定边粗到权重，视觉更直观
edge_width = [my_graph.edges[e]['weight'] * 0.4 for e in my_graph.edges] # 系数根据权重范围调整
nx.draw(
    my_graph, pos,
    with_labels=True,
    node_size=700,
    font_size=10,
    width=edge_width,
    edge_color='#aaaaaa' # 边调浅一点，突出标签
)

# 5. 绘制优化后的边权重标签
edge_labels = {e: round(my_graph.edges[e]['weight'], 2) for e in my_graph.edges} # 小数保留2位减少标签长度
nx.draw_networkx_edge_labels(
    my_graph, pos,
    edge_labels=edge_labels,
    font_size=8,
    label_pos=0.4, # 标签放在边40%的位置，避免和节点挨太近
    bbox=dict(facecolor='white', edgecolor='none', pad=0.2) # 加白色背景不被边挡住
)

plt.show()
plt.close()

可选优化调整

• 如果节点数量较多，可以把node_size、font_size再调小，同时放大画布尺寸
• 权重数值范围过大的话，可以用对数映射到边宽，避免高权重边过粗
• 需要固定布局每次生成效果一致的话，给布局函数加seed参数即可

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