python - 在 K-Means 聚类之后绘制聚类

Question

我在一个数据集上使用了 K-Means 聚类，我发现了 7 个聚类，但我无法绘制这些聚类。请在下面找到我使用的代码：-

# Plot the clustered data
fig, ax = plt.subplots(figsize=(10, 10))
plt.scatter(general_population_scaled[kmeans.labels == 0, 0], general_population_scaled[kmeans.labels == 0, 1],
            c='green', label='cluster 1')

请帮助我更正代码以生成结果。这是我得到的错误：-

AttributeError: 'KMeans' object has no attribute 'labels'

标签： pythonscikit-learnk-means

Answer 1

簇标签位于对象的labels_属性中kmeans，而不是labels属性中

https://scikit-learn.org/stable/modules/generated/sklearn.cluster.KMeans.html