Python - 如何从群集图表中获取列表?

时间:2016-04-28 03:51:52

标签: python scikit-learn

我正在使用sklearn来使用K-Mean聚类方法。但问题是我可以从群集图表中检索列表。我的代码如下所示

kmeans = KMeans(n_clusters=5)
clusters = kmeans.fit(existing_df)

existing_df_2d['cluster'] = pd.Series(clusters.labels_, index=existing_df_2d.index)

existing_df_2d.plot(
        kind='scatter',
        x='PC2',y='PC1',
        c=existing_df_2d.cluster.astype(np.float), 
        figsize=(16,8))

enter image description here

实际上我已经获得了如何使用R编程语言获取所有列表的代码。

existing_df$cluster <- existing_clustering$cluster
table(existing_df$cluster)

//first cluster 
rownames(subset(existing_df, cluster==1))
existing_clustering$centers[1,]

//second cluster 
rownames(subset(existing_df, cluster==2))
existing_clustering$centers[2,]

1 个答案:

答案 0 :(得分:0)

尝试existing_df_2d['cluster'].to_dict()