亚洲足球队等级分类
数据统计了2006年,2010年世界杯以及2007年亚洲杯,将相同等级的足球队自动分组
中国,50,50,9
日本,28,9,4
韩国,17,15,3
伊朗,25,40,5
沙特,28,40,2
伊拉克,50,50,1
卡塔尔,50,40,9
阿联酋,50,40,9
乌兹别克斯坦,40,40,5
泰国,50,50,9
越南,50,50,5
阿曼,50,50,9
巴林,40,40,9
朝鲜,40,32,17
印尼,50,50,9
import numpy as np
import pandas as pd
from sklearn.cluster import KMeans
data = pd.read_csv('./Asiafootball.txt',header = None,names = ['国家','2006','2010','2007'])
X = data.iloc[:,1:]
kmeans = KMeans(n_clusters=3)
kmeans.fit(X)
y_ = kmeans.predict(X)
for i in range(3):
print(c[np.argwhere(y_ == i).ravel()]) #ravel() 一层索引
['中国' '伊拉克' '卡塔尔' '阿联酋' '泰国' '越南' '阿曼' '印尼'] ['日本' '韩国'] ['伊朗' '沙特' '乌兹别克斯坦' '巴林' '朝鲜']