Matplotlib輪廓圖

輪廓圖(有時稱為“水準圖”)是一種在二維平面上顯示三維表面的方法。 它繪製了y軸上的兩個預測變數X Y和輪廓的回應變數Z。 這些輪廓有時稱為z切片或等回應值。

如果要查看Z如何隨兩個輸入X和Y的變化而變化，則輪廓圖是非常適用的，例如Z = f(X，Y)。 兩個變數函數的等值線或等值線是函數具有常數值的曲線。

引數x和y通常限於稱為meshgrid的規則網格。 numpy.meshgrid使用x值數組和y值數組創建一個矩形網格。

Matplotlib API包含分別繪製輪廓線和填充輪廓的contour()和contourf()函數。 兩個函數都需要三個參數x，y和z。

import numpy as np
import matplotlib.pyplot as plt
xlist = np.linspace(-3.0, 3.0, 100)
ylist = np.linspace(-3.0, 3.0, 100)
X, Y = np.meshgrid(xlist, ylist)
Z = np.sqrt(X**2 + Y**2)
fig,ax=plt.subplots(1,1)
cp = ax.contourf(X, Y, Z)
fig.colorbar(cp) # Add a colorbar to a plot
ax.set_title('Matplotlib輪廓圖')
#ax.set_xlabel('x (cm)')
ax.set_ylabel('y (cm)')
plt.show()

執行上面示例代碼，得到以下結果 -