使用 Matplotlib 中的 scatter() 在 3D 散点图中添加图例
matplotlibpythondata visualization
要使用 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()
输出
它将产生以下输出 −
