如何使用Python选择x级深度的JSON数据

import json

def select_json_depth(json_data, depth):
    result = []
    
    def traverse(data, current_depth):
        if current_depth == depth:
            result.append(data)
        elif isinstance(data, dict):
            for key, value in data.items():
                traverse(value, current_depth + 1)
        elif isinstance(data, list):
            for item in data:
                traverse(item, current_depth + 1)
    
    traverse(json_data, 0)
    return result

json_data = {
    "name": "John",
    "age": 30,
    "pets": [
        {
            "name": "Tom",
            "type": "cat"
        },
        {
            "name": "Jerry",
            "type": "mouse"
        }
    ],
    "address": {
        "street": "123 ABC Street",
        "city": "XYZ"
    }
}

depth = 2
selected_data = select_json_depth(json_data, depth)
print(selected_data)

[{'name': 'Tom', 'type': 'cat'}, {'name': 'Jerry', 'type': 'mouse'}, {'street': '123 ABC Street', 'city': 'XYZ'}]