matplotlib 绘制三维散点图

项目介绍

在撰写论文时常常会用到 matplotlib 来绘制三维散点图，如下所示：

代码实现

plt.figure(figsize=(8,6))  # 设置画布大小
ax = plt.axes(projection='3d')  # 设置三维轴
ax.scatter3D(X, Y, Z)  # 三个数组对应三个维度（三个数组中的数一一对应）
plt.xticks(range(11))  # 设置 x 轴坐标
plt.rcParams.update({
     'font.family': 'Times New Roman'})
plt.rcParams.update({
     'font.weight': 'normal'})
plt.rcParams.update({
     'font.size': 20})
plt.xlabel('X')
plt.ylabel('Y', rotation=38)  # y 轴名称旋转 38 度
ax.set_zlabel('Z')  # 因为 plt 不能设置 z 轴坐标轴名称，所以这里只能用 ax 轴来设置（当然，x 轴和 y 轴的坐标轴名称也可以用 ax 设置）
plt.savefig('3D.jpg', bbox_inches='tight', dpi=2400)  # 保存图片，如果不设置 bbox_inches='tight'，保存的图片有可能显示不全
plt.show()