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曲面图,其中的X、Y、Z参数分别表示X、Y、Z坐标轴上的数值,cmap参数表示使用的颜色映射。
最后使用ax.set_xlabel()、ax.set_ylabel()、ax.set_zlabel()函数设置坐标轴标签,使用ax.set_title()函数设置图形标题,最后使用plt.show()函数显示图形。
运行以上代码,即可得到一个绘制有3D曲面的图形。这个图形可以通过鼠标拖动、旋转和缩放进行交互操作。
