Python中JSON的选择性拼合

Question

我正在从返回非标准化json的API中提取数据，这意味着我必须编写一个自定义函数来解析每个数据流，因为JSON具有嵌套字典和列表的不同结构。我想要的输出是包含所有相关端点项的扁平列表。我已经为其中一些编写了函数，但正在尝试创建更通用的函数。

示例JSON的格式为：

JSON = {
    'metadata' : {
        'meta_param' : 'meta_value',
    },
    'contents' : [
        { 
            'content' : [
                {
                    'attributes' : {
                        'series_name' : 'series_1a',
                    },
                    'data' : {
                        'index' : [0, 1, 2, 3, 4, 5],
                        'values' : [12, 84, 38, 3, 92, 67]
                    },
                },
                {
                    'attributes' : {
                        'series_name' : 'series_2a',
                    },
                    'data' : {
                        'index' : [0, 1, 2, 3, 4, 5],
                        'values' : [42, 73, 48, 20, 19, 61]
                    },
                },
            ],
        },
        {
            'content' : [
                {
                    'attributes' : {
                        'series_name' : 'series_1b',
                    },
                    'data' : {
                        'index' : [0, 1, 2, 3, 4, 5],
                        'values' : [13, 85, 39, 4, 93, 68]
                    },
                },
                {
                    'attributes' : {
                        'series_name' : 'series_2b',
                    },
                    'data' : {
                        'index' : [0, 1, 2, 3, 4, 5],
                        'values' : [43, 74, 49, 21, 20, 62]
                    },
                },
            ],
        },
    ]
}

而且我会编写这样的自定义函数，将其转换为平面列表

flatlist = []

for contents in JSON['contents']:
    for content in contents['content']:
        flatlist.append(content['data'])

flatlist -> 
[{'index': [0, 1, 2, 3, 4, 5], 'values': [12, 84, 38, 3, 92, 67]},
 {'index': [0, 1, 2, 3, 4, 5], 'values': [42, 73, 48, 20, 19, 61]},
 {'index': [0, 1, 2, 3, 4, 5], 'values': [13, 85, 39, 4, 93, 68]},
 {'index': [0, 1, 2, 3, 4, 5], 'values': [43, 74, 49, 21, 20, 62]}]

我正在尝试创建一个函数，该函数接受嵌套字典和列表的JSON以及如何访问其中的特定项目的“路线图”，然后返回指定项目的平面列表。

上一个示例的路由图及其用法如下：

JSON_route = [
    'contents',
    'content',
    'data',
]

flatlist = json_to_flatlist(JSON, JSON_route)

flatlist -> 
[{'index': [0, 1, 2, 3, 4, 5], 'values': [12, 84, 38, 3, 92, 67]},
 {'index': [0, 1, 2, 3, 4, 5], 'values': [42, 73, 48, 20, 19, 61]},
 {'index': [0, 1, 2, 3, 4, 5], 'values': [13, 85, 39, 4, 93, 68]},
 {'index': [0, 1, 2, 3, 4, 5], 'values': [43, 74, 49, 21, 20, 62]}]

我的关键问题是我不确定如何以编程方式将JSON_route列表转换为提取数据所需的嵌套for循环。

感谢您的帮助！

Answer 1

def json_to_flatlist(obj, route):
    if len(route) == 0:
        yield obj
    elif isinstance(obj, list):
        for item in obj:
            yield from json_to_flatlist(item, route)
    else:
        yield from json_to_flatlist(obj[route[0]], route[1:])

JSON_route = [
    'contents',
    'content',
    'data',
]

result = list(json_to_flatlist(JSON, JSON_route))
print(result)

结果（为清楚起见添加了换行符：]：>
[
{'index': [0, 1, 2, 3, 4, 5], 'values': [12, 84, 38, 3, 92, 67]}, 
{'index': [0, 1, 2, 3, 4, 5], 'values': [42, 73, 48, 20, 19, 61]}, 
{'index': [0, 1, 2, 3, 4, 5], 'values': [13, 85, 39, 4, 93, 68]}, 
{'index': [0, 1, 2, 3, 4, 5], 'values': [43, 74, 49, 21, 20, 62]}
]