使用Kmeans进行聚类，并将类别追加到csv文件最后一列

最新推荐文章于 2023-11-03 16:03:41 发布

imrush

最新推荐文章于 2023-11-03 16:03:41 发布

阅读量1.4k

点赞数 1

CC 4.0 BY-SA版权

分类专栏： python

本文链接：https://blog.youkuaiyun.com/imrush2019/article/details/116093486

python 专栏收录该内容

12 篇文章

订阅专栏

这段代码导入了numpy, pandas, seaborn, matplotlib和sklearn库，并应用KMeans算法对名为'samples.csv'的数据集进行了五类聚类。聚类后的数据被保存到'samples_labeled5.csv'文件中。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

import numpy as np
import pandas as pd
import seaborn as sns
import matplotlib.pyplot as plt
import os

from sklearn.cluster import KMeans

df = pd.read_csv(r'F:\kaiti\055n\data\samples.csv')
km = KMeans(n_clusters=5).fit(df)
df['cluster'] = km.labels_
df = df.to_csv(r'F:\kaiti\055n\data\samples_labeled5.csv'，index = 0)
# print(df['cluster'])