如何在 Python 中将列中的 k 均值预测聚类添加到数据帧

Question

我有一个关于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)

如何将包含聚类结果的列作为附加列添加到我的第一个数据帧"data"中？谢谢！

Answer 1

假设列长度与数据帧df中的每一列相同，您需要做的就是：

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