基于索引和值创建列表列表

Question

我得到了一个10^6的正整数列表，列表中的项目总数最多为10**4。我想按如下所述的特定顺序存储这些值

我想遍历这个列表，并将item的索引值放在对应于该项的内部列表中

示例

给定列表：[3,0,1,1]

输出：[[1],[2,3],[],[0]]#预期输出

基本上，输出列表应该包含与给定列表中的项值对应的索引列表，该列表应该包含索引值

我想出了这个解决办法

sequence = [[]] * 4
given = [3,0,1,1]
index = 0
for num in given:
    sequence[num].append(index)
    index+=1
print(sequence)

但是我得到的输出是[[0, 1, 2, 3], [0, 1, 2, 3], [0, 1, 2, 3], [0, 1, 2, 3]]，我不知道为什么

Answer 1

这样就可以了：

a = [3, 0, 1, 1]
[[j for j, k in enumerate(a) if k == i] for i, _ in enumerate(a)]
#[[1], [2, 3], [], [0]]