将三个长度不一致的列表转换为Python中的JSON格式。

Question

我有三个列表：

filtered_headings：包含了如['教育教学机构', '学生数量', '教学员工数量', '非教学员工数量']这样的标题。
sub_headings：列举了各个层级的教育机构，比如['大学/学院', '基础阶段', '中学阶段', '中学阶段', ...]。
values：提供了一组数值，与上述两个列表中的项一一对应，例如['2', '10', '12', '566', ...]。

我的目标是将这些列表转换成如下JSON格式：

{
    "Educational Institutions": {
        "College/University": "2",
        "Basic Level": "10",
        "Secondary Level": "12"
    },
    ...
}

但是，我目前得到的结果却是错误地重复了values列表的最后三个值。我的代码如下：

result = {}
for indv_heading in filtered_headings:
    data_dict = {sub_heading: value for sub_heading, value in zip(sub_headings, values)}
    result[indv_heading] = data_dict
json_data = json.dumps(result, indent=1)
print(json_data)

请问如何修正我的代码，以便正确生成预期的JSON结构？

Thilo · Answer

可以将sub_headings和values配对后，使用Python的itertools.batched函数将它们分组为每组3个元素来构建子字典。然后，再将filtered_headings与这些子字典配对，生成主字典的键值序列。以下是具体实现方法：

from itertools import batched

# 使用itertools.batched将sub_headings和values配对并分组
sub_dicts = map(dict, batched(zip(sub_headings, values), 3))

# 将filtered_headings与子字典配对构造最终的字典结构
result = dict(zip(filtered_headings, sub_dicts))

注意，itertools.batched是在Python 3.12版本中引入的。如果你使用的Python版本早于3.12，可以参考itertools模块文档中的“迭代器食谱”（iterator recipes），手动实现一个类似的batched功能。

Walker · Answer

我不确定这些列表在未来是否会增长，如果不会的话，我们可以采用一个虽不适合生产环境但简单、易读且易于理解的解决方案。

简单非优化方案:

result = {}
for title in filtered_headings:
    temp_dict = {}
    for _ in range(3):
        temp_dict[sub_headings.pop(0)] = values.pop(0)
    result[title] = temp_dict

json_data = json.dumps(result, indent=1)
print(json_data)

另外，你能否告知数据的来源呢？

因为可能有更高效的处理方式。