python中matplotlib画等值线

使用matplotlib库可以绘制等值线图，具体步骤如下：

1.导入所需的库

import numpy as np
import matplotlib.pyplot as pltfrom mpl_toolkits.mplot3d import Axes3D

2.生成数据

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))

3.绘制等值线图

plt.contour(X, Y, Z, 20, cmap='RdGy')
plt.colorbar()

其中，`plt.contour(X, Y, Z, 20, cmap='RdGy')`表示绘制等值线图，其中X、Y为网格数据，Z为高度数据，20表示等值线的密度，cmap表示使用的颜色映射。

完整代码如下：

import numpy as np
import matplotlib.pyplot as plt
from mpl_toolkits.mplot3d import Axes3D

# 生成数据
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.contour(X, Y, Z, 20, cmap='RdGy')
plt.colorbar()

plt.show()

运行代码，即可看到绘制的等值线图。