python - 用字典理解过滤字典

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_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)}

差不多OK了，但是正在返回

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