要在Matplotlib中繪制等高線圖,可以使用contour
或contourf
函數(shù)。以下是一個簡單的示例代碼:
import numpy as np
import matplotlib.pyplot as plt
# 生成數(shù)據(jù)
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()
在上面的示例中,我們首先生成了一個二維數(shù)組Z
,然后使用contour
函數(shù)繪制了該數(shù)組的等高線圖。您可以調整levels
參數(shù)來改變等高線的數(shù)量。通過調用colorbar
函數(shù),您可以添加一個顏色條來顯示對應的數(shù)值范圍。最后,我們添加了一些標簽和標題來美化圖形。