| X1 | X2 |
|---|---|
| 2 | 10 |
| 2 | 5 |
| 8 | 4 |
| 5 | 8 |
| 7 | 5 |
| 6 | 4 |
| 1 | 2 |
| 4 | 9 |
X = [
[2, 10],
[2, 5],
[8, 4],
[5, 8],
[7, 5],
[6, 4],
[1, 2],
[4, 9]
]from sklearn.cluster import KMeans
kmeans = KMeans(n_clusters = 3, n_init='auto')
kmeans.fit(X)
kmeans.labels_
kmeans.cluster_centers_
kmeans.predict([[1, 3], [2, 8]])