python - 执行此操作的最有效的 pythonic 方法是什么 :

Question

我有一个大小为 N 的列表 L，其中列表的每个元素都在 0 到 K-1.
我想创建一个二维列表 S 和 K 行，这样 r^th 行包含所有这些索引 i，这样 L[i] == r。

例如，如果 L 是 [0, 0, 1, 3, 0, 3]
那么新列表 S 是 [[0, 1, 4], [2], [], [3, 5]]

解决方案当然应该是 O(N)，它也应该尽可能高效(阅读:列表上没有无用的追加操作)

Answer 1

>>> L = [0, 0, 1, 3, 0, 3]
>>> import collections
>>> d = collections.defaultdict(list)
>>> for index, item in enumerate(L):
...   d[item].append(index)
... 
>>> d
defaultdict(<type 'list'>, {0: [0, 1, 4], 1: [2], 3: [3, 5]})
>>> [d[i] for i in xrange(1 + max(d))]
[[0, 1, 4], [2], [], [3, 5]]