用唯一索引为列表建立索引

Question

我有一个清单说l = [10,10,20,15,10,20]。我想为每个唯一值分配一个特定的“索引”来获取[1,1,2,3,1,2]。

这是我的代码：

a = list(set(l))
res = [a.index(x) for x in l]

事实证明这很慢。

l具有1M个元素和100K个唯一元素。我也尝试过用lambda和sort进行地图操作，这没有帮助。这样做的理想方法是什么？

Answer 1

由于a.index(x)执行线性搜索，然后对中的每个元素执行线性搜索，因此导致代码变慢l。因此，对于每1M个项目，您要进行（最多）100K个比较。

将一个值转换为另一个值的最快方法是在地图中查找它。您需要创建地图，并填写原始值和所需值之间的关系。然后在列表中遇到另一个相同值时，从地图中检索该值。

这是一个通过的示例l。可能还有进一步优化的空间，以消除res在附加到它时重复进行重新分配的需要。

res = []
conversion = {}
i = 0
for x in l:
    if x not in conversion:
        value = conversion[x] = i
        i += 1
    else:
        value = conversion[x]
    res.append(value)