2012-07-11 119 views
0

我有一个1000x6数据集,并使用下面的kmeans脚本是好的,但是当我想输出其中一个集群时,它只能作为一个列出现?在Kmeans中输出IDX?

%% cluster 
opts = statset('MaxIter', 100, 'Display', 'iter'); 
[clustIDX, clusters, interClustSum, Dist] = kmeans(data, K, 'options',opts, ... 
'distance','sqEuclidean', 'EmptyAction','singleton', 'replicates',6); 

%% plot data+clusters 
figure, hold on 
scatter3(data(:,1),data(:,2),data(:,3), 5, clustIDX, 'filled') 
scatter3(clusters(:,1),clusters(:,2),clusters(:,3), 100, (1:K)', 'filled') 
hold off, xlabel('x'), ylabel('y'), zlabel('z') 

%% plot clusters quality 
figure 
[silh,h] = silhouette(data, clustIDX); 
avrgScore = mean(silh); 

%% Assign data to clusters 
% calculate distance (squared) of all instances to each cluster centroid 
D = zeros(numObservarations, K);  % init distances 
for k=1:K 
%d = sum((x-y).^2).^0.5 
D(:,k) = sum(((data - repmat(clusters(k,:),numObservarations,1)).^2), 2); 
end 

% find for all instances the cluster closet to it 
[minDists, clusterIndices] = min(D, [], 2); 

% compare it with what you expect it to be 
sum(clusterIndices == clustIDX) 

% Output cluster data to K datasets 
K1 = data(clustIDX==1) 
K2 = data(clustIDX==2)... etc 

不应该K1 = data(clustIDX==1)输出全行信息?不只是一列,而是六个像原始数据集?或者这只是输出距离?

回答

2

替换

K1 = data(clustIDX==1) 
K2 = data(clustIDX==2) 

K1 = data(clustIDX==1,:) 
K2 = data(clustIDX==2,:) 

第一种只检索对应的行的第一列中。第二个应该解决它,我已经尝试过,它的工作原理。

+0

嗨,谢谢你,工作。只使用matlab几次,所以不太确定所有的语法! +1 – 2012-07-11 19:23:07