如何修正imshow的x/y轴刻度范围？兼析z数组索引问题

一、修正坐标轴刻度为-3到3 #

你的问题出在contour函数未使用实际坐标范围——你给imshow设置了extent=[-3,3,-3,3]映射正确坐标，但contour默认用数组索引（0到59）作为坐标值。两种简单修复方式：

方法1：给contour传入x、y数组 #

直接将x、y数组传给contour，让它基于实际坐标绘制等高线：

# 替换原contour代码行
cset = contour(x, y, z, arange(-1,1.5,0.2), linewidths=2, cmap=cm.Set2)

方法2：给contour添加extent参数 #

和imshow保持一致的范围设置：

cset = contour(z, arange(-1,1.5,0.2), linewidths=2, cmap=cm.Set2, extent=[-3,3,-3,3])

额外优化：建议用numpy.meshgrid替代手动嵌套循环生成z数组，代码更简洁高效：

X, Y = meshgrid(x, y)
z = z_func(X, Y)

这样无需手动处理索引转置，meshgrid会自动生成对应维度的网格矩阵，直接代入函数计算即可。

二、关于索引转置的疑问 #

需要写z[j][i] = z_func(x[i], y[j])是因为matplotlib对二维数组的维度映射规则：

• 二维数组的第一个维度（行）对应图像的y轴（纵向）
• 第二个维度（列）对应图像的x轴**（横向）**

你的循环中，i遍历x的索引（横轴方向），j遍历y的索引（纵轴方向）。如果直接写z[i][j]，会把x索引对应到行（y轴）、y索引对应到列（x轴），导致图像错位旋转。而z[j][i]交换了行和列的索引，让y索引对应行、x索引对应列，匹配matplotlib的坐标映射规则，确保图像的x/y轴和函数变量对应一致。

加上你设置了origin='lower'，y轴起始点在图像底部，和数学坐标系y轴方向一致，此时行索引从小到大对应y值从-3到3，进一步保证坐标对应正确。

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