如何消除Matplotlib pcolormesh绘图中的可见网格边缘？

我使用Matplotlib的pcolormesh绘制高度图可视化，但即使设置了edgecolors='none'，输出图像中仍会出现网格边缘，形成类似线框的冗余效果，无法实现平滑无间断的表面可视化。

我已经尝试过以下操作，但问题依然存在：

• 添加rasterized=True参数
• 提高输出分辨率
• 调整配色方案（将配色设为50%不透明度时，网格会透过颜色显现）

附加信息 #

• Python版本：3.x
• Matplotlib版本：最新版

问题复现代码 #

以下是简化后的测试代码，放大图像边缘可观察到相同的网格问题：

import numpy as np
import matplotlib.pyplot as plt
from matplotlib.colors import LinearSegmentedColormap
from scipy.interpolate import griddata
from os.path import join

# 示例数据
x = np.random.rand(1000)
y = np.random.rand(1000) * 100  # 模拟高度数据
z = np.random.rand(1000)

# 转换为NumPy数组
x_np = np.array(x)
y_np = np.array(y)
z_np = np.array(z)

# 定义网格分辨率
heightmap_resolution = (512, 512)

# 创建网格
grid_x, grid_z = np.meshgrid(np.linspace(np.min(x_np), np.max(x_np), heightmap_resolution[0]),
                             np.linspace(np.min(z_np), np.max(z_np), heightmap_resolution[1]))

# 插值得到网格上的高度值
points = np.column_stack((x_np, z_np))
grid_y = griddata(points, y_np, (grid_x, grid_z), method='cubic')

# 绘图设置
fig_width, fig_height = 210 / 25.4, 297 / 25.4  # 英寸单位尺寸
dpi = 1440  # 高分辨率输出

fig = plt.figure(figsize=(fig_width, fig_height), dpi=dpi)
ax = fig.add_axes([0, 0, 1, 1])  # 占满整个画布
cmap = LinearSegmentedColormap.from_list('mycmap', ['white', 'black'])
c = ax.pcolormesh(grid_x, grid_z, grid_y, cmap=cmap, shading='gouraud', edgecolors='none', rasterized=True)
ax.axis('off')  # 关闭坐标轴

# 保存图像
output_dir = r"C:\test" # 修改为你的路径
heightmap_name = 'out_plot_height_map.png'
plt.savefig(join(output_dir, heightmap_name), format='png', dpi=dpi, bbox_inches='tight', pad_inches=0)
plt.close(fig)

已尝试的替代方案（存在缺陷） #

我试过用imshow绘制，但输出会出现像素化的阶梯效果，而该高度图要用于制造环节，这种效果无法接受。我需要保留shading='gouraud'带来的平滑渐变效果：

fig, ax = plt.subplots(figsize=(fig_width, fig_height), dpi=dpi)
im = ax.imshow(grid_y, extent=[np.min(x_np), np.max(x_np), np.min(z_np), np.max(z_np)], interpolation='bicubic', cmap='viridis', aspect='auto')
ax.axis('off')
plt.savefig(join(output_dir, heightmap_name), format='png', dpi=dpi, bbox_inches='tight', pad_inches=0)
plt.close(fig)

解决方案 #

1. 关闭抗锯齿消除边缘伪影

在pcolormesh中添加antialiased=False参数，避免抗锯齿算法导致的网格边缘渲染异常：

c = ax.pcolormesh(grid_x, grid_z, grid_y, cmap=cmap, shading='gouraud', edgecolors='none', rasterized=True, antialiased=False)

2. 优化PNG保存参数

保存图像时添加pil_kwargs={'optimize': True}，让PIL优化图像渲染，减少伪影：

plt.savefig(join(output_dir, heightmap_name), format='png', dpi=dpi, bbox_inches='tight', pad_inches=0, pil_kwargs={'optimize': True})

3. 备选方案：改用contourf生成平滑填充

如果pcolormesh的问题始终无法解决，可以尝试使用contourf，设置足够多的颜色层级来模拟gouraud shading的渐变效果：

levels = np.linspace(np.min(grid_y), np.max(grid_y), 1000)  # 用1000个层级实现平滑过渡
c = ax.contourf(grid_x, grid_z, grid_y, levels=levels, cmap=cmap, antialiased=True)

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