欢迎访问宙启技术站
智能推送

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

发布时间:2023-12-26 03:08:21

mpl_toolkits.mplot3d是Matplotlib中的一个模块,用于在2D平面上绘制3D图形。其中的Axes3D()函数用于创建一个3D场景图。

下面是一个使用Axes3D()函数绘制3D场景图的例子:

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

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 = np.sin(np.sqrt(X**2 + Y**2))

# 绘制3D图形
ax.plot_surface(X, Y, Z, cmap='hot')

# 添加标签和标题
ax.set_xlabel('X')
ax.set_ylabel('Y')
ax.set_zlabel('Z')
ax.set_title('3D Surface Plot')

# 显示图形
plt.show()

上述代码生成一个包含绘制3D曲面的图形。首先通过np.linspace函数生成-5到5之间的100个均匀分布的点,然后通过np.meshgrid函数生成网格点。接着通过np.sin函数计算出每个点的Z值。

创建图形对象(fig)并添加3D子图(ax),然后使用ax.plot_surface()函数绘制3D曲面图,其中的XYZ参数分别表示X、Y、Z坐标轴上的数值,cmap参数表示使用的颜色映射。

最后使用ax.set_xlabel()ax.set_ylabel()ax.set_zlabel()函数设置坐标轴标签,使用ax.set_title()函数设置图形标题,最后使用plt.show()函数显示图形。

运行以上代码,即可得到一个绘制有3D曲面的图形。这个图形可以通过鼠标拖动、旋转和缩放进行交互操作。