from sklearn.cluster import kmeans kmeans = KMeans(3) # 或者kmeans = KMeans(n_cluster=3) kmeans.fit(data)
