使用每个文件中第一个序列的ID自动重命名fasta文件

from Bio import SeqIO

def rename_fasta(input_file, output_file):
    records = SeqIO.parse(input_file, "fasta")
    new_records = []

    for i, record in enumerate(records):
        # 获取第一个序列的ID
        new_id = record.id.split("|")[0]  # 根据具体的ID格式进行分割

        # 创建新的记录，并将序列ID和序列内容赋值给新记录
        new_record = record
        new_record.id = new_id
        new_record.description = ""
        new_records.append(new_record)

    # 将新的记录写入输出文件
    SeqIO.write(new_records, output_file, "fasta")

# 调用函数进行重命名
rename_fasta("input.fasta", "output.fasta")