使用python实现k-means聚类算法

k-means聚类算法是一种基于类中心的聚类方法。在python中，可以使用Scikit-learn库中的KMeans类来实现k-means聚类。下面是一个简单的示例代码：

from sklearn.cluster import KMeans
import numpy as np

# 创建样本数据
X = np.array([[1, 2], [1, 4], [1, 0],
              [4, 2], [4, 4], [4, 0]])

# 创建k-means模型，并设置聚类中心数量为2
kmeans = KMeans(n_clusters=2)

# 训练模型
kmeans.fit(X)

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

# 获取聚类中心
cluster_centers = kmeans.cluster_centers_

# 打印结果
print("Labels:", labels)
print("Cluster Centers:", cluster_centers)

在这个示例中，我们创建了一个样本数据，并使用KMeans类训练了一个k-means模型。然后我们使用labels_属性获取每个样本的聚类结果，并使用cluster_centers_属性获取聚类中心。