Question

我在使用dict理解时过滤dict的dict有问题。

我有词：

clients = {
        'Shop': {'url' : 'url_v', 'customer' : 'cumoster_v', 
                 'some_other_key1' : 'some_value'},
        'Gym': {'url' : 'url_v1', 'customer_1' : 'customer_v1', 'customer_2': 'customer_v2',
                 'some_other_key2' : 'some_value'},
        'Bank': {'url' : 'url_v2', 'customer_3' : 'customer_v3',    
                  'some_other_key3' : 'some_value'}
}

我想制作另一个只有＆＃39;客户的字典。*＆＃39;键。所以，新的dict应该是这样的：

dict_only_cust = {
               'Shop': {'customer' : 'cumoster_v'},
               'Gym': {'customer_1' : 'customer_v1', 'customer_2': 'customer_v2'},
               'Bank': {'customer_3' : 'customer_v3'}
}

由于我是名单和词汇理解的忠实粉丝，我想知道是否可以用这个做到这一点。

到目前为止，我已写过：

dict_only_cust = {v.pop(t) for k, v in clients.items()
                for t, vv in v.items()
                if not re.match('.*customer.*', t)}

代码以'RuntimeError: dictionary changed size during iteration'

失败

我第二次尝试：

dict_only_cust = {k:{t: vv} for k, v in clients.items()
                for t, vv in v.items()
                if re.match('.*customer.*', t)}

几乎可以，但它正在返回

dict_only_cust = {
              'Shop' : {'customer' : 'cumoster_v'},
              'Gym' : {'customer_1' : 'customer_v1'},
              'Bank' : {'customer_3' : 'customer_v3'}
}

如何使用dict理解来解决这个问题？我使用的是python 3.4。

谢谢！

Answer 1

>>> {key:{k:v for k,v in dic.items() if 'customer' in k} for key,dic in clients.items()}
{'Shop': {'customer': 'cumoster_v'}, 'Gym': {'customer_2': 'customer_v2', 'customer_1': 'customer_v1'}, 'Bank': {'customer_3': 'customer_v3'}}

Answer 2

{k:{k1:v1 for  k1,v1 in v.items() if k1.startswith('customer')} for k, v in clients.items()}

输出：

{'Shop': {'customer': 'cumoster_v'}, 'Gym': {'customer_2': 'customer_v2', 'customer_1': 'customer_v1'}, 'Bank': {'customer_3': 'customer_v3'}}

用词典理解过滤dict的词典

2 个答案: