繁体   English   中英

python中集合的字典元素

[英]Elements of dict of sets in python

我有一本这样的字典:

dict1 = {0: set([1, 4, 5]), 1: set([2, 6]), 2: set([3]), 3: set([0]), 4: set([1]), 5: set([2]), 6: set([])}

从这个字典中,我想建立另一个字典,该字典计算dict1中每个其他值中键的出现次数,即结果应该是:

result_dict = {0: 1, 1: 2, 2: 2, 3: 1, 4: 1, 5: 1, 6: 1}

我的代码是这样的:

dict1 =  {0: set([1, 4, 5]), 1: set([2, 6]), 2: set([3]), 3: set([0]), 4: set([1]), 5:set([2]), 6: set([])}
result_dict = {}
for pair in dict1.keys():
    temp_dict = list(dict1.keys())

    del temp_dict[pair]
    count = 0
    for other_pairs in temp_dict :
        if pair in dict1[other_pairs]:
            count = count + 1
    result_dict[pair] = count  

此代码的问题在于,处理大量数据时,它的运行速度非常慢。 另一尝试是在一行中,如下所示:

result_dict = dict((key ,dict1.values().count(key)) for key in dict1.keys())  

但这会给我错误的结果,因为dict1的值已设置:

{0: 0, 1: 0, 2: 0, 3: 0, 4: 0, 5: 0, 6: 0}

提前非常感谢

我想,对于第一个步骤,我想知道那里有哪些值:

all_values = set().union(*dict1.values())

然后,我尝试计算每个值出现了多少次:

result_dict = {}
for v in all_values:
    result_dict[v] = sum(v in dict1[key] for key in dict1)

另一种方法是使用collections.Counter

result_dict = Counter(v for set_ in dict1.values() for v in set_)

这可能比我的第一个解决方案“更干净”-但这确实涉及嵌套的理解,这可能有点难以理解。 它确实可以工作:

>>> from collections import Counter
>>> dict1
{0: set([1, 4, 5]), 1: set([2, 6]), 2: set([3]), 3: set([0]), 4: set([1]), 5: set([2]), 6: set([])}
>>> result_dict = Counter(v for set_ in dict1.values() for v in set_)

只需使用dict1的键创建第二个字典,值从0开始。 然后遍历dict1集合中的值,并随着dict1增加而递增result_dict值。 运行时为O(n) ,其中ndict1组中值的dict1

dict1 = {0: set([1, 4, 5]), 1: set([2, 6]), 2: set([3]), 3: set([0]), 4: set([1]), 5:set([2]), 6: set([])}
result_dict = dict.fromkeys(dict1.keys(), 0)
# {0: 0, 1: 0, 2: 0, 3: 0, 4: 0, 5: 0, 6: 0}

for i in dict1.keys():
    for j in dict1[i]: 
        result_dict[j] += 1

print result_dict
# {0: 1, 1: 2, 2: 2, 3: 1, 4: 1, 5: 1, 6: 1}

暂无
暂无

声明:本站的技术帖子网页,遵循CC BY-SA 4.0协议,如果您需要转载,请注明本站网址或者原文地址。任何问题请咨询:yoyou2525@163.com.

 
粤ICP备18138465号  © 2020-2024 STACKOOM.COM