下面是一个用 Python 实现 K-Means 聚类的代码示例:
import numpy as np
def kmeans(X, k, max_iterations=100):
# 随机选择 k 个初始聚类中心
centers = X[np.random.choice(X.shape[0], k, replace=False), :]
for i in range(max_iterations):
# 计算每个样本到每个聚类中心的距离
distances = np.array([np.linalg.n