python基于键匹配合并字典的最快方法

Question

我有 2 个字典列表。列表 A 长 34,000，列表 B 长 650,000。我基本上是根据键匹配将所有列表 B 指令插入到列表 A 指令中。目前，我正在做显而易见的事情，但这需要很长时间(说真的，就像一天一样)。一定有更快的方法!

for a in listA:
    a['things'] = []
    for b in listB:
        if a['ID'] == b['ID']:
            a['things'].append(b)

Answer 1

from collections import defaultdict
dictB = defaultdict(list)
for b in listB:
    dictB[b['ID']].append(b)

for a in listA:
    a['things'] = []
    for b in dictB[a['ID']]:
        a['things'].append(b)

这会将您的算法从 O(n*m) 变为 O(m)+O(n)，其中 n=len(listA)，m=len(listB)

基本上，它通过“预先计算”listB 中的哪些字典与每个“ID”匹配，避免了针对 listA 中的每个字典遍历 listB 中的每个字典