在 KMeans 算法中获取每个集群中的值的数量答案

【问题标题】：Getting number of values in each cluster in KMeans Algorithm在 KMeans 算法中获取每个集群中的值的数量
【发布时间】：2019-11-24 03:15:02
【问题描述】：

如何在 Pandas 的 KMeans 算法中获取每个聚类中的值总数？

我尝试了以下方法：

kmeans_model = KMeans(n_clusters = 3, random_state = 1).fit(dataframe.iloc[:,:])
clusters = kmeans_model.labels_.count()

但它不起作用。

我的预期输出是这样的：

Clusters   Number_of_values
cluster_0  932
cluster_1  931
cluster_2  930

知道如何在 Pandas 中实现这一点吗？

我也试过了，效果很好。如果我有其他选择会更好。

from collections import Counter
print(Counter(kmeans_model.labels_))

提前致谢。

【问题讨论】：

【解决方案1】：

你可以这样做

将簇号列添加到数据框

kmeans_model = KMeans(n_clusters = 3, random_state = 1).fit(dataframe)

dataframe['kmean'] = kmeans_model.labels_

然后数一数

dataframe['kmean'].value_counts()

【讨论】：