如何移除matplotlib.pyplot.contourf外轮廓，实现类pcolormesh效果？

这个问题我之前也碰到过——contourf会默认把等高线延伸到坐标轴边界，哪怕数据边缘有未采样的空白区域，而pcolormesh只会老老实实绘制有有效数据的部分。要让两者效果一致，你可以这么做：

核心思路 #

• 先把未采样的区域标记为NaN（如果你的数据还没这么处理的话），让matplotlib识别出这是缺失值
• 调整contourf的参数，禁止它在有效数据范围外填充额外的颜色

修改后的代码示例 #

import matplotlib.pyplot as plt
import numpy as np

# 构造带未采样区域的示例数据（模拟边缘未采样）
x = np.linspace(-5, 5, 50)
y = np.linspace(-5, 5, 50)
X, Y = np.meshgrid(x, y)
zt = np.sin(np.sqrt(X**2 + Y**2))
# 将边缘区域设为NaN，模拟未采样场景
zt[:5, :] = np.nan
zt[-5:, :] = np.nan
zt[:, :5] = np.nan
zt[:, -5:] = np.nan

levels = np.linspace(-1, 1, 20)

plt.subplot(1,2, 1)
# 关键修改：添加extend='neither'，配合NaN识别缺失区域
plt.contourf(x, y, zt, levels=levels, extend='neither')
plt.title("contourf (fixed)")
plt.colorbar()

plt.subplot(1,2, 2)
plt.pcolormesh(x, y, zt)
plt.title("pcolormesh")
plt.colorbar()

plt.tight_layout()
plt.show()

参数解释 #

• extend='neither'：默认contourf会给超出levels最高/最低值的区域填充颜色（比如未采样的边缘会被填充成最高或最低层级的颜色），这个参数会关闭该行为，只在levels定义的有效范围内绘制等高线
• 标记未采样区域为NaN：matplotlib会自动跳过NaN区域的绘制，避免在空白区域生成不必要的边界轮廓

这样修改后，contourf的绘图效果就会和pcolormesh一致，不会再出现那些多余的外部边界着色了。

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