首页 > 解决方案 > 从嵌套字典中删除值为零的项目

问题描述

这是一个嵌套字典:

fdict = {}
fdict['apple'] = {}
fdict['banana'] = {}

fdict['apple']['green'] = 5
fdict['apple']['red'] = 0
fdict['banana']['light_yellow'] = 10
fdict['banana']['dark_yellow'] = 0
fdict['appraisal round'] = 1

{'apple': {'green': 5, 'red': 0},
 'banana': {'light_yellow': 10, 'dark_yellow': 0},
 'appraisal round': 1}

从这个嵌套字典中删除值为零的键和值对的最pythonic方法是什么,从而得到以下字典结果:

{'apple': {'green': 5}, 'banana': {'light_yellow': 10}, 'appraisal round': 1}

注意:appraisal round键本身没有字典作为值。

这是我迄今为止实施的:

overall_dict = {}
for key in [key for key in fdict.keys() if key != 'appraisal round']:
    new_dict = {k:v for k,v in fdict[key].items() if v != 0}
    overall_dict[key] = new_dict
overall_dict['appraisal round'] = 1

但是,使用临时字典,构建一个全新的字典并重新添加appraisal round似乎不是一个干净的方法。有没有办法更有效地修改现有字典?

标签: pythonpython-3.xdictionary

解决方案


假设您不知道具有非 dict 类型的键,解决方案可能是:

fdict = {}
fdict['apple'] = {}
fdict['banana'] = {}

fdict['apple']['green'] = 5
fdict['apple']['red'] = 0
fdict['banana']['light_yellow'] = 10
fdict['banana']['dark_yellow'] = 0
fdict['appraisal round'] = 1

def non_zero_value(item):
  k, v = item
  return v != 0

result = {}
for k, v in fdict.items():
  if isinstance(v, dict):
    result[k] = dict(filter(non_zero_value, v.items()))
  else:
    result[k] = v

print(result)
# {'apple': {'green': 5}, 'banana': {'light_yellow': 10}, 'appraisal round': 1}

推荐阅读