Python实现简单层次聚类算法以及可视化实现简单层次聚类算法以及可视化
本文实例为大家分享了Python实现简单层次聚类算法，以及可视化，供大家参考，具体内容如下
基本的算法思路就是：把当前组间距离最小的两组合并成一组。
算法的差异在算法如何确定组件的距离，一般有最大距离，最小距离，平均距离，马氏距离等等。
代码如下：
import numpy as np
import data_helper
np.random.seed(1)
def get_raw_data(n):
_data=np.random.rand(n,2)
#生成数据的格式是n个（x,y）
_groups={idx:[[x,y]] for idx,(x,y) in enumerate(_data)}
return _groups
def cal_distance(cluster1,cluster2):
#采用最小距离作为聚类标准
_min_distance=10000
for x1,y1 in cluster1:
for x2,y2 in cluster2:
_distance=(x1-x2)**2+(y1-y2)**2
if _distance<_min_distance:
_min_distance=_distance
return _distance
groups=get_raw_data(10)
count=0
while len(groups)!=1:#判断是不是所有的数据是不是归为了同一类
min_distance=10000
len_groups=len(groups)
for i in groups.keys():
for j in groups.keys():
if i>=j:
continue
distance=cal_distance(groups[i],groups[j])
if distance<min_distance:
min_distance=distance
min_i=i
min_j=j#这里的j>i
groups[min_i].extend(groups.pop(min_j))
data_helper.draw_data(groups)
#一共n个簇，共迭代n-1次
运行的效果就是迭代一次，组数就会少一次，调用画图方法，同一组的数据被显示为一个颜色。
您可能感兴趣的文章您可能感兴趣的文章:python KNN算法实现鸢尾花数据集分类决策树剪枝算法的python实现方法详解Python 实现大整数乘法
算法的示例代码应用OpenCV和Python进行SIFT算法的实现详解python实现PID算法及测试的例子python 随机森林算法及其
优化详解Python使用sklearn实现的各种回归算法示例基于python实现雪花算法过程详解
评论0