用python代码写基于能量距离的聚类算法

你好，以下是基于能量距离的聚类算法的 Python 代码示例：

import numpy as np

def energy_distance(x, y):
    return np.sqrt(np.sum((x - y) ** 2))

def k_means_clustering(X, k):
    n_samples, n_features = X.shape
    centroids = X[np.random.choice(n_samples, k, replace=False)]
    labels = np.zeros(n_samples)
    while True:
        for i in range(n_samples):
            distances = [energy_distance(X[i], c) for c in centroids]
            labels[i] = np.argmin(distances)
        new_centroids = np.array([X[labels == j].mean(axis=0) for j in range(k)])
        if np.allclose(new_centroids, centroids):
            break
        centroids = new_centroids
    return labels

希望能对你有所帮助。