使用python绘制三维等高线图

在Python中，我们可以使用matplotlib库的mplot3d模块来创建三维等高线图。以下是基本步骤：

1. 首先，确保已经安装了`matplotlib`库和`numpy`库，如果未安装，可以使用pip命令安装：

   pip install matplotlib numpy

2. 导入所需库：

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

3. 创建数据集，比如通过网格计算一些函数的值：

# 定义x、y坐标范围
x = np.linspace(-5, 5, 100)
y = np.linspace(-5, 5, 100)
X, Y = np.meshgrid(x, y)

# 计算Z值，例如Z=f(X,Y) = sin(X**2 + Y**2)
Z = np.sin(X**2 + Y**2)

4. 创建3D图形，并添加等高线：

fig = plt.figure()
ax = fig.add_subplot(111, projection='3d')

# 添加等高线
ax.plot_surface(X, Y, Z, cmap='viridis', alpha=0.8) # cmap可以选择不同的颜色映射

# 显示等高线轮廓
cset = ax.contourf(X, Y, Z, zdir='z', offset=-1, cmap='viridis')
cset = ax.contour(X, Y, Z, zdir='x', offset=-5, cmap='viridis')
cset = ax.contour(X, Y, Z, zdir='y', offset=5, cmap='viridis')

# 设置轴标签和标题
ax.set_xlabel('X轴')
ax.set_ylabel('Y轴')
ax.set_zlabel('Z轴')
ax.set_title('三维等高线图')

# 显示图形
plt.show()