其中meshgrid函数的作用请参考下面博文:
https://blog.csdn.net/weixin_39541558/article/details/80551788

源代码如下:

import matplotlib.pyplot as plt
import numpy as np

def f(x,y):
    return (1-x/2+x**5+y**3)*np.exp(-x**2-y**2)
n = 256
x=np.linspace(-3,3,n)
y=np.linspace(-3,3,n)

X,Y =np.meshgrid(x,y)           #将向量x,y定义的区域转换为矩阵
plt.contourf(X,Y,f(X,Y),8,alpha = 0.75,cmap = plt.cm.hot)
#  cmap设置颜色相关
C=plt.contour(X,Y,f(X,Y),8,color="black",linewidth=.5)
plt.clabel(C,inline='Ture',fontsize=10)


plt.xticks(())
plt.yticks(())
plt.show()

效果图如下:
matplotlib模块(绘图模块)使用——006_contours等高线图

相关文章: