如何将图例颜色与绘图颜色匹配?

时间:2021-05-19 23:57:45

标签: python legend scatter-plot

我正在处理一个非常简单的示例。我在同一组轴上创建了三个散点图,我绘制的每个数据集都有一个不同的关联颜色图。然而,传说并不像我想要的那样;这是为什么?

import matplotlib.pyplot as plt
from sklearn.datasets import make_blobs

X, y = make_blobs(n_samples=500, n_features=2, cluster_std=1.0, centers=[(0,0), (3,3)])

plt.scatter(X[:, 0], X[:, 1], c=y, s=50, cmap='rainbow');


clf_tree = DecisionTreeClassifier(criterion='entropy', max_depth=1)
clf_tree = clf_tree.fit(X, y)
y_pred = clf_tree.predict(X)

clf_tree = DecisionTreeClassifier(criterion='entropy', min_samples_leaf = 3)
clf_tree = clf_tree.fit(X, y)
y_pred = clf_tree.predict(X)

clf_tree = DecisionTreeClassifier(criterion='entropy', max_leaf_nodes = 3)
clf_tree = clf_tree.fit(X, y)
y_pred = clf_tree.predict(X)



#shap.decision_plot(expected_value, sh, features_display, link='logit', highlight=misclassified)

for i in range(len(y)):
    if y[i] != y_pred[i]:
        plt.scatter(X[i, 0], X[i, 1], c=y[i], s=50, cmap='Dark2')
plt.legend(("0","1","miss"))

enter image description here

0 个答案:

没有答案