为什么matplotlib mplot3d散点图遵守掩码规则但曲面图不生效？

核心差异来自两类图表的渲染逻辑和matplotlib对掩码数组的支持度不同 #

• 首先是离散点和曲面的渲染逻辑本质不同
  • 散点图scatter是逐点独立渲染：每个点位单独判断是否有效，被掩码的负Z值点会直接被丢弃，所有参与渲染的点Z值均为正，因此色标范围仅覆盖正值区间。
  • 曲面图plot_surface是基于网格顶点插值生成连续面片：只要一个面片的四个顶点没有全部被掩码，该面片就会被保留渲染。渲染过程中会对四个顶点的原始数值（包括掩码位置对应的负Z值）做插值计算，自然会产生介于正负之间的过渡值，使得整体数值范围包含负值。
• 其次是plot_surface对numpy掩码数组的支持存在缺陷：它在计算颜色映射的全局极值时，不会自动忽略掩码位置的原始值，会直接读取整个Z数组的最小/最大值，哪怕对应位置已经被标记为掩码，这也直接导致曲面图的色标下限延伸到了负区间。

修复方案 #

如果要让曲面图也完全不显示负值区域，不需要给X、Y数组加掩码，只需要把Z数组中掩码位置的值替换为np.nan即可，plot_surface遇到包含NaN的面片会自动跳过渲染：

Z = 1-X-Y
Z1 = np.ma.masked_less(Z,0)
# 新增这行，把掩码位置替换为NaN
Z1[Z1.mask] = np.nan

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