matplotlib学习笔记（五）

导入模块

import matplotlib.pyplot as plt
import numpy as np

---

等值线图

contour

dx = 0.01
dy = 0.01
x = np.arange(-2,2,dx)
y = np.arange(-2,2,dy)
X,Y = np.meshgrid(x,y)  #生成网格点坐标矩阵

def f(x,y):
    return(1-y**5+x**5)*np.exp(-x**2-y**2)

C = plt.contour(X,Y,f(X,Y),8,colors='black') #等值线值的轮廓线及标注
plt.clabel(C,inline=1,fontsize=10) #绘制等值线的标注
plt.contourf(X,Y,f(X,Y),8,cmap=plt.cm.winter) #绘制等值线间的颜色填充
plt.colorbar() #添加颜色图例
plt.show()

---

参考:

法比奥·内利. Python数据分析实战：第2版.北京：人民邮电出版社, 2019.11.