在MATLAB中,可以使用kmeans
函數來實現k-means算法。以下是一個示例:
% 生成一些隨機數據
data = [randn(100,2)+ones(100,2); randn(100,2)-ones(100,2)];
% 使用kmeans函數進行聚類
k = 2; % 聚類的簇數
[idx, centers] = kmeans(data, k);
% 繪制結果
figure;
gscatter(data(:,1), data(:,2), idx);
hold on;
plot(centers(:,1), centers(:,2), 'k*', 'MarkerSize', 10);
legend('Cluster 1', 'Cluster 2', 'Centroids');
上述代碼首先生成了一些隨機數據,然后使用kmeans
函數對數據進行聚類,指定了2個簇。最后,使用gscatter
函數將數據點根據聚類結果進行可視化,使用plot
函數將聚類中心繪制在圖上。