Python K-means算法分析客户数据

import pandas as pd
import matplotlib.pyplot as plt
#引入sklearn框架,导入K均值聚类算法
from sklearn.cluster import KMeans

inputfile = r'C:/Users/Administrator/Desktop/transformdata.xls' #待聚类的数据文件

outputfile=r'C:/Users/Administrator/Desktop/data_type.xls'

#读取数据并进行聚类分析
data = pd.read_excel(inputfile) #读取数据

#利用K-Means聚类算法对客户数据进行客户分群,聚成4类

k = 4                       #需要进行的聚类类别数

iteration = 500

kmodel = KMeans(n_clusters = k,max_iter=iteration)

kmodel.fit(data) #训练模型

r1 = pd.Series(kmodel.labels_).value_counts()

r2 = pd.DataFrame(kmodel.cluster_centers_)

r = pd.concat([r2, r1],axis = 1)

r.columns=list(data.columns) + [u'聚类数量']

r3 = pd.Series(kmodel.labels_, index=data.index)

r = pd.concat([data, r3], axis = 1)

r.columns = list(data.columns) + [u'聚类类别']

r.to_excel(outputfile)

kmodel.cluster_centers_

kmodel.labels_

plt.rcParams['font.sans-serif'] = ['SimHei']

plt.rcParams['axes.unicode_minus'] = False

for i in range(k):

  cls=data[r[u'聚类类别']==i]

  cls.plot(kind = 'kde', linewidth = 2, subplots = True, sharex = False)

  plt.suptitle('客户群 = %d; 聚类数量 = %d' %(i, r1[i]))

plt.legend()

plt.show()




数据在:适用于数据分析和数据挖掘的客户数据-数据集文档类资源-CSDN下载

  • 0
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值