Python：遍历并修改包含列表的复杂字典

Question

我需要遍历一个复杂的字典并根据几个条件对其进行修改。

这是一个字典的基本示例：

a = {
   'x' : ["x1", "x2", 'x3'],
   'y' : [
       {'y1' : 1},
       {'y2' : 2.0},
       {'y3' : True},
       {'y4' : 99},
   ],
   'z' : {
      'x' : ["a1", "a2", 'a3'],
      'y' : [
           {'y1' : 66},
           {'y2' : False},
           {'y3' : 3},
           {'y4' : 4.3},
       ]
   },
   'y3' : "Delete Me"
}

遍历它也没有问题，比如这样：

pTypes = [str, bool, int, float]
def parse(d, path=[]):
    t = type(d)
    if t == dict:
        for k, v in d.items():
            parse(v, path + [k])
    elif t == list:
        for i, item in enumerate(d):
            parse(item, path + [i])
    elif t in pTypes:
        print(path, '=>', d)
    else:
        print('Error:', path, d, t)

parse(a, [])

我的难题在于，我需要删除所有键为'y3'的项或值大于50的项。此外，所有字符串都需要被'-'包围，如'x1'变为'-x1-'。

直接修改数据似乎并不起作用（没有持久性）：

...
elif t in pTypes:
   if t == str:
       d = '-'+d+'-'
...

而且，我还不知道如何在遍历字典本身的同时移除项目。

由于字典可能非常大，我更倾向于不对原始字典做一个“简化”的副本来进行操作。

有人能帮助我吗？

SilentGhost · Answer

确切的期望输出不是很清楚，但因为你正在遍历一个嵌套结构，最好生成一个新的对象。

保持你原来的递归逻辑，你应该在每一步都添加一个return来传播你的对象：

def parse(d):
    if isinstance(d, dict):
        return {k: parse(v) for k, v in d.items()
                if k != 'y3' and not 
                (isinstance(v, (int, float)) and v > 50)}
    elif isinstance(d, list):
        return [parse(x) for x in d]
    elif isinstance(d, str):
        return f'-{d}-'
    else:
        return d

out = parse(a)

输出结果为：

{
    'x': ['-x1-', '-x2-', '-x3-'],
    'y': [{'y1': 1}, {'y2': 2.0}, {}, {}],
    'z': {
        'x': ['-a1-', '-a2-', '-a3-'],
        'y': [{}, {'y2': False}, {}, {'y4': 4.3}]
    }
}

Matt Joiner · Answer

Mozway的回答如果不在意不满足条件的空字典被替换掉的话，是完全令人满意的。你没有提供很多细节，但我可以想象有些情况下你可能在乎这一点。例如，如果你原始的字典本身就包含空字典，那么你将无法区分那些原本就为空的字典（可能是缺失数据？）和那些因不符合条件而生成的空字典（表示已删除的数据）。

为了解决这个问题，你可以通过用其他东西“标记”被清空的字典来处理，比如在这个例子中，我使用字符串'This marks an emptied dict'来进行标记。为了确保我处理的是被清空的字典，我使用了拷贝，修改拷贝后的对象，并将其与原始对象进行比较。最后两行代码删除了被标记的项目，但你也可以选择保留它们。

import copy

def parse(anything):
    c = copy.copy(anything)
    
    if isinstance(anything, dict):
        for key, value in anything.items():
            if key == 'y3' or (isinstance(value, (int, float)) and value > 50):
                del c[key]
            else:
                c[key] = parse(value)
        if len(c) == 0 and len(anything) != 0:
            return 'This marks an emptied dict'
    elif isinstance(anything, list):
        c = [parse(item) for item in anything]
    elif isinstance(anything, str):
        c = '-' + c + '-'

    if isinstance(c, list):
        c = [item for item in c if not (isinstance(item, str) and item == 'This marks an emptied dict')]
    
    return c

parse(a)

输出结果为：

{
    'x': ['-x1-', '-x2-', '-x3-'],
    'y': [{'y1': 1}, {'y2': 2.0}],
    'z': {'x': ['-a1-', '-a2-', '-a3-'], 'y': [{'y2': False}, {'y4': 4.3}]}
}