Python - 使用Matplotlib绘制具有不同颜色的分类数据点的图例

Question

我花了很多时间试图为我的愚蠢问题找到解决方案，但我没有设法做到这一点。我是Python的新手（最重要的是绘图）。

我想做的是为我的数据点绘制一个小图例。我有一个二维数据集，每个条目都有一个类[0,1,2]。这是我为绘制数据集而编写的代码：

plt.figure(figsize=(15,8))
plt.scatter(reduced_dataset[:,0], reduced_dataset[:,1], c=y.Gender)
plt.xlabel('PC 1')
plt.ylabel('PC 2') 
plt.legend()
plt.figure()

这是我的输出图片：

所以，我想要的是首先要知道三种颜色代表哪个类，并在右上角添加一个小图例来解释这一点。

编辑：我添加一些细节。 我的数据集名为 reduced_dataset ，是一个像这样的二维数据集，通过应用PCA从8维数据集中获得（n_components = 2）：

[[5.29251,-0.680271]
[-10.6902,0.135495]
[-0.676506,-0.0493725]
[0.306184,-0.315342]
[-2.73479,-0.705164]]

我有一个向量，表示我的数据集的每一行的类：

[0,0,1,0,2]

所以这个课可能是1,2或3，这就是为什么我的情节中的数据点有三种颜色的原因。我需要一个这样的传奇：

谢谢。

Answer 1

如果将图保持为静态，则按顺序绘制每个数据系列，将自动创建图例：

dd1 :=to_number(to_char(test_date1,'dd'));
dd2 :=to_number(to_char(test_date2,'dd'));
mm1 :=to_number(to_char(test_date1,'mm'));
mm2 :=to_number(to_char(Test_Date2,'mm'));