在Matplotlib中使用scatter()绘制3D散点图添加图例

在Matplotlib中使用 scatter() 绘制3D散点图添加图例，可采取以下步骤：

步骤

设置图形大小并调整子图间和周围的填充。
初始化一个变量 N 来存储样本数据的数量。
创建 x 和 y 数据点；创建 z1 和 z2 数据点列表。
在当前图形上添加一个子图，使用 projection=’3d’ 。
使用 plot() 在2d轴上绘制 x ， y 和 z1 数据点，使用标记*。
使用 plot() 在2d轴上绘制 x ， y 和 z2 数据点，使用标记o。
将图例置于图形上。
使用 show() 方法显示图形。

示例

import matplotlib.pyplot as plt 
import numpy as np

plt.rcParams["figure.figsize"] = [7.50, 3.50]
plt.rcParams["figure.autolayout"] = True 

N = 100

x = np.random.rand(N)
y = np.random.rand(N)

z1 = [i*j for (i, j) in zip(x, y)]
z2 = [i+j for (i, j) in zip(x, y)]

axes = plt.subplot(111, projection='3d')

axes.plot(x, y, z1, "*", label="xy")
axes.plot(x, y, z2, "o", label="x+y")

plt.legend(loc="upper right")

plt.show()