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