In[6]:nba=pd.read_csv('nba_2013.csv')
In[7]:kmeans=KMeans(n_clusters=cluster_num)
In[8]:(nba[['ppg','atr']])
Out[8]:
KMeans(algorithm='auto',copy_x=True,init='k-means++',max_iter
=300,n_clusters=5,n_init=10,n_jobs=1,precompute_distances='aut
o',random_state=None,tol=0.0001,verbose=0)
In[9]:nba['cluster']=kmeans.labels_
In[10]:visualize_clusters(nba,cluster_num)
图5-8