Responder a: K-means – Salvar base original

#32442
Denny Ceccon
Moderador

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')