使用字典元组键的最后两个元素求和

Question

我有名单

A = [(i,j,k,l,m)]
B = [(l,m,k)]

和字典

C = {(i,j,k,l,m): val}
D = {(l,m,k): other_val}

我想创建一个E字典，以便

E = {(i,j,k): C[(i,j,k,l,m)]*D[(l,m,k)]}

假设列表和字典中的所有索引约定都匹配。我有下面的非肾盂，非常缓慢的解决方案。对于非常大的A大小（例如，500万行），是否有任何Pythonic方法可以快速执行此操作

E = {}
for i,j,k,l,m in A:
    E[i,j,k] = sum(
        C[i,j,k,l,m] * D[l2,m2,k2] 
        for l2,m2,k2 in B if l2==l and m2==m and k2==k)

下面是生成样本数据集的代码，该样本数据集接近要处理的实际大小

import numpy as np
np.random.seed(1)

Irange = range(50)
Jrange = range(10)
Krange = range(80)
Lrange = range(8)
Mrange = range(18)

A = [
    (i,j,k,l,m)
    for i in Irange
    for j in Jrange
    for k in Krange
    for l in Lrange
    for m in Mrange]
B = [
    (l,m,k)
    for k in Krange
    for l in Lrange
    for m in Mrange]

C = {key: np.random.uniform(1,10) for key in A}

D = {key: np.random.uniform(0,1) for key in B}

E = {}
for i,j,k,l,m in A:
    E[i,j,k] = sum(
        C[i,j,k,l,m] * D[l2,m2,k2]
        for l2,m2,k2 in B if l2==l and m2==m and k2==k)

Answer 1

from collections import defaultdict

b = set(B)  # O(#B)
E = defaultdict(float)
for i,j,k,l,m in A: # O(#A)
    if (l, m, k) in b:
        E[i,j,k] += C[i,j,k,l,m] * D[l, m, k]

这种方法具有O(#A + #B)复杂性。 撇开正确性问题不谈，天真的实现是O(#A * #B)