图例和散点图元素颜色不匹配

Question

我无法在图例中使用与来自循环数据相同的颜色。'alldata'的尺寸为569x30，因此定义多种颜色是不可行的

for labels in ['benign', 'malignant']:
scatter = plt.scatter(svd_pca_eqvt[:,0],svd_pca_eqvt[:,1],label=labels, c=alldata['target'])
plt.legend()

输入图片描述

Answer 1

您可以在for循环中添加颜色，如下所示:

for labels, color in zip (['benign', 'malignant'], ['#FECB52','#7E1E9C']):
scatter = plt.scatter(svd_pca_eqvt[:,0],svd_pca_eqvt[:,1],label=labels, c=color)
plt.legend()

# FECB52——比;黄色的# 7 e1e9c——比;深紫色