将两个具有一对多关系的CSV表转换为具有嵌入子文档列表的JSON

import csv
import json

def convert_csv_to_json(parent_csv_file, child_csv_file, parent_key, child_key):
    parent_data = {}
    with open(parent_csv_file, 'r') as csvfile:
        reader = csv.DictReader(csvfile)
        for row in reader:
            parent_data[row[parent_key]] = {
                'fields': row,
                'children': []
            }

    with open(child_csv_file, 'r') as csvfile:
        reader = csv.DictReader(csvfile)
        for row in reader:
            parent_id = row[child_key]
            if parent_id in parent_data:
                parent_data[parent_id]['children'].append(row)

    json_data = json.dumps(list(parent_data.values()), indent=4)
    return json_data

# 示例用法
parent_csv_file = 'parent.csv'
child_csv_file = 'child.csv'
parent_key = 'parent_id'
child_key = 'parent_id'

json_data = convert_csv_to_json(parent_csv_file, child_csv_file, parent_key, child_key)
print(json_data)