Python绘制等高线图教程

作者：热心市民鹿先生2024.01.17 22:26浏览量：5

简介：本文将介绍如何使用Python绘制等高线图，通过matplotlib库实现可视化。我们将使用numpy生成数据，并使用matplotlib的pyplot模块绘制等高线图。

在Python中，我们可以使用matplotlib库来绘制等高线图。以下是一个简单的示例：
首先，我们需要导入所需的库。在Jupyter Notebook中，您可以直接输入以下代码：

  1. import numpy as np
  2. import matplotlib.pyplot as plt

接下来，我们创建一个二维等高线图。在这个例子中，我们将创建一个简单的正弦波。

  1. x = np.linspace(-np.pi, np.pi, 100)  # 生成100个均匀分布的x值
  2. y = np.sin(x)  # 对x值应用sin函数得到y值

然后，我们使用contour函数来绘制等高线图。这个函数需要指定等高线的数量，颜色和线条宽度等参数。

  1. plt.contour(x, y, y, levels=5)  # 绘制5条等高线

最后，我们显示图像。

  1. plt.show()

完整的代码如下：
```python
import numpy as np
import matplotlib.pyplot as plt
x = np.linspace(-np.pi, np.pi, 100) # 生成100个均匀分布的x值
y = np.sin(x) # 对x值应用sin函数得到y值
plt.contour(x, y, y, levels=5) # 绘制5条等高线
plt.show() # 显示图像

最热文章