Question

我将DBSCAN群集用于文本文档，如下所示，感谢this post。

db = DBSCAN(eps=0.3, min_samples=2).fit(X)
core_samples_mask1 = np.zeros_like(db1.labels_, dtype=bool)
core_samples_mask1[db1.core_sample_indices_] = True
labels1 = db1.labels_

现在，我想查看哪个文档属于哪个群集，例如：

[I have a car and it is blue] belongs to cluster0

或

idx [112] belongs to cluster0

与我的问题在here中询问的方式类似，但是我已经测试了其中提供的一些答案：

X[labels == 1,:]

我得到了：

array([[0, 0, 1, 0, 0, 0, 0, 1, 0, 1, 0, 0, 0, 0],
       [0, 0, 1, 0, 0, 0, 0, 1, 0, 1, 0, 0, 0, 0]], dtype=int64)

但这对我没有帮助。如果您有任何建议或方法，请告诉我。

Answer 1

如果您的熊猫数据框df的列为idx和messages，那么您要做的就是

df['cluster'] = db.labels_

以获取具有集群成员资格的新列cluster。

这是一个带有虚拟数据的简短演示：

import numpy as np
import pandas as pd
from sklearn.cluster import DBSCAN

X = np.array([[1, 2], [5, 8], [2, 3],
               [8, 7], [8, 8], [2, 2]])

db = DBSCAN(eps=3, min_samples=2).fit(X)
db.labels_
# array([0, 1, 0, 1, 1, 0], dtype=int64)

# convert our numpy array to pandas:
df = pd.DataFrame({'Column1':X[:,0],'Column2':X[:,1]})
print(df)
# result:
   Column1  Column2
0        1        2
1        5        8
2        2        3
3        8        7
4        8        8
5        2        2

# add new column with the belonging cluster:
df['cluster'] = db.labels_

print(df)
# result:
   Column1  Column2  cluster
0        1        2        0
1        5        8        1
2        2        3        0
3        8        7        1
4        8        8        1
5        2        2        0

在基于密度的聚类中如何获取属于其聚类的文档？

1 个答案: