您的位置:宽带测速网 > 编程知识 > 在Matplotlib中如何绘制等高线图

在Matplotlib中如何绘制等高线图

2025-06-27 17:08来源:互联网 [ ]

要在Matplotlib中绘制等高线图,可以使用contourcontourf函数。以下是一个简单的示例代码:

import numpy as npimport 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函数,您可以添加一个颜色条来显示对应的数值范围。最后,我们添加了一些标签和标题来美化图形。