python json从找到的元素中捕获整个节点

import json

def find_element(json_data, target_element):
    if isinstance(json_data, dict):
        for key, value in json_data.items():
            if key == target_element:
                return json_data
            elif isinstance(value, (dict, list)):
                result = find_element(value, target_element)
                if result is not None:
                    return result
    elif isinstance(json_data, list):
        for item in json_data:
            result = find_element(item, target_element)
            if result is not None:
                return result

# 示例 JSON 数据
json_data = {
    "name": "John",
    "age": 30,
    "address": {
        "street": "123 Street",
        "city": "New York"
    },
    "friends": [
        {
            "name": "Alice",
            "age": 25
        },
        {
            "name": "Bob",
            "age": 35
        }
    ]
}

# 查找名为 "Alice" 的节点
target_element = "Alice"
result = find_element(json_data, target_element)

# 打印结果
print(json.dumps(result, indent=4))

{
    "name": "Alice",
    "age": 25
}