递归字典搜索

Question

我正在尝试创建一个函数，它将嵌套数组（dict/list 以任何顺序排列）和一个键名作为参数，return 列表中该键的所有值。

my_key = "Items"

my_dict = [{'z': 0, 'x': 0, 'y': 0, 'Items': [{'Slot': 1, 'id': 'minecraft:rail', 'Count': 1}, {'Slot': 2, 'id': 'minecraft:white_shulker_box', 'tag': {'BlockEntityTag': {'id': 'minecraft:shulker_box', 'Items': [{'Slot': 0, 'Count': 1, 'tag': {'Items': [{'id': 'minecraft:amethyst_shard', 'Count': 1}]}, 'id': 'minecraft:bundle'}]}}, 'Count': 1}]}]

def recursive_lookup(data, key):
    if isinstance(data, list):
        for i in data:
            recursive_lookup(i, key)
    elif isinstance(data, dict):
        for i, v in data.items():
            if i == key:
                print(f'{v = }')
            if isinstance(v, list) or isinstance(v, dict): recursive_lookup(v, key)
print(recursive_lookup(my_dict, my_key))

目前它打印出在 print(f'{v = }') 找到的项目。如何将它们存储在列表中并作为函数传递 return?

Answer 1

您可以使用 .extend() 将递归调用的结果连接到一个列表。

def recursive_lookup(data, key):
    values = []
    if isinstance(data, list):
        for i in data:
            values.extend(recursive_lookup(i, key))
    elif isinstance(data, dict):
        for i, v in data.items():
            if i == key:
                values.append(v)
            if isinstance(v, list) or isinstance(v, dict):
                values.extend(recursive_lookup(v, key))
    return values

Answer 2

您可以保留一个运行列表：

def recursive_lookup(data, key):
    lst = []
    if isinstance(data, list):
        for i in data:
            lst.append(recursive_lookup(i, key))
    elif isinstance(data, dict):
        for i, v in data.items():
            if i == key:
                lst.append([v])
            if isinstance(v, list) or isinstance(v, dict): lst.append(recursive_lookup(v, key))
    return lst

print(recursive_lookup(data, 'Items'))

Answer 3

通过使用标准中的 json 模块，您可以 而无需 任何显式递归库（假设您的数据可以序列化为该格式）。这是因为 JSON 解码器支持一个 object_hook 参数，这是一个函数，它会在每次遇到字典时调用。

基本思想是通过此参数指定一个函数，该函数仅“监视”正在解码的内容并检查它是否有 sought-after 密钥。

我的意思是：

import json

my_key = "Items"
my_dict = [{'z': 0, 'x': 0, 'y': 0, 'Items': [{'Slot': 1, 'id': 'minecraft:rail', 'Count': 1}, {'Slot': 2, 'id': 'minecraft:white_shulker_box', 'tag': {'BlockEntityTag': {'id': 'minecraft:shulker_box', 'Items': [{'Slot': 0, 'Count': 1, 'tag': {'Items': [{'id': 'minecraft:amethyst_shard', 'Count': 1}]}, 'id': 'minecraft:bundle'}]}}, 'Count': 1}]}]

def lookup(data, key):
    results = []

    def decode_dict(a_dict):
        try:
            results.append(a_dict[key])
        except KeyError:
            pass
        return a_dict

    json_repr = json.dumps(data)  # Convert to JSON format.
    json.loads(json_repr, object_hook=decode_dict)  # Return value ignored.
    return results

from pprint import pprint
pprint(lookup(my_dict, my_key), sort_dicts=False)

Pretty-printed 结果列表：

[[{'id': 'minecraft:amethyst_shard', 'Count': 1}],
 [{'Slot': 0,
   'Count': 1,
   'tag': {'Items': [{'id': 'minecraft:amethyst_shard', 'Count': 1}]},
   'id': 'minecraft:bundle'}],
 [{'Slot': 1, 'id': 'minecraft:rail', 'Count': 1},
  {'Slot': 2,
   'id': 'minecraft:white_shulker_box',
   'tag': {'BlockEntityTag': {'id': 'minecraft:shulker_box',
                              'Items': [{'Slot': 0,
                                         'Count': 1,
                                         'tag': {'Items': [{'id': 'minecraft:amethyst_shard',
                                                            'Count': 1}]},
                                         'id': 'minecraft:bundle'}]}},
   'Count': 1}]]

递归字典搜索

Recursive dictionary searching

python

search

dictionary