仅显示来自一个集群的数据

Question

我有一个熊猫数据框。

input_data = {'col1': [1, 2, 3], 'col2': [3, 4, 5]}
d = pd.DataFrame(data=input_data)

anotherdata= magic(d)
df = pd.DataFrame(data=anotherdata)

我使用 DBSCAN 来集群 df。 结果我有cluster_labels。在这种情况下，标签的值可以从 -1（离群值）到 2。

我希望有机会单独显示来自特定类的数据，并可以通过索引访问初始数据帧d。

例如，我在input_data 中有一个索引为 1 的元素。 该元素被分配到簇 0，簇 0 中没有其他元素。

如何通过索引在input_data 中找到这个元素？

Answer 1

你可能想要使用

d[cluster_labels == 0]

除非您的 magic 函数更改了索引。