web-dev-qa-db-ja.com

列のk-means予測クラスターをPythonのデータフレームに追加する方法

Pythonでのkmeansクラスタリングについて質問があります。

だから私はそのように分析をしました:

from sklearn.cluster import KMeans

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

クラスターの結果を含む列を最初のデータフレーム「データ」に追加の列として追加するにはどうすればよいですか?ありがとう!

9
Keithx

列の長さがデータフレームdfの各列と同じであると仮定すると、必要なのは次のとおりです。

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