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