如何在matplotlib 3D曲面图中实现与Z值关联的连续色图

问题解决方案 #

你遇到的色块拼接问题是plot_surface默认着色逻辑和当前网格分辨率低共同导致的：默认会将每四个网格顶点围成的面填充单一颜色，你的y轴仅3个值，Y方向网格非常稀疏，自然没有平滑过渡。不需要插值也可以先调整参数尝试，两种方案如下：

方案1：无需插值，调整plot_surface参数（最简洁）

给plot_surface加上shade=False参数，同时调高采样计数参数rcount、ccount，强制启用逐顶点的颜色插值：

surface = ax.plot_surface(X,Y,z, 
                          cmap = cmap, 
                          antialiased=True, 
                          edgecolor='gray', 
                          linewidth=0.2,
                          shade=False, # 关闭默认光照着色，避免影响颜色映射
                          rcount=200, # 行方向采样数，设为高于你y的点数即可
                          ccount=200) # 列方向采样数，设为高于你x的126即可

这个方案不改原始数据，仅修改参数就能实现颜色平滑过渡。如果调整后效果还达不到要求，再考虑插值方案。

方案2：网格插值提升分辨率

如果希望曲面本身也更平滑，不止是颜色过渡，可以对Z值做二维插值，生成更高密度的网格后再绘图：

import numpy as np
from scipy.interpolate import interp2d

# 原始网格坐标
x_ori = np.arange(0,126)
y_ori = np.arange(0,3)
# 生成更高密度的新网格
x_new = np.linspace(0,125, 500) # 点数可根据需要调整
y_new = np.linspace(0,2, 100)
# 构建插值函数
f = interp2d(x_ori, y_ori, z, kind='cubic') # 可选linear线性/cubic三次插值
z_new = f(x_new, y_new)
# 新网格绘图
X_new, Y_new = np.meshgrid(x_new, y_new)
surface = ax.plot_surface(X_new,Y_new,z_new, cmap = cmap, antialiased=True, edgecolor='none', linewidth=0)

插值后如果不需要原来的网格线可以把edgecolor设为none，避免线条太多影响观感。

额外优化点

你当前设置的ScalarMappable可以简化，不需要手动生成，直接把surface对象传给colorbar即可：

plt.colorbar(surface, ax=ax)

比手动新建ScalarMappable的写法更简洁，也不会出现映射不匹配的问题。

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