如何动态获取嵌套在JSON文件中的数据

import json

def get_nested_data(json_data, keys):
    if len(keys) == 1:
        return json_data.get(keys[0])
    else:
        key = keys[0]
        if isinstance(json_data, dict) and key in json_data:
            return get_nested_data(json_data[key], keys[1:])
        elif isinstance(json_data, list) and key.isdigit() and int(key) < len(json_data):
            return get_nested_data(json_data[int(key)], keys[1:])
        else:
            return None

# 加载JSON文件
with open('data.json') as file:
    json_data = json.load(file)

# 定义目标数据的路径
keys = ['key1', 'key2', 'key3']

# 获取嵌套数据
nested_data = get_nested_data(json_data, keys)

# 打印结果
print(nested_data)