2016-04-28 17 views
1

Ich verwende sklearn, um K-Mean Clustering-Methode zu verwenden. Aber das Problem ist, ich kann Listen aus dem Cluster-Plot-Diagramm abrufen. Mein Code ist wie untenPython - Wie bekomme ich eine Liste von einem Cluster-Diagramm?

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

Eigentlich hatte ich schon den Code wie alle Listen zu erhalten, aber in R Programmiersprache.

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,] 

Bitte beraten. Vielen Dank.

Antwort

0

Versuchen Sie existing_df_2d['cluster'].to_dict()

Verwandte Themen