#######  这个脚本是把一个model跑出来的一类任务的n条离散数据合并成一个json文件
import os
import json

def merge_json_files(input_folder, output_file):
    merged_data = []
    
    # 获取所有文件并排序
    files = sorted(
        [f for f in os.listdir(input_folder) if f.startswith("id:") and f.endswith(".json")],
        key=lambda x: int(x.split(":")[1].split(".")[0])  # 提取数字部分排序
    )
    
    # 逐个读取并追加
    for file_name in files:
        file_path = os.path.join(input_folder, file_name)
        with open(file_path, "r", encoding="utf-8") as f:
            data = json.load(f)
            merged_data.append(data)
    
    # 写入新文件
    with open(output_file, "w", encoding="utf-8") as f:
        json.dump(merged_data, f, ensure_ascii=False, indent=2)

if __name__ == "__main__":
    input_folder = # 修改为你的输入文件夹路径
    output_file =           # 输出文件名
    merge_json_files(input_folder, output_file)
    print(f"已合并 {input_folder} 下的所有 JSON 文件到 {output_file}")

