如何给出聚集数组的索引python?

2024-05-19 14:00:06 发布

您现在位置:Python中文网/ 问答频道 /正文

我尝试运行这个K-Medoids python实现代码

from sklearn.metrics.pairwise import pairwise_distances
import numpy as np

import kmedoids

# 3 points in dataset
data = np.array([[1,1], 
                [2,2], 
                [10,10]])

# distance matrix
D = pairwise_distances(data, metric='euclidean')

# split into 2 clusters
M, C = kmedoids.kMedoids(D, 2)

print('medoids:')
for point_idx in M:
    print( data[point_idx] )

print('')
print('clustering result:')
for label in C:
    for point_idx in C[label]:
        print('label {0}: {1}'.format(label, data[point_idx]))

https://github.com/someus/kmedoids)并显示:

medoids:

[1 1]

[10 10]

clustering result:

label 0: [1 1]

label 0: [2 2]

label 1: [10 10]

但我想在它的聚类结果上插入索引,如下所示:

medoids:

[1 1], index 1

[10 10], index 3

clustering result:

label 0: [1 1], index 1

label 0: [2 2], index 2

label 1: [10 10], index 3

有人知道怎么做吗? 谢谢你


Tags: inimportfordataindexresultlabelpoint