将图例添加到matplotlib.pyplot.scatter

Question

我想在散点图中添加图例。这是我到目前为止的代码：

mat_contents1 = sio.loadmat('Data/synth1.mat')
mat_dict1 = {}
mat_dict1.update(mat_contents1)
X1 = mat_dict1['X']
Y1 = mat_dict1['y']
plt.figure(figsize=(8, 6))
plt.title("synth1")
plt.scatter(X1[:,0], X1[:,1], c=Y1[:,0])
plt.legend()
plt.show()

在X中，我具有x和y轴的值。在y中，我具有从0到3的类标签。 产生的图： this

现在我想要一个图例，说蓝色= 0级，黄色= 1级，等等。