Question

我在Python字典下面有source

d1 = {
  'a': 1,
  'b': 2,
  'c': [{'d': 3, 'e': 4, 'un': 'wanted1', 'dont': 'needthis1'},
        {'d': 5, 'e': 6, 'un': 'wanted2', 'dont': 'needthis2'}]
  'xyz': 'abc',
  'zxy': 'cab',
  'wva': 'xyw'
}

我想将某些特定键的值复制到target字典下方的另一格中

d2 = {
  'some_attr_1': 1,  
  'some_attr_x': 2,
  'attr_some_z': [{'attr_x': 3, 'attrib': 4},
                  {'attr_x': 5, 'attrib': 6}]
}

注意：

我对source中的所有属性不感兴趣例如：我不需要键xyz，zxy等
基本上，source中某些键的希望值将被映射 target字典中的不同键。

我目前的方法是在source和target字典键之间建立映射。

attr_map1 = {
   'some_attr_1': 'a',
   'some_attr_x': 'b'
}

attr_map2 = {
  'attr_x': 'd',
  'attrib': 'e',
}

d2 = dict()
for k, v in attr_map1.items():
   d2[k] = d1[v]

l1 = list()
for d_elem in d1['c']:
   temp_dict = dict()
   for k, v in attr_map2.items():
       temp_dict[k] = d_elem[v]
   l1.append(temp_dict)
d2['attr_some_z'] = l1

是否有替代，更好和快速的方法来实现这一目标？

我正在寻找Python 2.7中的解决方案。

谢谢

Answer 1

您可以使用递归：

d1 = {'a': 1, 'b': 2, 'c': [{'d': 3, 'e': 4}, {'d': 5, 'e': 6}]}
def build(d):
  return {f't_{a}':b if not isinstance(b, (dict, list)) else 
           list(map(build, b)) if isinstance(b, list) else build(b) for a, b in d.items()}

print(build(d1))

输出：

{
't_a': 1, 
 't_b': 2, 
 't_c': [
    {'t_d': 3, 't_e': 4}, 
    {'t_d': 5, 't_e': 6}
   ]
}

编辑：要在Python2中运行此解决方案，请将f-string替换为简单的串联：

d1 = {'a': 1, 'b': 2, 'c': [{'d': 3, 'e': 4}, {'d': 5, 'e': 6}]}
def build(d):
   return {'t_'+a:b if not isinstance(b, (dict, list)) else 
       list(map(build, b)) if isinstance(b, list) else build(b) for a, b in d.items()}

复制多级Python字典

1 个答案: