2022年 11月 6日

python等高线图的画法

等高线图的画法。关键点:显示哪些等高线要清楚,知道如何显示等高线的图例和数值。

  1. import numpy as np
  2. import matplotlib.pyplot as plt
  3. from mpl_toolkits.axes_grid1 import ImageGrid
  4. import matplotlib as mpl
  5. def process_signals(x,y):
  6. return (1-(x**2+y**2))*np.exp(-y**3/3)
  7. x=np.arange(-1.5,1.5,0.1)
  8. y=np.arange(-1.5,1.5,0.1)
  9. X,Y=np.meshgrid(x,y)
  10. Z=process_signals(X,Y)
  11. N=np.arange(-1,1.5,0.3)#用来指明等高线对应的值为多少时才出现对应图线
  12. fig=plt.figure(1,(18,6))#设定图形大小
  13. fig.add_subplot(121)#画第一张图
  14. CS=plt.contour(Z,N,linewidth=2,cmap=mpl.cm.jet)#画出等高线图,cmap表示颜色的图层。
  15. plt.clabel(CS,inline=True,fmt='%1.1f',fontsize=10)#在等高线图里面加入每条线对应的值
  16. plt.colorbar(CS)#标注右侧的图例
  17. fig.add_subplot(122)#画第二张图
  18. CS=plt.contourf(Z,N,linewidth=2,cmap=mpl.cm.jet)#画出等高线填充图,cmap表示颜色的图层。
  19. plt.colorbar(CS)#标注右侧的图例
  20. plt.show()