/Python/K-Means Clustering in Python – 3 clusters

K-Means Clustering in Python – 3 clusters

from pandas import DataFrame
import matplotlib.pyplot as plt
from sklearn.cluster import KMeans

Data = {'x': [25,34,22,27,33,33,31,22,35,34,67,54,57,43,50,57,59,52,65,47,49,48,35,33,44,45,38,43,51,46],
        'y': [79,51,53,78,59,74,73,57,69,75,51,32,40,47,53,36,35,58,59,50,25,20,14,12,20,5,29,27,8,7]
       }
  
df = DataFrame(Data,columns=['x','y'])
  
kmeans = KMeans(n_clusters=3).fit(df)
centroids = kmeans.cluster_centers_
print(centroids)

plt.scatter(df['x'], df['y'], c= kmeans.labels_.astype(float), s=50, alpha=0.5)
plt.scatter(centroids[:, 0], centroids[:, 1], c='red', s=50)
plt.show()

Source: datatofish.com

Similar codes

1 vote157 views1 answer
Display 2d numpy array as image
Python
0 vote159 views1 answer
Sharpdevelop pause python code
Python
2 votes130 views1 answer
Django.core.exceptions.improperlyconfigured: wsgi application 'backend.wsgi.application' could not be loaded; error importing module.
Python Django
1 vote158 views2 answers
Input spaces seperated integers in python
Python
1 vote138 views2 answers
Django timezone settings
Python
-1 vote145 views1 answer
Lasso regression implementation python
Python
1 vote201 views1 answer
Django gmail smtp
Python Django
-1 vote142 views1 answer
Qtablewidget add row python
Python
0 vote144 views1 answer
How to get the percentage accuracy of a model in python
Python
1 vote96 views1 answer
Pandas series contains
Python