本文详解如何通过合理设置坐标轴范围与 nan 掩码协同使用，避免因自动缩放导致的圆柱几何形变，实现对圆柱表面任意角度区段（如仅保留前25°）的干净、无失真遮蔽。
本文详解如何通过合理设置坐标轴范围与 nan 掩码协同使用，避免因自动缩放导致的圆柱几何形变，实现对圆柱表面任意角度区段（如仅保留前25°）的干净、无失真遮蔽。
在 Plotly 中绘制三维圆柱面时，直接用 np.nan 掩蔽部分数据（如按角度剔除 θ > 25° 的区域）看似直观，但常引发视觉失真——表面出现拉伸、塌陷或非圆柱形轮廓。根本原因在于：Plotly 默认对每根坐标轴进行自适应范围裁剪（auto-range），当掩码区域产生大量 NaN 后，x、y、z 数组的有效极值发生偏移，导致坐标系比例失调，破坏圆柱固有的等距对称结构。
✅ 正确做法是：先计算完整圆柱未掩蔽时的理论坐标范围，再显式固定 scene 轴范围（range），确保即使部分数据被设为 np.nan，绘图空间仍严格保持原始几何比例。
以下为优化后的完整实现：
import plotly.graph_objects as go
import numpy as np
mask_cylinder = True # 启用角度遮蔽
# 构建参数网格：θ ∈ [0, 2π)，z ∈ [0, 10]
theta = np.linspace(0, 2 * np.pi, 100)
z = np.linspace(0, 10, 100)
theta_grid, z_grid = np.meshgrid(theta, z)
r = 1.0
x = r * np.cos(theta_grid)
y = r * np.sin(theta_grid)
# ✅ 关键步骤：预先计算完整圆柱的坐标边界（含 padding 避免边缘锯齿）
def pad_limits(min_val, max_val, padding_factor=0.05):
range_val = max_val - min_val
padding = range_val * padding_factor
return min_val - padding, max_val + padding
xlim = pad_limits(x.min(), x.max())
ylim = pad_limits(y.min(), y.max())
zlim = pad_limits(z.min(), z.max())
# 应用角度掩码：仅保留 θ ≤ 25°（即 ≈0.436 rad）区域
if mask_cylinder:
mask = theta_grid > np.deg2rad(25) # 注意：此处为“排除”，故用 >
x = np.where(mask, np.nan, x)
y = np.where(mask, np.nan, y)
z = np.where(mask, np.nan, z_grid) # z 必须同步掩蔽，否则高度错位
else:
z = z_grid
fig = go.Figure()
fig.add_trace(go.Surface(
x=x, y=y, z=z,
colorscale="Viridis",
showscale=True,
lighting=dict(diffuse=0.9, specular=0.3), # 可选：增强立体感
))
fig.update_layout(
title="3D Pipe with Angular Mask (0°–25° only)",
scene=dict(
xaxis_title="X", yaxis_title="Y", zaxis_title="Z",
xaxis=dict(visible=True, range=xlim, backgroundcolor="white"),
yaxis=dict(visible=True, range=ylim, backgroundcolor="white"),
zaxis=dict(visible=True, range=zlim, backgroundcolor="white"),
# 可选：启用旋转约束提升交互体验
camera=dict(eye=dict(x=1.5, y=1.5, z=1.5))
),
margin=dict(l=0, r=0, b=0, t=50)
)
fig.show()
? 关键要点总结：
-
范围锁定是核心：
scene.xaxis.range等必须显式设置为未掩蔽状态下的全范围（含合理 padding），而非依赖默认 auto-range； -
三轴同步掩蔽：
x,y,z必须使用同一布尔掩码，否则会导致曲面拓扑断裂； -
避免透明度替代方案：虽然用
opacity+ 多层渲染可绕过 NaN 问题，但会显著增加内存与渲染开销，且难以精确控制接缝；本方法零额外计算，纯前端修复； -
精度提示：若需更精细角度控制（如 0°–25° 闭区间），建议将
mask = theta_grid > np.deg2rad(25)改为mask = (theta_grid np.deg2rad(25))，并注意np.linspace边界包含性。
该方案兼顾简洁性、性能与几何保真度，适用于任意轴对称曲面（锥面、球面等）的局部区域可视化需求。
