testcase-data/get_rank1&5.py

def extract_rank1_and_rank5(input_file="ALL_RESULT.md", output_file="RANK1_5.md"):
    with open(input_file, "r", encoding="utf-8") as f:
        lines = f.readlines()

    if not lines:
        print("输入文件为空")
        return

    # 保留表头（前两行：表头 + 分隔线）
    extracted = lines[:2]

    # 只保留 rank1 和 rank5
    for line in lines[2:]:
        if "| rank5 " in line:
            extracted.append(line)

    # 保存新文件
    with open(output_file, "w", encoding="utf-8") as f:
        f.writelines(extracted)

    print(f"已生成 {output_file}")


if __name__ == "__main__":
    extract_rank1_and_rank5()