使用循环过滤嵌套字典列表

Question

我有如下的字典列表。

它的结构类似于一棵树，每个节点都有一个任意数量的children。我想要 select 具有作为输入提供的匹配 'name' 的节点。

newdata = [
    {'name':"Oli Bob", 'location':"United Kingdom", '_children':[
        {'name':"Mary May", 'location':"Germany"},
        {'name':"Christine Lobowski", 'location':"France"},
        {'name':"Brendon Philips", 'location':"USA",'_children':[
            {'name':"Margret Marmajuke", 'location':"Canada"},
            {'name':"Frank Harbours", 'location':"Russia",'_children':[{'name':"Todd Philips", 'location':"United Kingdom"}]},
        ]},
    ]},
    {'name':"Jamie Newhart", 'location':"India"},
    {'name':"Gemma Jane", 'location':"China", '_children':[
        {'name':"Emily Sykes", 'location':"South Korea"},
    ]},
    {'name':"James Newman", 'location':"Japan"},
]

目前我正在这样做，使用下面

op = []

def getInfo(list_of_dict, name):
    for dict1 in list_of_dict:
        if dict1["name"]==name:
            op.append(dict1)
        if "_children" in dict1:
            getInfo(dict1["_children"], name)

getInfo(newdata, "Gemma Jane")  
print(op)

我想在没有外部变量（列表）的情况下执行相同的操作。

当我尝试使用以下函数做同样的事情时

def getInfo(list_of_dict, name):
    for dict1 in list_of_dict:
        if dict1["name"]==name:
            return dict1
        if "_children" in dict1:
            return getInfo(dict1["_children"], name)

op = getInfo(newdata, "James Newman")

它进入了一个递归循环，并没有为所有值提供正确的输出。

有什么解决这个问题的建议吗？

Answer 1

您可以使用递归生成器函数来搜索节点。例如：

def getInfo(data, name):
    if isinstance(data, list):
        for value in data:
            yield from getInfo(value, name)
    elif isinstance(data, dict):
        if data.get("name") == name:
            yield data
        yield from getInfo(data.get("_children"), name)


for node in getInfo(newdata, "Gemma Jane"):
    print(node)

打印：

{'name': 'Gemma Jane', 'location': 'China', '_children': [{'name': 'Emily Sykes', 'location': 'South Korea'}]}

使用循环过滤嵌套字典列表

filter a list of nested dictionary using loops

python

tree

recursion

dictionary