使用kmeans聚类

以下是使用KMeans进行聚类的示例代码：

from sklearn.cluster import KMeans
import numpy as np

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

# 创建KMeans模型并训练
kmeans = KMeans(n_clusters=2, random_state=0).fit(X)

# 查看聚类结果
labels = kmeans.labels_
print(labels)

# 查看聚类中心
centers = kmeans.cluster_centers_
print(centers)

在上面的代码中，我们首先创建了一个包含6个样本的数据集X。然后，我们使用KMeans模型对数据进行聚类，将聚类数设置为2。最后，我们打印出每个样本的聚类标签和聚类中心。