kmeans代码python

当然，我可以提供一份简单的 kmeans 代码，但请注意这只是一个例子。因为 kmeans 是一个经典的机器学习算法，有很多开源的库可以使用，如 sklearn。

import numpy as np

def kmeans(X, k, iterations):
    centers = X[np.random.choice(len(X), k, replace=False), :]
    
    for i in range(iterations):
        distances = np.sqrt(((X - centers[:, np.newaxis])**2).sum(axis=2))
        labels = np.argmin(distances, axis=0)
        for j in range(k):
            new_center = X[labels==j].mean(axis=0)
            centers[j] = new_center
    
    return labels, centers

这是一个简单的 kmeans 实现。它接受一个数据集 X，聚类数目 k 和迭代次数 iterations 作为参数。它返回一个标签向量和聚类中心。