将向量的值排列到 Python 中的列表

Question

我在我的数据上使用运行 k-means 算法，我的标签输出如下所示：

[0 5 8 6 1 3 3 2 2 5 5 6 1 1 3 3 1 8 8 3 3 1 1 1 1 5 2 5 1 1 7 3 6 4 3 3 8
 1 3 3 5 1 8 8 1 8 7 1 1 8 6]

这个向量包含点索引的簇号，例如第一个值是簇号。 0 表示点索引 0，向量的第二个值表示它是簇号。 5和点索引1属于它。

我想要集群的子集：喜欢：

cluster no 0 = { its index numbers}
cluster no 1 = { its index numbers}
..
cluster no 8 = { its index numbers}

例如向量的第一个值为 5，我需要列出该向量的所有值为 5 的索引，反之亦然。我希望每个值都有自己的索引列表。

所以值 5 的列表应该是：

集群 5 = [ 1,9,10,25,27....

和其他值的所有输出，最终输出应该是8个列表。

Answer 1

如果你愿意使用 numpy，这很容易用 numpy.where

cluster5, = numpy.where( array == 5 )

在 'pure' python 你可以这样做：

cluster5 = [i for i in range(len(array)) if array[i]==5]

Answer 2

这样就可以了，使用 enumerate:

array = [0,5,8,6,1,3,3,2,2,5,5,6,1,1,3,3,1,8,8,3,3,1,1,1,1,5,2,5,1,1,7,3,6,4,3,3,8,1,3,3,5,1,8,8,1,8,7,1,1,8,6]

for j in range(9):
    print("%i: %s"%(j,[i for i,x in enumerate(array) if x == j]))

Answer 3

基于enumerate and EAFP 方法的简单解决方案。

def cluster(seq):
    out = {}
    for index, value in enumerate(seq):
        try:
            out[value].append(index)
        except KeyError:
            out[value] = [index]
    return out

data = [2, 3, 4, 4, 3, 1]
result = cluster(data)
assert result[2] == [0]
assert result[3] == [1, 4]
assert result[4] == [2, 3]
assert result[1] == [5]

将向量的值排列到 Python 中的列表

arranging the values of vector to lists in Python

python

cluster-analysis

vector

k-means