Question

我有一个很大的整数未排序列表，数字可能会重复。我想创建另一个列表，它是从第一个列表开始的索引子列表的列表，从max元素到min，按降序排列。例如，如果我有这样的列表：

list = [4, 1, 4, 8, 5, 13, 2, 4, 3, 7, 14, 4, 4, 9, 12, 1, 6, 14, 10, 8, 6, 4, 11, 1, 2, 11, 3, 9]

输出应为：

indexList = [[10, 17], [5], [14], [22, 25], [18], [13, 27], [3, 19], [9], [16, 20], [4], [0, 2, 7, 11, 12, 21], [8, 26], [6, 24], [1, 15, 23]]

其中，[10, 17]是＆＃39; 14＆＃39;的指数。存在等......

在下面分享我的代码。使用cProfile对大约9000个元素的列表进行分析大约需要6秒钟。

def indexList(list):
    # List with sorted elements
    sortedList = sorted(list, reverse = True)

    seen = set()
    uSortedList = [x for x in sortedList if x not in seen and not seen.add(x)]

    indexList = []
    for e in uSortedList:
        indexList.append([i for i, j in enumerate(list) if j == e])

    return indexList

Answer 1

你走了：

def get_list_indices(ls):
    indices = {}
    for n, i in enumerate(ls):
        try:
            indices[i].append(n)
        except KeyError:
            indices[i] = [n]
    return [i[1] for i in sorted(indices.items(), reverse=True)]

test_list = [4, 1, 4, 8, 5, 13, 2, 4, 3, 7, 14, 4, 4, 9, 12, 1, 6, 14, 10, 8, 6, 4, 11, 1, 2, 11, 3, 9]
print(get_list_indices(test_list))

基于一些非常基本的测试，它的速度大约是您发布的代码的两倍。

从max到min元素开始的大型整数列表中查找元素索引的有效方法

1 个答案: