python - 在python中获取sklearn中的簇大小

Question

我正在使用 sklearn DBSCAN 对我的数据进行如下聚类。

#Apply DBSCAN (sims == my data as list of lists)
db1 = DBSCAN(min_samples=1, metric='precomputed').fit(sims)

db1_labels = db1.labels_
db1n_clusters_ = len(set(db1_labels)) - (1 if -1 in db1_labels else 0)
#Returns the number of clusters (E.g., 10 clusters)
print('Estimated number of clusters: %d' % db1n_clusters_)

现在我想根据大小(每个集群中的数据点数)对前 3 个集群进行排序。请让我知道如何在 sklearn 中获取簇大小？

Answer 1

另一种选择是使用 numpy.unique:

db1_labels = db1.labels_
labels, counts = np.unique(db1_labels[db1_labels>=0], return_counts=True)
print labels[np.argsort(-counts)[:3]]