如何以最有效的方式在嵌套字典中获取指定的键值？

Question

有一个嵌套的词典，如：

data_dict = {
 "picture":"xxx.jpg",
 "link_data":{
     "picture":"xxxxx.jpg", 
     ... 
     "child_attachments":{
            "picture":"xxxxx.jpg", 
              ... 
       }
  }
  ...
}

问题出现在字典的每个层面，关键picture可能存在，如何以最有效的方式获得图片的价值？

这是我的试用，但失败了：

def get_picture_url(data):
    for key, value in data.items():
        if key == "picture":
            return data[key]
        else:
            if isinstance(value, dict):
                return get_picture_url(value)

get_picture_url(data_dict)

Answer 1

您没有检查对get_picture_url的递归调用的返回值。

这应该是你的dict中最顶级的图片：

def get_picture_url(data, picture_key="picture"):
    if not isinstance(data, dict):
        return None
    picture_url = data.get(picture_key)
    if picture_url is not None:
        return picture_url
    for value in data.values():
        picture_url = get_picture_url(value)
        if picture_url is not None:
            return picture_url
    return None

Answer 2

这应该适用于具有类似JSON结构的任意嵌套字典的一般情况：

def get_picture(data):
    # you can remove this case if the
    # input doesn't contain lists
    if isinstance(data, list):
        ans = []
        for e in data:
            ans += get_picture(e)
        return ans
    elif not isinstance(data, dict):
        return []
    else:
        ans = []
        for k, v in data.items():
            if k == 'picture':
                ans.append(v)
            else:
                ans += get_picture(v)
        return ans

它将遍历数据结构的所有级别，查找名为'picture'键并在单个输出列表中累积其所有值。 如果您确定输入中没有列表，我们可以稍微简化一下解决方案：

def get_picture(data):
    ans = []
    if isinstance(data, dict):
        for k, v in data.items():
            if k == 'picture':
                ans.append(v)
            else:
                ans += get_picture(v)
    return ans

无论哪种方式，它都可以按预期的方式输入样本：

data_dict = {
 "picture":"xxx.jpg",
 "link_data":{
    "picture":"xxxx.jpg",
    "child_attachments":{
        "picture":"xxxxx.jpg"
    }
  }
}

get_picture(data_dict)
=> ['xxx.jpg', 'xxxx.jpg', 'xxxxx.jpg']