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

最新推荐文章于 2024-08-19 15:52:51 发布

原创最新推荐文章于 2024-08-19 15:52:51 发布 · 1.5k 阅读

12 ·

CC 4.0 BY-SA版权

python 专栏收录该内容

12 篇文章

订阅专栏

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

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