如何让Matplotlib 3D曲面图的网格线仅显示可见部分？

问题描述 #

想要绘制带有辅助线的plot_surface，现有代码通过叠加plot_surface和plot_wireframe实现，但运行后曲面背面的网格线也会显示，调整alpha参数无法解决，多曲面绘制时问题更严重，需要让网格线仅显示视角下的可见部分。

解决方案 #

不需要单独调用plot_wireframe，直接在plot_surface中设置edgecolor参数添加辅助线即可，Matplotlib会自动处理深度测试，只保留视角可见的边缘线。

修改后的代码 #

import numpy as np
import matplotlib.pyplot as plt

fig = plt.figure()
ax = fig.add_subplot(111, projection="3d")
X, Y = np.mgrid[-1:1:30j, -1:1:30j]
Z = np.sin(np.pi*X)*np.sin(np.pi*Y)

# 直接在plot_surface中配置边缘线，自动处理可见性
ax.plot_surface(X, Y, Z, cmap="autumn_r", rstride=1, cstride=1, edgecolor='black', linewidth=0.1)

plt.show()

关键说明 #

• edgecolor参数用于定义曲面每个小面片的边缘颜色，这些边缘线会遵循3D绘图的深度排序规则，只会显示当前视角下未被遮挡的部分，彻底避免背面网格线乱入的问题。
• 若要调整网格密度，修改rstride（行步长）和cstride（列步长）即可，数值越小网格越密；linewidth则用于控制辅助线的粗细。

如果需要更定制化的网格样式（比如只显示横向或纵向辅助线），可以单独调整rstride/cstride为较大值，配合rcount/ccount控制采样数，核心思路始终是利用plot_surface本身的边缘渲染逻辑，而非单独绘制独立的线框。

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