Python/matplotlib mplot3d- jak ustawić maksymalną wartość dla osi Z?

Question

Próbuję utworzyć trójwymiarowy wykres powierzchni dla wyrażenia: z = y^2/x, dla x w przedziale [-2,2] iy w przedziale [-1.1,4.4]. Chcę również, aby wartości z z zakresu od -4 do 4.

Problem polega na tym, że gdy oglądam gotowy surfaceplot, wartości osi Z nie zatrzymują się na [-4,4].

Moje pytanie brzmi: jak mogę "usunąć" wartość osi z, która znajduje się poza intervall [-4,4] z gotowego wykresu?

Mój kod to:

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 x-axis should be [-2,2] 
plt.show() 

Answer 1

wycinek danych będzie tego dokonać, ale to nie jest bardzo ładna.

z[z>4]= np.nan 
z[z<-4]= np.nan 

Answer 2

Mam ten sam problem i nadal nie znalazłem nic lepszego niż przycinanie moich danych. Niestety w moim przypadku jestem związany z matplotlib 1.2.1. Ale w przypadku, gdy można uaktualnić do wersji 1.3.0, można mieć rozwiązanie: wydaje się, że jest kilka nowych API związanych z zakresami osi. W szczególności możesz być zainteresowany "set_zlim".

Edytuj 1: Zarządzaj migracją mojego środowiska, aby korzystać z programu matplotlib 1.3.0; set_zlim pracował jak urok :)

Kod follwing pracował dla mnie (Nawiasem mówiąc używam tego na OSX, nie jestem pewien, że to ma wpływ?):

# ---------------------------------------------------------------------------- 
# Make a 3d plot according to data passed as arguments 
def Plot3DMap(self, LabelX, XRange, LabelY, YRange, LabelZ, data3d) : 
    fig = plt.figure() 
    ax = fig.add_subplot(111, projection="3d") 
    xs, ys = np.meshgrid(XRange, YRange) 
    surf = ax.plot_surface(xs, ys, data3d) 
    ax.set_xlabel(LabelX) 
    ax.set_ylabel(LabelY) 
    ax.set_zlabel(LabelZ) 
    ax.set_zlim(0, 100) 
    plt.show()