1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 | import matplotlib.pyplot as plt import numpy as np from sklearn.cluster import KMeans X = np.array([ [6,3], [11,15], [17,12], [24,10], [20,25], [22,30], [85,70], [71,81], [60,79], [56,52], [81,91], [80,81]]) plt.scatter(X[:,0],X[:,1]) kmeans = KMeans(n_clusters=2) kmeans.fit(X) print(kmeans.cluster_centers_) print(kmeans.labels_) plt.scatter(X[:,0],X[:,1], c=kmeans.labels_, cmap='rainbow') | cs |