[英]Dictionary Comprehension for list values
我想知道是否有更多Pythonic方式來執行以下操作,可能使用字典理解:
A = some list
D = {}
for i,v in enumerate(A):
if v in D:
D[v].append(i)
else:
D[v] = [i]
使用defaultdict
:
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,其中的元素將在遍歷時生成。 如果你堅持做一個dict comp,你必須首先找到所有的(value, [indices])
,然后在[(k,[v])]
上做一個dict comp,這就意味着沒有任何好處的額外雜技。
聲明:本站的技術帖子網頁,遵循CC BY-SA 4.0協議,如果您需要轉載,請注明本站網址或者原文地址。任何問題請咨詢:yoyou2525@163.com.