我正在使用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))
实际上我已经获得了如何使用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,]
答案 0 :(得分:0)
尝试existing_df_2d['cluster'].to_dict()