将Json行(Jsonl)生成器转换为csv格式

import json
import csv

# 读取Jsonl文件
def read_jsonl(file_path):
    with open(file_path, 'r') as file:
        for line in file:
            yield json.loads(line)

# 将Jsonl生成器转换为csv格式
def jsonl_to_csv(jsonl_file, csv_file):
    jsonl_generator = read_jsonl(jsonl_file)

    # 创建CSV文件
    with open(csv_file, 'w', newline='') as file:
        writer = csv.writer(file)

        # 写入CSV文件头部
        first_object = next(jsonl_generator)
        headers = list(first_object.keys())
        writer.writerow(headers)

        # 遍历JSON对象并写入CSV文件
        writer.writerow(first_object.values())
        for obj in jsonl_generator:
            writer.writerow(obj.values())

# 示例用法
jsonl_file = 'data.jsonl'
csv_file = 'data.csv'
jsonl_to_csv(jsonl_file, csv_file)