Home › Forums › Fórum Machine Learning e Data Science com Python de A à Z › K-means – Salvar base original › Reply To: K-means – Salvar base original
29 de novembro de 2021 at 12:28
#32442
Olá Alisson,
Você pode usar o método transform ou fit_transform para gerar os nomes dos clusters, e depois adicionar uma coluna ao dataframe original com os rótulos dos grupos formados. Mais ou menos assim:
grupos = kmeans.fit_transform(X)
df['grupos'] = grupos
df.to_csv('kmeans.csv')
