溫馨提示×

在Matplotlib中如何繪制等高線圖

小億
84
2024-05-20 12:54:25
欄目: 編程語言

要在Matplotlib中繪制等高線圖,可以使用contourcontourf函數(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ù)值范圍。最后,我們添加了一些標簽和標題來美化圖形。

0