从2级嵌套字典

时间:2016-06-26 19:20:58

标签: python

我看起来最优雅的方式来获得这个:

{'i_1': {'a': 33, 't': 4}, 'i_2': {'a': 9, 't': 0}}

由此:

{'i_1': {'a': 33, 'b': 55, 't': 4}, 'i_2': {'a': 9, 'b': 11, 't': 0}}

每个内部词典都可以有很多a,b,...,z键。

现在我有了这个::

In [3]: {k:dict(a=d[k]['a'], t=d[k]['t']) for k in d.keys()}
Out[3]: {'i_1': {'a': 33, 't': 4}, 'i_2': {'a': 9, 't': 0}}

但它不是很优雅

3 个答案:

答案 0 :(得分:2)

使用items代替keys,您可以使代码更具可读性:

{k: dict(a=v['a'], t=v['t']) for k, v in d.items())

答案 1 :(得分:0)

import copy

def foo(d):
    d_copy = copy.deepcopy(d)

    for key in d_copy:
        print(key, d[key])
        if isinstance(d[key], dict):
            foo(d[key])
        if key == 'b':
            d.pop(key)

答案 2 :(得分:0)

你走了。此函数采用您指定格式的dict和必须从内部字典中删除的键列表:

def remove_inner_keys(data: dict, inner_keys_to_remove: list) -> dict:
    result = dict()

    for outer_key in data.keys():
        partial_result = dict()
        for inner_key in data[outer_key]:
            if inner_key not in inner_keys_to_remove:
                partial_result[inner_key] = data[outer_key][inner_key]
        result[outer_key] = partial_result

    return result

测试:

data = { 'i_1': { 'a': 33, 'b': 55, 't': 4 }, 'i_2': { 'a': 9, 'b': 11, 't': 0 } }
print(str(remove_inner_keys(data, ["b"])))

输出:

{'i_2': {'a': 9, 't': 0}, 'i_1': {'a': 33, 't': 4}}