k-means聚类例子

假设我们有以下数据集：

X = [[1, 2], [1, 4], [1, 0], [4, 2], [4, 4], [4, 0]]

我们想要将这些数据分为两个簇。以下是使用k-means聚类算法的Python代码：

from sklearn.cluster import KMeans

# 定义聚类数为2
kmeans = KMeans(n_clusters=2)

# 将数据拟合到模型中
kmeans.fit(X)

# 获取聚类结果
labels = kmeans.labels_
print(labels)

输出：

[1 1 1 0 0 0]

可以看到，k-means算法将数据集分成了两个簇，标签为0和1。我们还可以通过以下代码获取簇的中心：

centers = kmeans.cluster_centers_
print(centers)

输出：

[[4.         2.        ]
 [1.         2.        ]]

其中，第一个簇的中心为[4, 2]，第二个簇的中心为[1, 2]。