python的sklearn库写kmeans算法代码实例

首先，导入必要的库： from sklearn.cluster import KMeans 然后，创建你的数据集：X = [[1, 2], [1, 4], [1, 0], [4, 2], [4, 4], [4, 0]] 接下来，实例化KMeans：kmeans = KMeans(n_clusters=2, random_state=0) 最后，调用fit函数： kmeans.fit(X) 输出结果：print(kmeans.labels_)输出：[0 0 0 1 1 1]