Python按顺序缩小列表

2023年6月18日 237次阅读

假设我有一个列表[10,5,7],我想把它减少到表示相对订单[3,1,2]的东西.我不知道如何在
Python中转换它. 最佳答案

start = [10, 5, 7]

解决

step1 = sorted(start) #if you have duplicates, sorted(set(start)) to uniquify

制作一个查找表

lookup = {v:i for i,v in enumerate(step1,1)}

制作新名单.

[lookup[x] for x in start]
Out[9]: [3, 1, 2]

这是O(nlogn)而不是使用重复索引搜索的O(n ** 2)解决方案.时序：

test = [randrange(0,10000) for _ in range(10000)]

def f():
    std = sorted(test)
    return [std.index(e)+1 for e in test]

def g():
    step1 = sorted(test)
    lookup = {v:i for i,v in enumerate(step1,1)}
    return [lookup[x] for x in test]

%timeit f()
1 loops, best of 3: 1.17 s per loop

%timeit g()
100 loops, best of 3: 6.58 ms per loop
点赞