python - 如何在 Python 中将 dict 元素值作为列表

Question

我有一个列表。假设 [3,4,2,3,4,2,1,4,5]。

我需要根据元素的索引创建字典。

在这种情况下，我需要创建一个字典，如下所示:

{
   '3':[0,3],
   '4':[1,4,7],
   '2':[2,5],
   '1':[6],
   '5':[8]
}

其中元素值是所提供列表中键的索引。

我试过了。但是只能将值更改为整数。但无法将它们制成列表。

有没有办法只用 1 个 for 循环来做到这一点？

我试过的代码:

d=dict()
ll=[1,2,1,2,1,2,3,4,5,5,4,2,4,6,5,6,78,3,2,4,5,7,8,9,4,4,2,2,34,5,6,3]
for i,j in enumerate(ll):
    d[j].append(i)
print(d)

Answer 1

您可以使用 collections.defaultdict与 enumerate对于 O(n) 解决方案:

from collections import defaultdict

d = defaultdict(list)

A = [3,4,2,3,4,2,1,4,5]

for idx, val in enumerate(A):
    d[val].append(idx)

print(d)

defaultdict(list, {1: [6], 2: [2, 5], 3: [0, 3], 4: [1, 4, 7], 5: [8]})