私はsklearnの虹彩データセットに取り組んでいます。ご存知かもしれませんが、虹彩データセットには 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()
黄色、緑、紫の点が見える散布図があります。どの色のドットがどのクラス (「setosa」、「versicolor」、「virginica」) に属しているかを知りたいです。凡例を表示して、どの色がどのクラスを表しているかがわかるようにしたい