1 mpl_toolkits.mplot3d 功能介绍
mpl_toolkits.mplot3d 是 Matplotlib 库中的一个子模块，用于绘制和可视化三维图形，包括三维散点图、曲面图、线图等。它提供了丰富的功能来创建和定制三维图形。以下是 mpl_toolkits.mplot3d 的主要功能和功能简介：
3D 散点图：通过
scatter函数，你可以绘制三维散点图，用于显示三维数据点的分布和关系。3D 曲面图：使用
plot_surface函数，可以创建三维曲面图，用于可视化三维数据的表面形状。这对于显示函数的三维性质非常有用。3D 线图：
plot函数允许你绘制三维线图，用于表示数据点之间的连接关系。这对于显示数据随时间或其他变量的变化趋势非常有用。3D 条形图：通过
bar3d函数，你可以创建三维柱状图，用于比较不同类别或组的数据。3D 散点图标记和颜色：你可以根据数据的特性来定制散点图的标记形状和颜色，以区分不同的数据点。
3D 坐标轴定制：你可以设置和定制三维图的坐标轴，包括添加标签、设置刻度和范围等。
图形样式定制：你可以设置图形的样式，包括标题、图例、背景色、线条样式和颜色等，以使图形更具吸引力和可读性。
3D 投影：
mpl_toolkits.mplot3d支持不同类型的三维投影，包括透视投影和正交投影，以满足不同的可视化需求。动画和交互性：你可以在三维图中添加动画效果或交互式元素，以更好地探索数据。
多图形组合：你可以在同一个三维图中组合多个不同类型的图形，以展示多个数据系列。
保存图形：最后，你可以将绘制的三维图形保存为图像文件，以便在文档中使用或与他人分享。
总之，mpl_toolkits.mplot3d 子模块为 Matplotlib 提供了强大的三维可视化工具，可以用于可视化和分析三维数据。根据你的需求，你可以选择不同的图形类型和样式来呈现数据，以便更好地理解和传达你的发现。
在 Matplotlib 的 mpl_toolkits.mplot3d 模块中，绘制 3D 图的一般流程包括以下步骤：
（1）导入必要的库和模块：
import matplotlib.pyplot as plt from mpl_toolkits.mplot3d import Axes3D（2）创建一个 3D 图形的绘图对象：
fig = plt.figure()（3） 创建一个 3D 子图：
ax = fig.add_subplot(111, projection='3d')（4） 准备数据：定义 X、Y 和 Z 数据，这些数据将在 3D 图中表示。
-
使用相应的 3D 绘图函数来绘制数据，例如：
- 散点图：
ax.scatter(x, y, z, c='color', marker='marker_style', label='label') - 曲面图：
ax.plot_surface(X, Y, Z, cmap='colormap') - 线图：
ax.plot(x, y, z, label='label') - 柱状图：
ax.bar3d(x, y, z, dx, dy, dz, shade=True)
- 散点图：
（5）添加坐标轴标签：
ax.set_xlabel('X轴标签') ax.set_ylabel('Y轴标签') ax.set_zlabel('Z轴标签')（6）添加标题：
plt.title('3D 图标题')（7）添加图例（如果需要）：
ax.legend()（8）显示图形：
plt.show()2 绘制3D散点图
在matplotlib的mpl_toolkits.mplot3d模块中，你可以使用`scatter`函数绘制多种类型的3D散点图。
下面是一些常见的3D散点图类型：
2.1 绘制单色散点图
所有的散点都使用相同的颜色。可以通过设置`c`参数指定颜色。
import numpy as npimport matplotlib.pyplot as pltfrom mpl_toolkits.mplot3d import Axes3D# 创建数据x = np.random.rand(100)y = np.random.rand(100)z = np.random.rand(100)# 创建图形和轴fig = plt.figure()ax = fig.add_subplot(111, projection='3d')# 绘制单色散点图ax.scatter(x, y, z, c='blue')# 设置坐标轴标签ax.set_xlabel('X')ax.set_ylabel('Y')ax.set_zlabel('Z')# 显示图形plt.show()
2.2 绘制彩色散点图
不同的散点可以使用不同的颜色。可以通过设置`c`参数为一个具有相同长度的数组来指定每个散点的颜色。
import numpy as npimport matplotlib.pyplot as pltfrom mpl_toolkits.mplot3d import Axes3D# 创建数据x = np.random.rand(100)y = np.random.rand(100)z = np.random.rand(100)colors = np.random.rand(100)# 创建图形和轴fig = plt.figure()ax = fig.add_subplot(111, projection='3d')# 绘制彩色散点图ax.scatter(x, y, z, c=colors)# 设置坐标轴标签ax.set_xlabel('X')ax.set_ylabel('Y')ax.set_zlabel('Z')# 显示图形plt.show()
2.3 绘制大小变化散点图
散点的大小可以根据某个特征进行变化。可以通过设置`s`参数为一个具有相同长度的数组来指定每个散点的大小。
import numpy as npimport matplotlib.pyplot as pltfrom mpl_toolkits.mplot3d import Axes3D# 创建数据x = np.random.rand(100) # x坐标y = np.random.rand(100) # y坐标z = np.random.rand(100) # z坐标colors = np.random.rand(100) # 散点颜色sizes = np.random.randint(low=50, high=200, size=100) # 散点大小# 创建图形和轴fig = plt.figure()ax = fig.add_subplot(111, projection='3d')# 绘制散点图ax.scatter(x, y, z, c=colors, s=sizes, alpha=0.8)# 设置坐标轴标签ax.set_xlabel('X') # 设置x轴标签ax.set_ylabel('Y') # 设置y轴标签ax.set_zlabel('Z') # 设置z轴标签# 显示图形plt.show()
2.4 绘制形状变化散点图
散点的形状可以根据某个特征进行变化。可以通过设置`marker`参数来指定散点的形状。
import numpy as npimport matplotlib.pyplot as pltfrom mpl_toolkits.mplot3d import Axes3D# 创建数据x = np.random.rand(100)y = np.random.rand(100)z = np.random.rand(100)markers = ['o', 's', '^', 'D']# 创建图形和轴fig = plt.figure()ax = fig.add_subplot(111, projection='3d')# 绘制形状变化散点图for i in range(len(x)): ax.scatter(x[i], y[i], z[i], marker=markers[i%len(markers)])# 设置坐标轴标签ax.set_xlabel('X')ax.set_ylabel('Y')ax.set_zlabel('Z')# 显示图形plt.show()
3 绘制3D线图
使用mpl_toolkits.mplot3d模块中的Axes3D对象绘制3D线图。
3.1 简单线图
使用`plot`函数绘制简单的曲线。
import numpy as npimport matplotlib.pyplot as pltfrom mpl_toolkits.mplot3d import Axes3D# 创建数据x = np.linspace(0, 1, 100)y = np.sin(2 * np.pi * x)z = np.cos(2 * np.pi * x)# 创建图形和轴fig = plt.figure()ax = fig.add_subplot(111, projection='3d')# 绘制简单线图ax.plot(x, y, z)# 设置坐标轴标签ax.set_xlabel('X')ax.set_ylabel('Y')ax.set_zlabel('Z')# 显示图形plt.show()
3.2 绘制多线图
使用`plot`函数绘制多条线并在同一图表中显示。
import matplotlib.pyplot as pltfrom mpl_toolkits.mplot3d import Axes3Dimport numpy as npplt.rcParams['font.sans-serif'] = ['SimHei']# 创建示例数据t = np.linspace(0, 20, 100) # 时间或X轴数据x1 = np.sin(t)y1 = np.cos(t)z1 = tx2 = np.sin(t) + 2y2 = np.cos(t) + 2z2 = t# 创建图形和子图fig = plt.figure()ax = fig.add_subplot(111, projection='3d')# 绘制第一条线ax.plot(x1, y1, z1, label='线1', color='blue', linestyle='-', linewidth=2)# 绘制第二条线ax.plot(x2, y2, z2, label='线2', color='red', linestyle='--', linewidth=2)# 添加标题plt.title('三维多线图示例')# 添加坐标轴标签ax.set_xlabel('X轴')ax.set_ylabel('Y轴')ax.set_zlabel('Z轴')# 添加图例ax.legend()# 显示图形plt.show() 
3.3 绘制带标记的线图
使用`plot`函数，并通过设置`marker`参数来添加线上的标记点。
import numpy as npimport matplotlib.pyplot as pltfrom mpl_toolkits.mplot3d import Axes3D# 创建数据x = np.linspace(0, 1, 100)y = np.sin(2 * np.pi * x)z = np.cos(2 * np.pi * x)# 创建图形和轴fig = plt.figure()ax = fig.add_subplot(111, projection='3d')# 绘制带标记的线图ax.plot(x, y, z, marker='o')# 设置坐标轴标签ax.set_xlabel('X')ax.set_ylabel('Y')ax.set_zlabel('Z')# 显示图形plt.show()
3.4 绘制立体网格线图
使用`plot_wireframe`函数在3D空间中绘制网格线图。
import numpy as npimport matplotlib.pyplot as pltfrom 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))# 创建图形和轴fig = plt.figure()ax = fig.add_subplot(111, projection='3d')# 绘制立体网格线图ax.plot_wireframe(X, Y, Z)# 设置坐标轴标签ax.set_xlabel('X')ax.set_ylabel('Y')ax.set_zlabel('Z')# 显示图形plt.show()
3.5 绘制等高线图
使用`contour`或`contourf`函数在3D空间中绘制等高线。
import numpy as npimport matplotlib.pyplot as pltfrom 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))# 创建图形和轴fig = plt.figure()ax = fig.add_subplot(111, projection='3d')# 绘制3D等高线图ax.contour3D(X, Y, Z, 50, cmap='viridis')# 设置坐标轴标签ax.set_xlabel('X')ax.set_ylabel('Y')ax.set_zlabel('Z')# 显示图形plt.show()
4 绘制3D条形图
在matplotlib中，由于 mpl_toolkits.mplot3d模块的限制，没有直接绘制3D的条形图的函数。不过你可以使用 bar3d函数绘制类似的3D条形图效果。
import numpy as npimport matplotlib.pyplot as pltfrom mpl_toolkits.mplot3d import Axes3D# 创建数据x = [1, 2, 3, 4, 5] # x轴坐标位置y = [1, 2, 3, 4, 5] # y轴坐标位置z = [0, 3, 2, 5, 1] # z轴高度，即条形的高度dx = dy = 0.8 # x和y方向的宽度dz = z # 条形的高度colors = ['red', 'green', 'blue', 'orange', 'purple'] # 颜色列表，一一对应于每个条形# 创建图形和轴fig = plt.figure()ax = fig.add_subplot(111, projection='3d')# 绘制3D条形图，并设置每个条形的颜色for xi, yi, zi, color in zip(x, y, z, colors): ax.bar3d(xi, yi, 0, dx, dy, zi, color=color)# 设置坐标轴标签ax.set_xlabel('X')ax.set_ylabel('Y')ax.set_zlabel('Z')# 显示图形plt.show()
来源地址：https://blog.csdn.net/qq_35831906/article/details/133073428
