我正在研究来自 sklearn 的 iris 数据集。您可能知道 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()
我有散点图,你可以看到黄色、绿色和紫色的点。我想知道哪个颜色点属于哪个类('setosa'、'versicolor'、'virginica')。我想显示图例,以便我知道哪种颜色代表哪个类