温馨提示×

在Matplotlib中如何绘制等高线图

小亿
82
2024-05-20 12:54:25
栏目: 编程语言

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

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

0