将默认字典更改为嵌套字典

时间:2018-10-18 06:10:48

标签: python dictionary defaultdict

我有一个字典(如下所示),值是默认字典,内部值是一个列表。我正在尝试将此字典转换为具有相同高级键的另一个字典,以及包含相同内部键和列表中“ 4”值之和的嵌套字典。

例如,第一个条目的预期输出为{'google':{'dog':1,'cat':2}}

{'google': defaultdict(<type 'list'>, {'dog': [4, 0], 'cat': [4, 4, 0]}), 'ibm': defaultdict(<type 'list'>, {'dog': [4], 'bird': [0]}), 'kaspersky': defaultdict(<type 'list'>, {}), 'amazon': defaultdict(<type 'list'>, {}), 'facebook': defaultdict(<type 'list'>, {}), 'sony': defaultdict(<type 'list'>, {}), 'microsoft': defaultdict(<type 'list'>, {}), 'apple': defaultdict(<type 'list'>, {})}

我正在使用下面的代码尝试执行此操作,但收到错误“ TypeError:列表索引必须是整数,而不是str”。

def swords_neg(dictionary):
    for key,value in dictionary.iteritems():
        company_sent_dict_three[key] = []
        for i,j in dict(value).iteritems():
            for sent in j:
                company_sent_dict_three[key][i] += sent

    print company_sent_dict_three

1 个答案:

答案 0 :(得分:3)

您可以使用以下嵌套的dict comprehension

{
    k: {k_: sum(x==4 for x in v_) for k_, v_ in v.items()} 
    for k, v in dictionary.items()
}
# {'google': {'dog': 1, 'cat': 2}, 
#  'ibm': {'dog': 1, 'bird': 0}, 
#  'kaspersky': {}, 
#  'amazon': {}, 
#  'facebook': {}, 
#  'sony': {}, 
#  'microsoft': {}, 
#  'apple': {}}

k, v是外部dict的键值对,k_, v_是每个内部defaultdict的键值对。 sum是一种便利工具,可以通过4

为每个最里面的列表计算sum(x==4 for x in v_).的出现