mpl_toolkits.mplot3d中的Axes3D()函数绘制3D图像

mpl_toolkits.mplot3d是matplotlib中用于绘制3D图像的工具包。其中的Axes3D()函数可以创建一个3D坐标系对象，用于绘制3D图像。下面我们来看一个简单的例子，演示如何使用Axes3D()函数绘制一个简单的3D图像。

首先，我们需要导入所需要的库:

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

然后，创建一个3D坐标系对象:

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

接下来，我们可以使用plot_surface()函数来绘制一个3D曲面图。这里我们以一个简单的函数z = x^2 + y^2为例:

import numpy as np

# 生成数据
x = np.linspace(-5, 5, 100)
y = np.linspace(-5, 5, 100)
X, Y = np.meshgrid(x, y)
Z = X**2 + Y**2

# 绘制3D曲面图
ax.plot_surface(X, Y, Z, cmap='viridis')

然后，我们可以设置坐标轴的名称和范围:

ax.set_xlabel('X')
ax.set_ylabel('Y')
ax.set_zlabel('Z')
ax.set_xlim(-5, 5)
ax.set_ylim(-5, 5)
ax.set_zlim(0, 50)

最后，我们使用show()方法显示图像:

plt.show()

完整的代码如下:

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

# 创建3D坐标系对象
fig = plt.figure()
ax = fig.add_subplot(111, projection='3d')

# 生成数据
x = np.linspace(-5, 5, 100)
y = np.linspace(-5, 5, 100)
X, Y = np.meshgrid(x, y)
Z = X**2 + Y**2

# 绘制3D曲面图
ax.plot_surface(X, Y, Z, cmap='viridis')

# 设置坐标轴的名称和范围
ax.set_xlabel('X')
ax.set_ylabel('Y')
ax.set_zlabel('Z')
ax.set_xlim(-5, 5)
ax.set_ylim(-5, 5)
ax.set_zlim(0, 50)

# 显示图像
plt.show()

运行以上代码，我们就可以看到一个简单的3D图像，其中的曲面呈现出由x和y平面上的点生成的圆锥形。