NetworkX边标签控制问题：节点侧链路ID需对应正确位置

我刚用NetworkX构建拓扑图，节点和边的绘制功能正常，但在边标签的精细控制上遇到问题。我希望给每条边的两端分别显示对应节点的唯一链路ID，且标签靠近各自的节点，但目前只能给整条边设置一个标签，无法实现这种拆分控制。

现有代码及问题 #

以下是我创建3节点3边拓扑的代码：

import networkx as nx
import matplotlib.pyplot as plt

node_def = ["N1", "N2", "N3"]
edgelist = [("N1", "N2"), 
            ("N1", "N3"), 
            ("N2", "N3")]

link_count = {}
G = nx.Graph(edgelist)

for node in G.nodes():
    link_count[node] = []

# 为每个节点生成链路ID
for edge in G.edges():
    for node in edge:
        link_count[node].append(len(link_count[node]) + 1)

print(link_count)
# 输出：{'N1': [1, 2], 'N2': [1, 2], 'N3': [1, 2]}

labels = {}
for edge in G.edges():
    labels[edge] = f"{link_count[edge[0]].pop(0)} | {link_count[edge[1]].pop(0)}"

print(labels)
# 输出：{('N1', 'N2'): '1 | 1', ('N1', 'N3'): '2 | 1', ('N2', 'N3'): '2 | 2'}

pos = nx.spring_layout(G)
nx.draw(G, pos, with_labels=True)
nx.draw_networkx_edge_labels(G, pos, edge_labels=labels, label_pos=0.5)
plt.show()

现有代码生成的标签是整条边的统一文本，但我需要的是：比如N1和N3的边，N1侧显示2（对应N1的链路ID），N3侧显示1（对应N3的链路ID），且两个标签分别靠近各自的节点。

解决方案：手动绘制两端独立标签 #

NetworkX的draw_networkx_edge_labels无法直接实现拆分标签，我们可以借助Matplotlib的文本绘制功能，手动计算每个标签的位置并单独绘制：

import networkx as nx
import matplotlib.pyplot as plt

node_def = ["N1", "N2", "N3"]
edgelist = [("N1", "N2"), 
            ("N1", "N3"), 
            ("N2", "N3")]

link_count = {}
G = nx.Graph(edgelist)

# 初始化每个节点的链路ID列表
for node in G.nodes():
    link_count[node] = []

# 为每个节点分配唯一的链路ID
for edge in G.edges():
    for node in edge:
        link_count[node].append(len(link_count[node]) + 1)

# 计算节点位置
pos = nx.spring_layout(G)

# 绘制图结构
nx.draw(G, pos, with_labels=True, node_size=700, node_color='lightblue')

# 定义标签偏移比例（越靠近0越靠近起点节点，越靠近1越靠近终点节点）
label_offset = 0.1

# 遍历每条边，绘制两端的链路ID标签
for edge in G.edges():
    u, v = edge
    # 获取u节点的链路ID并弹出（确保顺序对应）
    u_label = link_count[u].pop(0)
    # 获取v节点的链路ID并弹出
    v_label = link_count[v].pop(0)
    
    # 计算靠近u节点的标签位置
    u_pos_x = pos[u][0] * (1 - label_offset) + pos[v][0] * label_offset
    u_pos_y = pos[u][1] * (1 - label_offset) + pos[v][1] * label_offset
    plt.text(u_pos_x, u_pos_y, str(u_label), fontsize=12, ha='center', va='center', bbox=dict(facecolor='white', edgecolor='none', pad=2))
    
    # 计算靠近v节点的标签位置
    v_pos_x = pos[v][0] * (1 - label_offset) + pos[u][0] * label_offset
    v_pos_y = pos[v][1] * (1 - label_offset) + pos[u][1] * label_offset
    plt.text(v_pos_x, v_pos_y, str(v_label), fontsize=12, ha='center', va='center', bbox=dict(facecolor='white', edgecolor='none', pad=2))

plt.show()

关键说明 #

1. 链路ID分配：保持原逻辑，确保每个节点的链路ID按顺序递增且唯一。
2. 位置计算：通过label_offset控制标签与节点的距离，0.1表示标签位于从节点出发10%的边长度位置，调整该值可以改变标签的远近。
3. 文本绘制：使用plt.text()单独绘制每个标签，添加白色背景框避免与边重叠，提升可读性。

这样就能实现每条边两端的标签分别靠近对应节点，且每个节点的链路ID唯一显示的需求。

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