kmeans亚洲足球队分类python代码

导入包
使用make_blobs生成随机点cluster_std

from sklearn.datasets import make_blobs
from sklearn.cluster import KMeans
import numpy as np
import matplotlib.pyplot as plt
%matplotlib inline

#获取数据
#centers=3 默认值，生成3个聚类点集
X,y = make_blobs()
plt.scatter(X[:,0],X[:,1],c=y)

训练模型

#选取3个种子
kmeans = KMeans(n_clusters=3)
# 无监督学习：estimator.fit(X)
# 监督学习：estimator.fit(X,y)
kmeans.fit(X)

显示聚类中心

#查看聚类中心坐标点
cluster_centers_ = kmeans.cluster_centers_
cluster_centers_