在没有for循环的情况下检查一个元素是否属于与字典中的键关联的列表

Question

问题提法：

检查一个元素是否属于与字典中的键关联的值列表以及return键的索引。

字典的每个键都与一个值列表相关联。例如：

498 : {1299,45,78}
875 :{45,104,200,300,456}

字典中键的个数是30,000

elements = [45,65,65,...,104,..., 875] # 元素由大约 900,000 个整数值组成

算法是做什么的？

查找字典中的每个元素和return它所属键的索引。

例如：

45属于键号498和875

我试过什么？

for elt in elements:
        Keys_indices_elt = [key for key, list in dictionary.items() if elt in list]

有什么问题吗？

嵌套循环的使用效率不高，return 30,000 个键和 900,000 个元素之间的映射需要大约 9 个小时。

有什么有效的方法可以解决吗？

Answer 1

你想要的是在处理循环之前建立一个反向索引。

它应该看起来像这样：

{
    1299: {498},
    45: {498, 875},
    78: {498},
    104: {875},
    # etc.
}

要构建它，您只需遍历字典并将字典中的值用作反向索引中的键。像这样：

rev_idx = {}
for k, v in my_dict.items():
    for e in v:
        if e in rev_idx:
            rev_idx[e].add(k)
        else:
            rev_idx[e] = {k}

这当然会占用一些内存和处理时间，但是您几乎可以同时获得 900,000 个元素中每个元素的答案。我希望通过这种方法，您的程序将运行持续大约两秒钟而不是 9 小时。

在没有for循环的情况下检查一个元素是否属于与字典中的键关联的列表

Check if an element belongs to a list associated with key in dictionary without for loop

dictionary

itertools

python-3.x