2016-07-14 16 views
4

Haben Sie eine Frage über Kmeans Clustering in Python.Wie k-means vorhergesagte Cluster in einer Spalte zu einem Datenrahmen in Python hinzufügen

Also habe ich die Analyse, dass Art und Weise:

from sklearn.cluster import KMeans 

km = KMeans(n_clusters=12, random_state=1) 
new = data._get_numeric_data().dropna(axis=1) 
kmeans.fit(new) 
predict=km.predict(new) 

Wie kann ich die Spalte mit Cluster-Ergebnissen zu meinem ersten Datenrahmen „Daten“ als zusätzliche Spalte hinzu? Vielen Dank!

+1

so sind Sie im Wesentlichen die Frage, wie eine Spalte zu einem Datenrahmen hinzufügen? Wie in: http://stackoverflow.com/questions/12555323/adding-new-column-to-existing-dataframe-in-python-pandas oder hier http://stackoverflow.com/questions/18942506/add-new -Spalte-in-Pandas-Datenrahmen-Python –

Antwort

5

die Spaltenlänge ist als jede Spalte als die gleiche Unter der Annahme, in Ihnen Dataframe df, alles, was Sie tun müssen, ist dies:

df['NEW_COLUMN'] = Series(predict, index=df.index)