利用词典理解过滤词典

Question

我在使用字典理解时过滤字典有问题。

我有句名言：

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'}
}

我想再做一个只有“customer.*”键的字典。因此，新的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'}
}

如何使用字典理解来解决这个问题？我使用的是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'}}