我想知道是否有一种更像Pythonic的方式来做以下事情,也许是使用字典理解:
A = some list
D = {}
for i,v in enumerate(A):
if v in D:
D[v].append(i)
else:
D[v] = [i]
使用默认值
:
from collections import defaultdict
D = defaultdict(list)
[D[v].append(i) for i, v in enumerate(A)]
使用setdefault
:
D = {}
[D.setdefault(v, []).append(i) for i, v in enumerate(A)]
我想不出在不整理数据的情况下使用词汇理解的任何方法:
from itertools import groupby
from operator import itemgetter
{v: ids for v, ids in groupby(enumerate(sorted(A)), itemgetter(1))}
表演:
from collections import defaultdict
from itertools import groupby
from operator import itemgetter
from random import randint
A = tuple(randint(0, 100) for _ in range(1000))
def one():
D = defaultdict(list)
[D[v].append(i) for i, v in enumerate(A)]
def two():
D = {}
[D.setdefault(v, []).append(i) for i, v in enumerate(A)]
def three():
{v: ids for v, ids in groupby(enumerate(sorted(A)), itemgetter(1))}
from timeit import timeit
for func in (one, two, three):
print(func.__name__ + ':', timeit(func, number=1000))
结果(一如既往,最简单的胜利):
one: 0.25547646999984863
two: 0.3754340969971963
three: 0.5032370890003222
您可以执行以下操作
d = collections.defaultdict(list)
for i,v in enumerate(A):
d[v].append(i)
您可以看到,生成的字典的值是list
s,在遍历时生成其元素。如果你坚持要做口述练习,你必须先找到所有的(值,[index])
,然后在[(k,[v])]
上做口述练习,这意味着额外的杂技没有任何好处。