Home › Fóruns › Fórum Machine Learning e Data Science com Python de A à Z › K-means – Salvar base original › Responder a: K-means – Salvar base original
29 de novembro de 2021 às 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')