等值線圖的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()