Matplotlib绘制分段函数热力图出现白线问题及优化需求

问题本质 #

你遇到的白线是浮点精度误差导致的分界点网格错位：当网格步长缩小时，自动生成的x轴网格可能无法精确落到x=1处，绘图函数会在错位处出现空白或白线；另外，部分绘图函数（如imshow）的单元格边框默认开启，即使关闭坐标轴网格也无法消除。

方法1：精确对齐分界点的网格构建（彻底解决白线） #

直接生成包含x=1精确值的网格，避免浮点近似误差，配合pcolormesh绘制（支持精确网格对齐，且可关闭单元格边框）：

import numpy as np
import matplotlib.pyplot as plt

# 精确拆分区间生成网格，确保x=1为公共点
x_left = np.arange(0, 1.0001, 0.01)  # 左半区间到1（含）
x_right = np.arange(1, 2.0001, 0.01) # 右半区间从1（含）开始
x = np.unique(np.concatenate([x_left, x_right]))
y = np.arange(0, 1.0001, 0.01)
X, Y = np.meshgrid(x, y)

# 定义分段函数
Z = np.where(X <= 1, 1, Y)

# 绘制热力图，关闭单元格边框
plt.figure(figsize=(8, 4))
pcm = plt.pcolormesh(X, Y, Z, cmap='viridis', edgecolors='none')
plt.colorbar(pcm)
plt.xlabel('x')
plt.ylabel('y')
plt.title('精确分界热力图（无白线）')
plt.show()

关键细节 #

• 拆分区间生成x轴，确保x=1被精确包含，避免浮点近似导致的分界错位
• edgecolors='none'彻底关闭热力图单元格的边框线（区别于坐标轴网格）
• pcolormesh比imshow更适合需要精确坐标对齐的热力图场景

方法2：矢量渲染实现平滑过渡（无网格白线） #

如果需要颜色完全平滑的效果，使用contourf（填充等高线），它基于连续插值渲染，不会产生网格类白线，同时可精确绘制分界：

import numpy as np
import matplotlib.pyplot as plt

# 生成高密度采样点（无需精确对齐分界）
x = np.linspace(0, 2, 1000)
y = np.linspace(0, 1, 500)
X, Y = np.meshgrid(x, y)

Z = np.where(X <= 1, 1, Y)

plt.figure(figsize=(8, 4))
# 用足够多的levels实现平滑颜色过渡
contour = plt.contourf(X, Y, Z, levels=100, cmap='viridis', extend='neither')
plt.colorbar(contour)
plt.xlabel('x')
plt.ylabel('y')
plt.title('平滑过渡热力图')
# 精确绘制x=1的分界直线
plt.axvline(x=1, color='white', linewidth=1)
plt.show()

关键细节 #

• levels=100确保颜色过渡足够平滑
• axvline可精准标记x=1的分界，比网格对齐更直观
• 矢量渲染模式下导出的图片（如PDF、SVG）不会有像素类白线

避坑提示 #

• 不要用imshow绘制此类需要精确坐标的热力图，它会自动拉伸/插值，容易导致分界错位
• 避免直接用X == 1判断分界，可添加微小容差（如X <= 1 + 1e-8），但优先选择精确生成网格
• 若必须用imshow，需手动设置extent=[0,2,0,1]和origin='lower'，并关闭插值：interpolation='none'

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