sklearn 实现kmeans

Sklearn是一个用于机器学习的Python库，它提供了许多常用的机器学习算法的实现。其中包括k-means聚类算法的实现。下面是使用sklearn实现k-means聚类的代码：

import numpy as np
import matplotlib.pyplot as plt
from sklearn.cluster import KMeans

# 读取数据
data = np.genfromtxt('kmeans.txt', delimiter=' ')

# 聚类数量
k = 4

# 训练模型
model = KMeans(n_clusters=k)
model.fit(data)

# 分类中心点坐标
centers = model.cluster_centers_

# 预测结果
result = model.predict(data)

# 绘制数据点和分类中心点
mark = ['or', 'og', 'ob', 'ok']
for i, d in enumerate(data):
    plt.plot(d