等值线图的Python绘制方法

等值线图或等高线图在科学界经常用到，它是由一些封闭的曲线组成的，来表示三维结构表面。虽然看起来复杂，其实用matplotlib实现起来并不难。代码如下：

import numpy as np
import matplotlib.pyplot as plt
dx=0.01;dy=0.01
x=np.arange(-2.0,2.0,dx)
y=np.arange(-2.0,2.0,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.contourf(X,Y,f(X,Y),8)
plt.clable(C,inline=1,fontsize=10)

结果如下：

使用等值线图，在图的一侧增加图例作为图表中所用颜色的说明几乎是必需的，在上述代码最后增加colorbar()函数就可以实现。

plt.colorbar()