plt.scatter添加图例

在使用plt.scatter添加图例时，可以使用plt.legend()函数来实现。具体步骤如下：

1.首先，需要在plt.scatter()函数中设置label参数，用于指定每个点的标签。

2.然后，使用plt.legend()函数来添加图例。其中，handles参数用于指定图例中的标记，可以通过scatter.legend_elements()方法获取；labels参数用于指定每个标记对应的标签；title参数用于指定图例的标题。

下面是一个示例代码：

import matplotlib.pyplot as plt
import numpy as np

# 生成数据
x = np.random.rand(50)
y = np.random.rand(50)
c = np.random.randint(0, 5, 50)

# 绘制散点图
scatter = plt.scatter(x, y, c=c, cmap=plt.cm.plasma, label=c)

# 添加图例
plt.legend(handles=scatter.legend_elements()[0], labels=["0", "1", "2", "3", "4"], title="Classes")

# 显示图形
plt.show()

在这个例子中，我们使用了随机生成的数据，其中c数组用于指定每个点的标签。然后，我们使用plt.scatter()函数绘制散点图，并将cmap参数设置为plt.cm.plasma，用于指定颜色映射。接着，我们使用plt.legend()函数添加图例，其中handles参数使用scatter.legend_elements()方法获取，labels参数指定每个标记对应的标签，title参数指定图例的标题。最后，使用plt.show()函数显示图形。