从sklearn
集群标签输出中制作集群列表,可以通过以下步骤完成:
from sklearn.cluster import KMeans
import numpy as np
X
,每个样本有多个特征:X = np.array([[1, 2], [1, 4], [1, 0], [4, 2], [4, 4], [4, 0]])
KMeans
算法对数据进行聚类,指定所需的聚类数量n_clusters
:kmeans = KMeans(n_clusters=2)
kmeans.fit(X)
labels = kmeans.labels_
clusters = {}
for i, label in enumerate(labels):
if label not in clusters:
clusters[label] = []
clusters[label].append(i)
现在,clusters
字典中的键表示聚类标签,对应的值是一个列表,包含属于该聚类的样本的索引。
这是一个简单的示例,展示了如何从sklearn
集群标签输出中制作集群列表。在实际应用中,你可能需要根据具体的需求进行适当的调整和扩展。
领取专属 10元无门槛券
手把手带您无忧上云