[½»Á÷] k-means¾ÛÀà ÒÑÓÐ1È˲ÎÓë

clear
clc
load('data.mat');
m=input('ÇëÊäÈëÑù±¾¾ØÕóÐÐÊý: ');
n=input('ÇëÊäÈëÑù±¾¾ØÕóÁÐÊý: ');
%dispÃèÊöÊý¾Ý  %dataÊý¾Ý
disp(load('data.mat'));     
%ÃèÊöÊý¾Ý
data1=data(:,1);
data2=data(:,2);
data3=data(:,3);
%»­³ödataÊý×éÖÐ×ø±êµãµÄÇé¿ö
scatter3(data1,data2,data3);   
counter=0;
%È·¶¨¾ÛÀàÊýÄ¿µÄÊäÈëÔ¼Êø
n=input('ÇëÊäÈë¾ÛÀàÊýÄ¿£º');
%È·¶¨¾ÛÀàÊýÄ¿µÄºÏÀíÐÔ
while n>m
disp('ÄúÊäÈëµÄ¾ÛÀàÊýÄ¿¹ý´ó£¬ÇëÊäÈëÕýÈ·µÄkÖµ£º');
n=input('ÇëÊäÈë¾ÛÀàÊýÄ¿');
end
if n==1
disp('¾ÛÀàÊýÄ¿²»ÄÜΪ1£¬ÇëÊäÈëÕýÈ·µÄkÖµ£º');
%È·¶¨ÕýÈ·µÄ¾ÛÀàÊýÄ¿
n=input('ÇëÊäÈë¾ÛÀàÊýÄ¿');
end
[m,n]=size(data);
%patternÀ©´óÁËÔ­¾ØÕóµÄά¶È
pattern=zeros(m,n+1);   
%³õʼ»¯¾ÛÀàÖÐÐÄ  
center=zeros(n,n);
%patternÀïËùÓÐÐУ¬µÚ1ÁÐÖÁµÚnÁÐ=dataËùÓÐÊý¾Ý£¬µÚ4ÁÐΪ0
pattern(:,1:n)=data(:,:);

%¿ªÊ¼Ñ­»·

for x=1:n
    center(x,:)=data(randi(m,1),:);%µÚÒ»´ÎËæ»ú²úÉú¾ÛÀàÖÐÐÄ
end
while 1
distance=zeros(1,n);
num=zeros(1,n);
new_center=zeros(n,n);
for x=1:m
    for y=1:n
    distance(y)=norm(data(x,:)-center(y,:));%¼ÆË㵽ÿ¸öÀàµÄ¾àÀë
    end
    [~, temp]=min(distance);%Çó×îСµÄ¾àÀë
    pattern(x,n+1)=temp;         
end
k=0;
for y=1:n
    for x=1:m
        if pattern(x,n+1)==y
           new_center(y,:)=new_center(y,:)+pattern(x,1:n);
           num(y)=num(y)+1;
        end
    end
    new_center(y,:)=new_center(y,:)/num(y);
    if norm(new_center(y,:)-center(y,:))<0.1
        k=k+1;
    end
end
if k==n
     break;
else
     center=new_center;
end
end
[m, n]=size(pattern);


%×îºóÏÔʾ¾ÛÀàºóµÄÊý¾Ý
figure ;hold on
view(3)
for i=1:m
    if pattern(i,n)==1
         plot(pattern(i,1),pattern(i,2),'r*');
         plot(center(1,1),center(1,2),'ko');
    elseif pattern(i,n)==2
         plot(pattern(i,1),pattern(i,2),'g*');
         plot(center(2,1),center(2,2),'ko');
    else
         plot(pattern(i,1),pattern(i,2),'m*');
         plot(center(3,1),center(3,2),'ko');
    end
end

grid on;
c1=zeros(m,n-1);
c2=zeros(m,n-1);
c3=zeros(m,n-1);
for i=1:m
    if pattern(i,n)==1
    c1(i,:)=pattern(i,1:3);
    elseif pattern(i,n)==2
    c2(i,:)=pattern(i,1:3);
    else
    c3(i,:)=pattern(i,1:3);
    end
end
%ɾ³ýÁãÐÐ
c1(all(c1==0,2),:)=[];
c2(all(c2==0,2),:)=[];
c3(all(c3==0,2),:)=[];