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

x=[4,5,10,4,3,11,14,6,10,12]
y=[21,19,24,17,16,25,24,22,21,21]

data=list(zip(x,y))
print(data)

inertia=[]
for i in range(1,11):
    kmeans=KMeans(n_clusters=i)
    kmeans.fit(data)
    inertia.append(kmeans.inertia_)
plt.plot(range(1,11),inertia,marker='o')
plt.title("Elbow Method")
plt.xlabel("no of clusters")
plt.ylabel("inertias")
plt.show()

kmean=KMeans(n_clusters=2)
kmean.fit(data)

plt.scatter(x,y,c=kmean.labels_)
plt.show()