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

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

步骤

  • 设置图形大小并调整子图之间和周围的填充。

  • 初始化一个变量 **N** 来存储样本数据数量。

  • 创建 **x** 和 **y** 数据点；创建 **z1** 和 **z2** 数据点列表。

  • 使用 **projection='3d'** 向当前图形添加子图。

  • 使用 **plot()** 绘制 **x**、**y** 和 **z1** 数据点，使用 **marker ***。

  • 使用 **plot()** 绘制 **x**、**y** 和 **z2** 数据点，使用 **marker 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()

输出

它将产生以下输出：

更新于：2022年2月1日

5K+ 次浏览

启动您的 职业生涯

通过完成课程获得认证

开始
广告