如何獲取唯一值的索引列表？

Question

有這樣的清單

lst = ['a','b','a','c','a','b','b']

我想獲取每個唯一元素的索引列表以獲取此信息：

indexes = {
            'a': [0,2,4],
            'b': [1,5,6],
            'c': [3]
          }

這是我當前的代碼，但我只獲取每個元素的第一個索引。

indexes= dict()
for el in lst:
    indexes[el] = [lst.index(el)]

>>> indexes
{'a': [0], 'b': [1], 'c': [3]}

謝謝你的幫助。

Answer 1

您的代碼的問題是您一次又一次地覆蓋相同的鍵，但使用不同的list ，因此您的最終字典僅包含一個list 。

您可以通過使用defaultdict來避免這種行為。

from collections import defaultdict

lst = ["a", "b", "a", "c", "a", "b", "b"]
lst = [c for c in lst if c.strip()]  # this will remove empty strings

indexes = defaultdict(list)

for index, char in enumerate(lst):
    indexes[char].append(index)

indexes = dict(indexes)
print(indexes)

Output：

{'a': [0, 2, 4], 'b': [1, 5, 6], 'c': [3]}