在python中裁剪表面以获得大于的Z值

时间:2018-04-22 14:41:42

标签: python numpy surface

所以我有一个代表一些真实数据的函数的3D曲面图。 Z轴上任何大于8.5的值都是物理上不可能的,因此我想从我的绘图中删除它们。我一直在努力做到这一点,任何帮助将不胜感激:

enter image description here

我想将8以上的值(大多数红色)设置为NaN,但是当我尝试这个时,情节似乎是空的。我做了一些谷歌搜索,发现冲浪地块不像NaN(显然),并想知道这里是否有人可以提供帮助!

代码:

fig = plt.figure()
ax = fig.gca(projection='3d')

X = np.arange(1.0,101.0,1.0) 
Y = np.arange(1.0,4.1, 0.04) 


X, Y = np.meshgrid(X, Y)

Z = ((0.3973473*(np.exp(np.log10(X)))) + 0.6176893) * Y

surf = ax.plot_surface(X, Y, Z, cmap=cm.coolwarm,
                    linewidth=0, antialiased=False)

1 个答案:

答案 0 :(得分:0)

今天用一双新眼睛来管理:

qt = np.linspace(1.0, 100.0, num=100)
ic = np.linspace(1.0, 4.0, num=100)

qt, ic = np.meshgrid(qt, ic)

Z = ((0.3973473*(np.exp(np.log10(qt)))) + 0.6176893) * ic

for i in np.arange(len(qt)):
    for j in np.arange(len(ic)):
        if Z[j,i] > 8:
            Z[j,i] = np.nan
        else:
            pass


fig = plt.figure()
ax = fig.gca(projection='3d')

surf = ax.plot_surface(qt, ic, Z)

plt.show()

enter image description here