如何让3D曲面图与关联散点图同步并优化渲染效果

问题场景 #

在零件优化任务中，使用80个训练集点和20个测试集点绘制3D响应曲面：x轴为零件厚度t，y轴为直径d，z轴为轴承应力f_bru。通过curve_fit拟合得到曲面后，出现以下问题：

• 曲面渲染效果差，即使设置极低alpha值，透明度仍会随视角移动发生变化
• 曲面遮挡散点，无法清晰观察原始点的分布
• 对比其他响应曲面示例代码，未出现此类异常

复现代码 #

假设你的实现代码类似如下：

import numpy as np
import matplotlib.pyplot as plt
from scipy.optimize import curve_fit

# 模拟训练/测试数据
np.random.seed(42)
t_train = np.random.uniform(1, 5, 80)
d_train = np.random.uniform(10, 20, 80)
f_bru_train = 0.5*t_train**2 + 0.3*d_train + np.random.normal(0, 2, 80)
t_test = np.random.uniform(1, 5, 20)
d_test = np.random.uniform(10, 20, 20)
f_bru_test = 0.5*t_test**2 + 0.3*d_test + np.random.normal(0, 2, 20)

# 定义拟合函数
def fit_func(X, a, b, c):
    t, d = X
    return a*t**2 + b*d + c

# 拟合参数
popt, _ = curve_fit(fit_func, (t_train, d_train), f_bru_train)

# 生成曲面网格
t_grid = np.linspace(t_train.min(), t_train.max(), 50)
d_grid = np.linspace(d_train.min(), d_train.max(), 50)
T, D = np.meshgrid(t_grid, d_grid)
F_bru = fit_func((T, D), *popt)

# 绘制3D图
fig = plt.figure(figsize=(10,7))
ax = fig.add_subplot(111, projection='3d')

# 绘制拟合曲面
surf = ax.plot_surface(T, D, F_bru, alpha=0.3, cmap='viridis')

# 绘制训练/测试点
ax.scatter(t_train, d_train, f_bru_train, c='red', marker='o', label='训练集')
ax.scatter(t_test, d_test, f_bru_test, c='blue', marker='^', label='测试集')

ax.set_xlabel('厚度 t')
ax.set_ylabel('直径 d')
ax.set_zlabel('轴承应力 f_bru')
ax.legend()
plt.show()

问题原因 #

Matplotlib默认的3D曲面渲染中，depthshade=True会根据面片的深度调整阴影和透明度，导致视角变化时透明度出现不一致；同时，曲面面片的绘制顺序可能在视角切换时优先于散点，造成遮挡。

解决方案 #

针对上述问题，可通过以下调整修复：

1. 关闭深度阴影 #

在plot_surface中添加depthshade=False参数，强制曲面透明度不随深度变化：

surf = ax.plot_surface(T, D, F_bru, alpha=0.3, cmap='viridis', depthshade=False, antialiased=True)

2. 确保散点优先显示 #

显式调整绘制顺序，先绘制曲面再绘制散点，并给散点设置更高的zorder提升显示优先级：

# 先绘制曲面
surf = ax.plot_surface(T, D, F_bru, alpha=0.3, cmap='viridis', depthshade=False, antialiased=True)
# 再绘制散点，设置zorder确保在曲面上方
ax.scatter(t_train, d_train, f_bru_train, c='red', marker='o', label='训练集', zorder=10)
ax.scatter(t_test, d_test, f_bru_test, c='blue', marker='^', label='测试集', zorder=10)

3. 优化曲面面片数量 #

通过rstride和cstride增大面片间隔，减少曲面的面片数量，提升渲染稳定性：

surf = ax.plot_surface(T, D, F_bru, alpha=0.3, cmap='viridis', depthshade=False, antialiased=True, rstride=2, cstride=2)

修改后效果 #

调整后，曲面透明度不再随视角移动变化，散点始终清晰可见，可直观对比拟合曲面与原始数据点的分布差异。

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