将文件名添加到循环内多个fasta文件的fasta标头

import os

# 获取fasta文件所在目录
fasta_dir = '/path/to/fasta/files/'

# 遍历fasta文件
for filename in os.listdir(fasta_dir):
    if filename.endswith('.fasta'):
        # 打开fasta文件
        with open(os.path.join(fasta_dir, filename), 'r') as file:
            # 读取fasta文件内容
            fasta_content = file.readlines()
        
        # 将文件名添加到fasta标头
        new_fasta_content = []
        for line in fasta_content:
            if line.startswith('>'):
                line = line.strip() + ' [' + filename + ']\n'
            new_fasta_content.append(line)
        
        # 写入修改后的fasta序列到新文件
        new_filename = os.path.splitext(filename)[0] + '_modified.fasta'
        with open(os.path.join(fasta_dir, new_filename), 'w') as file:
            file.writelines(new_fasta_content)