阅读量:7
要在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函数，您可以添加一个颜色条来显示对应的数值范围。最后，我们添加了一些标签和标题来美化图形。
