如何利用Python 进行K-means聚类分析

1、利用pandas导入墙绅褡孛数据集。这里有两个变量，假设学生的数学与英语两个成绩如下：import pan颊俄岿髭das as pddf=pd.DataFrame({'math':[98,78,54,89,24,60,98,44,96,90],'english':[92,56,90,57,46,75,76,87,91,88]})

3、处理训练集。假设数据中有分类结果，可以将分类结果作为目标变量。与逻辑回归不同，在没有y的情况下，直接用X变量集也可以进行K-means训练。X=df.ix[:,['math','english']]

5、模型训练。得到预测值。cluster=clf_KMeans.fit_predict(X)print(cluster)从结果中可以看到，样本被分为了两类。

7、增加聚类个数。并绘制图形。cluster2=KMeans(n_罕铞泱殳clusters=4).fit_predi艘早祓胂ct(X)plt.figure(1)plt.scatter(X['math'], X['english'],c=cluster2)plt.title("K-means test")plt.show()从图形上看，比2类时效果好很多。当然，在样本量足够的情况下，进行聚类分析，要根据实际情况或聚类效果选择K值。