matplotlib/mplot3d 散点图尊重屏蔽，但表面图不 - 为什么？

问题描述

这篇文章的来源来自较早的post。

我有 X、Y、Z 二维数组。我只想在第一象限中看到所有 x、y 和 z 都为正的图。我对所有负 z 条目使用掩码。然而，注意到散点图确实尊重掩蔽并且不显示负 z 值。然而，表面图仍然显示负值。

请在下面找到代码-

import numpy as np
import matplotlib.pyplot as plt
from mpl_toolkits import mplot3d
from matplotlib import cm
%matplotlib notebook

x = np.linspace(0,1.5,10)
y = np.linspace(0,10)
X,Y = np.meshgrid(x,y)
Z = 1-X-Y
Z1 = np.ma.masked_less(Z,0)
X1 = np.ma.masked_array(X,Z1.mask)
Y1 = np.ma.masked_array(Y,Z1.mask)
fig = plt.figure()

# Two plots - one scatter plot,one surface plot
ax1 = fig.add_subplot(121,projection='3d')
ax2 = fig.add_subplot(122,projection='3d')

# Scatter plot
surf1 = ax1.scatter(X1,Y1,Z1,cmap=cm.coolwarm)
ax1.set_xlim(0,1.5)
ax1.set_ylim(0,1.5)
ax1.set_zlim(0,3)
ax1.set_xlabel('x')
ax1.set_ylabel('y')


# Surface plot
surf2 = ax2.plot_surface(X1,cmap=cm.coolwarm)
ax2.set_xlim(0,1.5)
ax2.set_ylim(0,1.5)
ax2.set_zlim(0,3)
ax2.set_xlabel('x')
ax2.set_ylabel('y')


fig.colorbar(surf1,ax=ax1,location='top')
fig.colorbar(surf2,ax=ax2,location='top')
fig.savefig('Question10CompareScatterandSurfacePlots')

颜色条明显表明散点图中没有负 z 点，而表面图延伸到负 z。为什么会这样？

解决方法

暂无找到可以解决该程序问题的有效方法，小编努力寻找整理中！

如果你已经找到好的解决方法，欢迎将解决方案带上本链接一起发送给小编。

小编邮箱:dio#foxmail.com (将#修改为@）