我试图为表达式制作一个三维表面图:z = y ^ 2/x,区间为[-2,2]的x,区间为[-1.4,1.4]的y.我还希望z值的范围从-4到4.
问题是,当我查看完成的表面图时,z轴值不会停在[-4,4].
所以我的问题是如何从完成的图中"移除"范围[-4,4]之外的z轴值？
我的代码是:
from mpl_toolkits.mplot3d import axes3d
import matplotlib.pyplot as plt
import numpy as np
fig = plt.figure()
ax = fig.gca(projection="3d")
x = np.arange(-2.0,2.0,0.1,float) # x in interval [-2,2]
y = np.arange(-1.4,1.4,0.1,float) # y in interval [-1.4,1.4]
x,y = np.meshgrid(x,y)
z = (y**2/x) # z = y^2/x
ax.plot_surface(x, y, z,rstride=1, cstride=1, linewidth=0.25)
ax.set_zlim3d(-4, 4) # viewrange for z-axis should be [-4,4]
ax.set_ylim3d(-2, 2) # viewrange for y-axis should be [-2,2]
ax.set_xlim3d(-2, 2) # viewrange for …