Python绘制曲面和曲面对应的等高线图

先将需要的库调入：

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

绘制曲面，曲面的函数为z=(1-x/2+x**5+y**3)*np.exp(-x**2-y**2)

fig = plt.figure()
ax =fig.add_subplot(111,projection='3d')


n = 256
u = np.linspace(-3,3,n)
x,y = np.meshgrid(u,u)


z=(1-x/2+x**5+y**3)*np.exp(-x**2-y**2)
ax.plot_surface(x,y,z,rstride=3,cstride=3,cmap=cm.jet)

曲面如下显示：

绘制对应的 曲面等高线图，并标记等高线的高度值：

fig = plt.figure()
bx =fig.add_subplot(111)
C=contour(X, Y, z)
bx.clabel(C, alpha=0.75, cmap='jet',inline=1, fontsize=10)
show()

等高线如图：