首先，导入必要的库： 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]
版权声明：本文为weixin_42577243原创文章，遵循 CC 4.0 BY-SA 版权协议，转载请附上原文出处链接和本声明。