matplotlib.pyplot.contour 簡單等高線繪制

contour(X, Y, Z)

X,Y是與Z形狀相同的二維數組，可以通過 numpy.meshgrid()創建。

numpy.meshgrid()----從坐標向量返回坐標矩陣

生成的x,y坐標矩陣，組合形成網格點

a = np.array([1,2,3])          #a.shape  (3,)
b = np.array([11,22,33,44])    #b.shape  (4,)
x,y = np.meshgrid(a,b)

x                              #x.shape(4,3)

array([[1, 2, 3],
       [1, 2, 3],
       [1, 2, 3],
       [1, 2, 3] ])

y                             #y.shape(4,3)

array([[11, 11, 11],
       [22, 22, 22],
       [33, 33, 33],
       [44, 44, 44]])        

 1）contour()

a = np.array([1,2,3])
z = np.array([[0,1,1],[1,0,1],[1,1,0]])
x,y = np.meshgrid(a,a)
plt.contour(x,y,z,3)          #數字3是等高線的數量

 

2）contourf() 返回填充等高線圖，其他輸入輸出和contour()沒有區別

a = np.array([1,2,3])
z = np.array([[0,1,1],[1,0,1],[1,1,0]])
x,y = np.meshgrid(a,a)
plt.contourf(x,y,z,3)         #contourf()