Matplotlib：对3D冲浪图赋予轴限制的方法

Question

我绘制如下图。

from mpl_toolkits.mplot3d import Axes3D
import matplotlib
from matplotlib import cm


fig = plt.figure(figsize=(24,12))
ax = fig.gca(projection='3d')
ax.view_init(elev=45, azim=225)

surf = ax.plot_surface(TT, FF, np.abs(Zxx),cmap=cm.rainbow,)

ax.set_xlabel('Time($sec$)')
ax.set_ylabel('Frequency($Hz$)')
ax.set_zlabel('STFT Magnitude')

ax.set_ylim3d([5,50])

fig.colorbar(surf, shrink=0.6, aspect=20)

plt.show()

这里的问题，

ax.set_ylim3d([5,50])

图形结束了。

如果matplotlib没问题，我想知道解决方案是什么。