如何将具有相同名称的fasta文件连接成具有不同头文件的文件？

import os

# 获取fasta文件列表
fasta_files = [f for f in os.listdir('.') if f.endswith('.fasta')]

# 创建一个新的fasta文件
output_file = open('output.fasta', 'w')

# 遍历每个fasta文件
for fasta_file in fasta_files:
    # 打开当前fasta文件
    with open(fasta_file, 'r') as file:
        # 读取文件内容
        lines = file.readlines()
        
        # 获取fasta头文件
        header = lines[0].strip()
        
        # 获取fasta序列
        sequence = ''.join(lines[1:]).replace('\n', '')
        
        # 修改头文件
        new_header = '>new_header_' + fasta_file[:-6]  # 修改为不同的头文件
        
        # 将新的头文件和序列写入新的fasta文件
        output_file.write(new_header + '\n')
        output_file.write(sequence + '\n')

# 关闭输出文件
output_file.close()