python - 为列表条目提供索引

Question

我有一个如下所示的 Python 列表:

A1 = ['a','a','a','foo','c','d','a','e','bar','bar','bar','e','d','d']

我想把它改造成这个...

A2 = [1,1,1,2,3,4,1,5,6,6,6,5,4,4]

...其中 A1 中的条目按顺序获取并在 A2 中给出增量索引。

在 Python 中是否有直接的方法来做到这一点？

Answer 1

index_map = {}
result = []

i = 0 # or 1 or whatever

for value in A1:
    if value not in index_map:
        index_map[value] = i
        i = i + 1
    result.append(index_map[value])