k-means

按课件写的matlab版k-means聚类程序

num =40;%一类随机点的数目
x1=1+69*rand(1,num); %生成第一类随机点
y1=1+69*rand(1,num);
scatter(x1,y1)
x2=40+60*rand(1,num);%生成第二类随机点
y2=40+60*rand(1,num);
x=[x1,x2];
y=[y1,y2];
x = floor(x);
y = floor(y);
hold on;
scatter(x2,y2)
title('raw data');
r=randi(2*num,1,2);
while r(1)==r(2)
    r=randi(2*num,1,2);
end
h1=x(r(1));
h2=y(r(1));
h3=x(r(2));
h4=y(r(2));
h=[h1,h2,h3,h4];%生成随机特征向量作为聚类中心
hn=[0,0,0,0];

while hn ~= h%聚类中心不再发生变化，则停止计算
    h = hn;
    k1=0;
    k2=0;
    sum1 = 0;
    sum2 = 0;
    sum3 = 0;
    sum4 = 0;
    for i=1:2*num
        if ((x(i)-h1)^2 + (y(i)-h2)^2) < ((x(i)-h3)^2 + (y(i)-h4)^2)%不能改成大于号，按最小距离原则分类
            sum1 = sum1+x(i);
            sum2 = sum2+y(i);
            flag(i) = 1;%模式分类标签
            k1 = k1+1;
        else
            sum3 = sum3+x(i);
            sum4 = sum4+y(i);
            flag(i) = 0;
            k2 = k2+1;
        end
    end
    h1 = sum1/k1;
    h2 = sum2/k1;
    h3 = sum3/k2;
    h4 = sum4/k2;
    hn=[h1,h2,h3,h4];%迭代一次后的聚类中心
%     flag
%     scatter(h1,h2,'r*');
%     scatter(h3,h4,'g*');
end
hold off;
figure;
hold on
title('result');
for i=1:2*num
    if flag(i) == 1
        scatter(x(i),y(i),'r*');
    else
        scatter(x(i),y(i),'k*');
    end
end