kmeans聚类算法python代码

下面是一个简单的 k-means 聚类算法的 Python 代码示例，使用的是 sklearn 库中的 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]])

# 设置聚类中心数量为 2
kmeans = KMeans(n_clusters=2, random_state=0).fit(X)

# 打印聚类结果
print(kmeans.labels_)

在这个示例中，我们将数据点分成了 2 个聚类，并打印了每个数据点所属的聚类。
希望这能对你有所帮助。