python - 如何让我的 matplotlib 图例显示颜色和相关值？

Question

我正在运行一个简单的聚类分析，当我绘制它时，图例没有按预期工作。图例仅显示 2 个值（o 和 1）。我期待看到集群，0-4 以及它们在图中表示的颜色。这是我到目前为止的代码：

%matplotlib inline
import matplotlib.pyplot as plt
import seaborn as sns; sns.set()
import numpy as np
from sklearn.datasets.samples_generator import make_blobs
from sklearn.cluster import KMeans

#Creating DataFrame for plotting
X= pd.DataFrame({"Times_Correct":tmpX['Times_Correct'], "Times_Incorrect": tmpX['Times_Incorrect']})

#Plotting fans correct versus incorrect predictions
plt.scatter(X['Times_Correct'], X['Times_Incorrect'], s=50);

#Clustering
kmeans = KMeans(n_clusters=5)
kmeans.fit(X)
X_kmeans = kmeans.predict(X)

#Plotting clusters
plt.scatter(tmpX['Times_Correct'], tmpX['Times_Incorrect'], c=X_kmeans,  s=50, cmap='viridis')
plt.xlabel("Times Correct", size=12)
plt.ylabel("Times Incorrect", size=12)

#add legend to the plot
plt.legend(np.unique(X_kmeans))

电流输出

有什么方法可以显示 X_kmeans 中的唯一值及其相关颜色，还是我需要在这里重新设计逻辑？

预先感谢您的任何帮助

标签： pythonmatplotliblegend-properties