ValueError：無法將輸入數組從形狀（0,2）廣播到形狀（2，）

Question

我正在學習 K 方法，但它遇到了一些麻煩。 有人可以幫忙嗎？

def find_new_centroid(X, labels, j):
    centers = np.zeros((j, X.shape[1]))
    for k in range(j):
        # collect all points assigned to the k-th cluster
        Xj = X[labels == k, :]
        # take average
        centers[k, :] = np.mean(Xj, axis=0)
    return centers  

Answer 1

確保labels的類型為numpy.ndarray而不是 Python list