python - 如何仅在 numpy 数组的“一维”上应用 K-means++ 并按原样保留另一维

Question

我在形状 (2048, 475)上应用了 k-means 算法。我使用了以下代码

from sklearn.cluster import KMeans

model = KMeans(n_clusters=36, random_state=42)

model.fit(abc)

labels = model.labels_

print(labels)

我只想将 K-means 聚类应用于值为 475 的维度（从 475 生成 36 个集群）并按原样保留第一个维度。有人可以帮我解决这个问题吗？