NetworkX与Matplotlib中矩形与边的交点计算问题求助

一、修正交点计算函数 #

之前的交点计算大概率是没考虑矩形宽高比例或向量方向判断错误，以下是精准的边缘交点计算逻辑：

1. 基于矩形中心、宽高，以及目标节点中心，计算方向向量
2. 分别计算向量到达矩形水平/垂直边缘的参数，取较小值确保交点落在边缘上
3. 避免节点重合时的除以0异常

def get_rect_edge_intersection(rect_center, rect_width, rect_height, target_center):
    x, y = rect_center
    tx, ty = target_center
    dx = tx - x
    dy = ty - y

    # 处理节点重合的极端情况
    if dx == 0 and dy == 0:
        return (x, y)

    # 计算到达水平/垂直边缘的参数t
    t_horizontal = (rect_width / 2) / abs(dx) if dx != 0 else float('inf')
    t_vertical = (rect_height / 2) / abs(dy) if dy != 0 else float('inf')

    # 取最小t，确保交点在矩形边缘
    t = min(t_horizontal, t_vertical)
    intersect_x = x + dx * t
    intersect_y = y + dy * t

    return (intersect_x, intersect_y)

二、完整绘图实现 #

结合NetworkX布局与Matplotlib手动绘制，实现边连接矩形边缘的效果：

import networkx as nx
import matplotlib.pyplot as plt
from matplotlib.patches import Rectangle

# 派系数据（字典格式）
faction_data = {
    "同盟国": ["美国", "英国", "法国"],
    "轴心国": ["德国", "意大利", "日本"],
    "中立国": ["瑞士", "瑞典"]
}

# 创建图结构
G = nx.Graph()
G.add_nodes_from(faction_data.keys())
G.add_edges_from([("同盟国", "轴心国"), ("同盟国", "中立国")])

# 固定布局（seed保证每次运行布局一致）
pos = nx.spring_layout(G, seed=42)

# 矩形节点的宽高配置
rect_width = 0.2
rect_height = 0.15

# 初始化画布
fig, ax = plt.subplots(figsize=(8, 6))

# 绘制矩形节点
for node in G.nodes():
    x, y = pos[node]
    # 矩形左下角坐标 = 中心坐标 - 宽高/2
    rect = Rectangle((x - rect_width/2, y - rect_height/2), rect_width, rect_height,
                     facecolor='lightblue', edgecolor='black')
    ax.add_patch(rect)
    # 添加派系名称标签
    ax.text(x, y, node, ha='center', va='center', fontsize=12)

# 绘制连接矩形边缘的边
for u, v in G.edges():
    u_center = pos[u]
    v_center = pos[v]
    # 计算两端矩形的边缘交点
    u_edge = get_rect_edge_intersection(u_center, rect_width, rect_height, v_center)
    v_edge = get_rect_edge_intersection(v_center, rect_width, rect_height, u_center)
    # 手动绘制边
    ax.plot([u_edge[0], v_edge[0]], [u_edge[1], v_edge[1]], color='gray', linewidth=2)

# 调整画布范围，隐藏坐标轴
ax.set_xlim(min(pos[n][0] for n in G.nodes()) - 0.3, max(pos[n][0] for n in G.nodes()) + 0.3)
ax.set_ylim(min(pos[n][1] for n in G.nodes()) - 0.3, max(pos[n][1] for n in G.nodes()) + 0.3)
ax.axis('off')

plt.show()

三、更优实现思路 #

• 自定义布局：放弃自动布局，手动指定pos字典的坐标（比如把派系按阵营排列成一行），更符合关系图的展示逻辑
• 批量处理：将边的交点计算、绘制逻辑封装成函数，减少重复代码
• 样式区分：给不同关系的边设置不同颜色/线型（比如敌对边用红色虚线，同盟边用蓝色实线）

四、替代绘制库推荐 #

如果自定义成本过高，这些库更适合此类需求：

• Graphviz：通过pygraphviz或graphviz调用，支持节点形状自定义（矩形、椭圆等），自动处理边连接到节点边缘，布局算法专业（dot布局适合层次关系图），无需手动计算交点
• Pyvis：交互式可视化库，支持节点样式自定义，边自动连接边缘，生成的HTML可拖拽缩放，适合复杂关系展示
• Plotly：支持交互式网络图绘制，节点形状可选矩形，边可设置连接到边缘，自带hover提示等功能

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