如何为基于sklearn iris数据集的散点图添加图例以区分类别

当前使用 sklearn 提供的 iris 数据集完成可视化分析，该数据集共包含3个类别：['setosa', 'versicolor', 'virginica']。
目前已基于数据集完成散点图绘制，实现代码如下：

from sklearn.datasets import load_iris
iris=load_iris()
Y_train=iris.target
X_train=iris.data
class_labels=iris.target_names
plt.scatter(X_train[:,0], X_train[:,1], c=Y_train)
plt.xlabel('attr1')
plt.ylabel('attr2')
plt.show()

生成的散点图效果如下：

现存问题 #

生成的散点图包含黄色、绿色、紫色三类散点，但没有标注各类颜色对应的实际鸢尾花类别，无法直接建立颜色和分类的对应关系。期望在散点图中展示图例，明确不同颜色对应的具体类别标签。

内容的提问来源于stack exchange，提问作者user15407748