Python交互式imshow绘图异常：坐标不匹配、自动重置问题求助

以下是修复后的完整代码：

import numpy as np
import matplotlib.pyplot as plt

N = 3

# 初始化全黑网格
data = np.zeros((N, N)) 

fig, ax = plt.subplots(1, 1, tight_layout=True)

# 创建图像对象并保存引用，后续直接更新数据
box = ax.imshow(data, cmap='gray', extent=[0, N, 0, N])

# 绘制网格线
for x in range(N + 1):
    ax.axhline(x, lw=2, color='w', zorder=5)
    ax.axvline(x, lw=2, color='w', zorder=5)

def on_click(event):
    # 忽略坐标轴外的点击
    if event.xdata is None or event.ydata is None:
        return
    
    gx = event.xdata
    gy = event.ydata
    
    # 修正坐标到数组索引的转换：解决垂直反转问题
    # 图像y轴从下到上是0到N，数组索引从上到下是0到N-1，所以反转计算
    i = N - 1 - int(gy)
    j = int(gx)
    
    # 确保索引在有效范围内（防止点击边界时越界）
    if 0 <= i < N and 0 <= j < N:
        data[i, j] = 1
        # 更新已有图像的数据，而非重新绘制imshow，解决自动重置问题
        box.set_data(data)
        fig.canvas.draw_idle()

fig.canvas.mpl_connect('button_press_event', on_click)

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

关键修复点 #

1. 坐标索引转换修正 #

Matplotlib的imshow默认y轴方向是从下到上，而数组data的索引是从上到下排列的。原来的i = int(gy) - N + 1计算逻辑错误，改为i = N - 1 - int(gy)：

• 点击顶部区域（gy接近N）时，int(gy)为N-1，计算得i=0，对应数组第一行（顶部行）
• 点击底部区域（gy接近0）时，int(gy)为0，计算得i=N-1，对应数组最后一行（底部行）

同时增加了索引范围判断，避免点击网格边界时出现数组越界错误。

2. 避免图像自动重置 #

原来每次点击都调用ax.imshow(data, ...)，这会重新创建新的图像层，覆盖原有内容，甚至在数据全为1时触发意外重置。改为直接更新已创建的box对象的数据：

box.set_data(data)

这种方式仅更新图像数据，保留原有绘图配置，彻底解决自动重置问题。

内容的提问来源于stack exchange，提问作者Chris Ze Third