python绘制等高线图

在Python中，可以使用matplotlib和numpy库来绘制等高线图（contour plot）。等高线图通常用于显示三维数据的二维投影，其中颜色或线条的密度表示第三维的值。

以下是一个简单的示例，说明如何使用这些库来创建等高线图：

import numpy as np  
import matplotlib.pyplot as plt  
  
# 创建数据  
x = np.linspace(-5, 5, 100)  
y = np.linspace(-5, 5, 100)  
X, Y = np.meshgrid(x, y)  
Z = np.sin(np.sqrt(X**2 + Y**2))  
  
# 创建等高线图  
plt.figure()  
CS = plt.contour(X, Y, Z)  
plt.colorbar(CS)  # 添加颜色条  
plt.title('Contour Plot')  # 添加标题  
plt.xlabel('X')  # 添加X轴标签  
plt.ylabel('Y')  # 添加Y轴标签  
plt.show()

在这个例子中，我们首先使用numpy的linspace函数创建两个一维数组x和y，它们分别包含-5到5之间的100个等间隔点。然后，我们使用meshgrid函数创建两个二维网格X和Y，它们分别包含x和y中所有点的所有可能组合。接下来，我们计算每个(X, Y)点对应的Z值，这里我们使用了一个简单的函数np.sin(np.sqrt(X**2 + Y**2))。最后，我们使用matplotlib的contour函数创建等高线图，并使用colorbar函数添加一个颜色条来指示不同等高线的值。我们还添加了标题和轴标签，并使用show函数显示图形。