要在Matplotlib中繪制等高線圖,可以使用contour
或contourf
函數。以下是一個簡單的示例代碼:
import numpy as np
import matplotlib.pyplot as plt
# 生成數據
x = np.linspace(-3, 3, 100)
y = np.linspace(-3, 3, 100)
X, Y = np.meshgrid(x, y)
Z = np.sin(X) * np.cos(Y)
# 繪制等高線圖
plt.contour(X, Y, Z, levels=10)
plt.colorbar()
plt.xlabel('x')
plt.ylabel('y')
plt.title('Contour Plot')
plt.show()
在上面的示例中,我們首先生成了一個二維數組Z
,然后使用contour
函數繪制了該數組的等高線圖。您可以調整levels
參數來改變等高線的數量。通過調用colorbar
函數,您可以添加一個顏色條來顯示對應的數值范圍。最后,我們添加了一些標簽和標題來美化圖形。