K-Means聚类代码实现

最新推荐文章于 2024-04-29 18:16:07 发布

浮生勿语

最新推荐文章于 2024-04-29 18:16:07 发布

阅读量235

点赞数

分类专栏： Algorithm

本文链接：https://blog.csdn.net/qq_36381573/article/details/103300124

版权

Algorithm 专栏收录该内容

18 篇文章 1 订阅

订阅专栏

#encoding:utf-8
import numpy as np
from sklearn.cluster import KMeans

def LoadData(filePath):
    f=open(filePath,'r+')
    lines=f.readlines()
    retCityName=[]
    retData=[]
    for line in lines:
         items=line.strip().split(",")
         retCityName.append(items[0])
         retData.append([float(items[i]) for i in range(1,len(items))])
    return retCityName,retData

if  __name__=='__main__':
    CityName,Data=LoadData('city.txt')

    km=KMeans(n_clusters=3)
    label=km.fit_predict(Data)
    # 横向计算每个城市的总开销，并把它归类到相应的簇里面。然后对每个簇进行求平均数。得到expense[i]i从0到3
    expenses=np.sum(km.cluster_centers_,axis=1)
    CityCluster=[[],[],[]]
    for i in range(len(CityName)):
        CityCluster[label[i]].append(CityName[i])

    for i in range(len(CityCluster)):
        print("Expenses:%.2f"%expenses[i])
        print(CityCluster[i])