NetworkX绘图中图像节点与坐标位置对齐问题求助

问题背景 #

参考StackOverflow方案实现以图像为节点的NetworkX网络图时，出现节点y坐标与图像y坐标无法对齐的问题。节点位置定义如下：

pos={
0: [0, 0], 
1: [1, 0], 
2: [2, 0], 
3: [3, 0], 
4: [4, 0], 
5: [5, 0], 
6: [6, 0], 
7: [5, 1]
}

原绘图代码如下：

pos:Dict={}
    for nodename in G.nodes:
        pos[nodename]=G.nodes[nodename]["pos"]
    print(f'pos={pos}')


    fig=plt.figure(figsize=(1,1))
    ax=plt.subplot(111)
    ax.set_aspect('equal')
    nx.draw_networkx_edges(G,pos,ax=ax)


    # plt.xlim(-1,10)
    # plt.ylim(-1.5,1.5)

    trans=ax.transData.transform
    trans2=fig.transFigure.inverted().transform

    piesize=0.3 # this is the image size
    p2=piesize/2.0
    for n in G:
        xx,yy=trans(pos[n]) # figure coordinates
        #print(f'xx,yy={xx,yy}')
        xa,ya=trans2((xx,yy)) # axes coordinates
        print(f'xa,ya={xa,ya}')
        a = plt.axes([xa-p2,ya-p2, piesize, piesize])
        a.set_aspect('equal')
        a.imshow(G.nodes[n]['image'])
        a.axis('off')
    ax.axis('off')
    plt.show()

问题原因 #

原代码通过plt.axes()创建独立子轴放置图像，子轴位置基于figure坐标系，而NetworkX的边绘制在主轴的数据坐标系中，两者坐标空间不统一，加上坐标变换过程中的误差，导致y轴对齐偏差。

解决方案 #

放弃独立子轴，直接在主轴的数据坐标系中绘制图像，通过extent参数精确指定图像位置，确保和节点坐标、边完全对齐。

修改后的完整代码 #

import matplotlib.pyplot as plt
import networkx as nx
from typing import Dict

# 假设G已定义，每个节点包含"pos"（坐标）和"image"（图像数组）属性
# 节点位置示例
pos = {
    0: [0, 0], 
    1: [1, 0], 
    2: [2, 0], 
    3: [3, 0], 
    4: [4, 0], 
    5: [5, 0], 
    6: [6, 0], 
    7: [5, 1]
}

# 从图节点属性提取位置（如果需要）
# pos: Dict = {}
# for nodename in G.nodes:
#     pos[nodename] = G.nodes[nodename]["pos"]

fig = plt.figure(figsize=(8, 4))  # 调整画布尺寸适配内容
ax = plt.subplot(111)
ax.set_aspect('equal')  # 保持坐标轴比例一致，避免图像变形

# 绘制NetworkX边
nx.draw_networkx_edges(G, pos, ax=ax)

piesize = 0.3  # 图像在数据坐标系中的边长
p2 = piesize / 2.0

# 在主轴上直接绘制图像，确保坐标对齐
for n in G:
    x, y = pos[n]
    # extent参数定义图像在数据坐标系中的范围：[x左边界, x右边界, y下边界, y上边界]
    extent = [x - p2, x + p2, y - p2, y + p2]
    # origin='upper'确保图像y轴方向和数据坐标系一致
    ax.imshow(G.nodes[n]['image'], extent=extent, origin='upper')

# 调整坐标轴范围，保证所有节点和边完整显示
x_min = min(p[0] for p in pos.values()) - p2 - 0.5
x_max = max(p[0] for p in pos.values()) + p2 + 0.5
y_min = min(p[1] for p in pos.values()) - p2 - 0.5
y_max = max(p[1] for p in pos.values()) + p2 + 0.5
ax.set_xlim(x_min, x_max)
ax.set_ylim(y_min, y_max)

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

关键修改说明 #

• 统一坐标空间：直接在主轴的data坐标系绘制图像，和NetworkX边的坐标系统完全一致，从根源避免对齐问题
• extent参数：精准控制图像在数据坐标系中的位置，确保图像中心与节点pos完全重合
• 显式坐标轴范围：避免自动缩放导致的内容截断，保证所有元素完整显示

内容的提问来源于stack exchange，提问作者a.t.