如何将层级化的 JSON 数据打印到 CSV 的多列中？

python

2024-03-25 21:40:36

动态将 JSON 数据打印到 CSV 的多列

问题

当你需要将复杂的 JSON 数据导出为结构化的 CSV 文件时，可能会面临一个挑战：如何动态地处理层级结构并将其映射到 CSV 文件的列中。

解决方法

为了解决这个问题，我们可以遵循以下步骤：

解析 JSON 数据

首先，使用适当的工具（例如 json.load()）解析 JSON 数据并将其转换为嵌套字典。这将创建一个 Python 对象，其中嵌套键值对代表 JSON 数据的结构。

确定层级结构

接下来，我们需要确定 JSON 数据中的层级结构。遍历嵌套字典，识别包含其他字典的键。这些键将代表 CSV 文件中的列。

创建 CSV 文件的列

根据确定的层级结构，创建 CSV 文件的列标头。这些标头将用作 CSV 文件的第一行。

遍历嵌套字典

使用循环遍历嵌套字典。对于每个键（代表一个列），提取对应值（代表该列的数据）。如果值是一个字典，则递归遍历该字典，提取所有嵌套的值。

按层级写入 CSV 文件

对于每个值，将其附加到当前行。遍历完整个字典后，将当前行写入 CSV 文件。

代码示例

import csv
import json

# 解析 JSON 数据
data = json.load(open('data.json'))

# 确定层级结构
levels = []
for key, value in data.items():
    if isinstance(value, dict):
        levels.append(key)

# 创建 CSV 文件的列
with open('output.csv', 'w', newline='') as f:
    writer = csv.writer(f)
    writer.writerow(levels)

    # 遍历嵌套字典
    for key, value in data.items():
        if isinstance(value, dict):
            row = []
            for level in levels:
                if level in value:
                    row.append(value[level])
                else:
                    row.append('')
            writer.writerow(row)