k均值聚类python代码

以下是一个简单的k均值聚类的Python代码示例：

import numpy as np
from sklearn.cluster import KMeans

# 生成数据
X = np.random.randn(100, 2)

# 建立模型
kmeans = KMeans(n_clusters=3)

# 训练模型
kmeans.fit(X)

# 预测结果
y_pred = kmeans.predict(X)

# 打印结果
print(y_pred)

在上面的代码示例中，我们首先生成了一个包含100个样本和2个特征的数据集。然后，我们使用scikit-learn库中的KMeans类来建立k均值聚类模型，并将聚类数设置为3。接下来，我们使用fit()方法训练模型，并使用predict()方法预测结果。最后，我们将预测结果打印出来。