节点尺寸不同时nx.draw_networkx_edges箭头无法连接节点的解决办法

当使用nx.draw_networkx_edges时，该函数仅支持设置全局单一node_size值，无法适配不同大小的节点，导致箭头无法准确连接到节点边界。解决思路是手动计算每条边的端点偏移量，让边的起点和终点分别落在源节点和目标节点的边界上，具体步骤如下：

实现步骤 #

1. 先绘制节点和标签，保留节点位置与尺寸数据
2. 根据每个节点的尺寸计算对应的半径（节点尺寸为面积，半径可通过sqrt(node_size)/2近似，或用sqrt(node_size/np.pi)得到精确值）
3. 对每条边，计算节点间的向量与单位向量，调整边的起点（从源节点中心向外偏移源节点半径）和终点（从目标节点中心向内偏移目标节点半径）
4. 用matplotlib.pyplot.arrow绘制调整后的带箭头边

代码示例 #

import networkx as nx
import matplotlib.pyplot as plt
import numpy as np

# 创建示例有向图
G = nx.DiGraph()
G.add_edges_from([(1, 2), (2, 3), (3, 1)])

# 节点布局与自定义尺寸
pos = nx.spring_layout(G)
node_sizes = {1: 1000, 2: 3000, 3: 500}

# 绘制节点与标签
nx.draw_networkx_nodes(G, pos, node_size=[node_sizes[n] for n in G.nodes()])
nx.draw_networkx_labels(G, pos)

# 计算每个节点的半径（近似值，适配matplotlib的marker面积规则）
node_radii = {n: np.sqrt(node_sizes[n]) / 2 for n in G.nodes()}

# 遍历每条边，调整端点位置并绘制
for u, v in G.edges():
    u_x, u_y = pos[u]
    v_x, v_y = pos[v]
    
    # 计算节点间的向量与长度
    dx = v_x - u_x
    dy = v_y - u_y
    edge_length = np.sqrt(dx ** 2 + dy ** 2)
    
    if edge_length == 0:
        continue  # 跳过重合节点
    
    # 单位向量
    unit_dx = dx / edge_length
    unit_dy = dy / edge_length
    
    # 调整边的起点和终点到节点边界
    start_x = u_x + unit_dx * node_radii[u]
    start_y = u_y + unit_dy * node_radii[u]
    end_x = v_x - unit_dx * node_radii[v]
    end_y = v_y - unit_dy * node_radii[v]
    
    # 绘制带箭头的边，可自定义箭头样式
    plt.arrow(start_x, start_y, end_x - start_x, end_y - start_y,
              head_width=0.03, head_length=0.05, fc='black', ec='black')

plt.axis('off')
plt.show()

这个方法能精准适配不同尺寸的节点，让箭头始终连接到节点边界，替代nx.draw_networkx_edges的全局设置限制。

内容的提问来源于stack exchange，提问作者Łukasz Zaroda