kmeans算法

内容导读

互联网集市收集整理的这篇技术教程文章主要介绍了kmeans算法，小编现在分享给大家，供广大互联网技能从业者学习和参考。文章包含1641字，纯文字阅读大概需要3分钟。

内容图文

原理

以k为参数，把n个对象分成k个簇，使簇内具有较高的相似性，而簇间的相似度较低。处理过程：

1. 随机选择k个点作为初始的聚类中心
2. 对于剩下的点，根据其与聚类中心的距离（如欧式距离等），将其归入最近的簇
3. 对每个簇，计算所有点的均值作为新的聚类中心
4. 重复2、3直到聚类中心不再发生改变

实战

31省市居民家庭消费调查

import numpy as np
from sklearn.cluster import KMeans


# 31个城市，每个城市8项数据
def loadData(filePath):
    fr = open(filePath, 'r+', encoding='UTF-8')   # 返回文件对象fr
    lines = fr.readlines()                        # 读取所有行并返回列表，列表元素为8维数据
    retData = []
    retCityName = []
    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 retData, retCityName


if __name__ == '__main__':
    data, cityName = loadData('city.txt')             # data共有31个元素，每个元素是8维的
    km = KMeans(n_clusters=4)
    label = km.fit_predict(data)                      # 返回每个城市对应的簇中心并为簇分配序号
    # print(km.cluster_centers_)                      # 4个簇中心，每个簇中心是8维的
    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])

输出

Expenses:3788.76
['河北', '山西', '内蒙古', '辽宁', '吉林', '黑龙江', '江西', '山东', '河南', '贵州', '陕西', '甘肃', '青海', '宁夏', '新疆']
Expenses:7754.66
['北京', '上海', '广东']
Expenses:4512.27
['江苏', '安徽', '湖南', '湖北', '广西', '海南', '四川', '云南']
Expenses:5678.62
['天津', '浙江', '福建', '重庆', '西藏']

内容总结

以上是互联网集市为您收集整理的kmeans算法全部内容，希望文章能够帮你解决kmeans算法所遇到的程序开发问题。 如果觉得互联网集市技术教程内容还不错，欢迎将互联网集市网站推荐给程序员好友。

内容备注

版权声明：本文内容由互联网用户自发贡献，该文观点与技术仅代表作者本人。本站仅提供信息存储空间服务，不拥有所有权，不承担相关法律责任。如发现本站有涉嫌侵权/违法违规的内容， 请发送邮件至 gblab@vip.qq.com 举报，一经查实，本站将立刻删除。

内容手机端

---