如何对fasta文件头进行分组

def group_fasta_headers(fasta_file):
    groups = {}  # 用字典存储分组结果，键为分组名称，值为文件头信息列表

    with open(fasta_file, 'r') as file:
        current_group = None  # 当前分组名称
        for line in file:
            line = line.strip()
            if line.startswith('>'):  # 判断是否为文件头行
                header = line[1:]  # 提取文件头信息
                # 根据特定规则进行分组，这里以"_"为分隔符，取第一个字段作为分组名称
                current_group = header.split('_')[0]
                if current_group not in groups:
                    groups[current_group] = []  # 初始化分组列表
            else:
                if current_group is not None:
                    groups[current_group].append(line)  # 将文件头信息添加到对应的分组中

    return groups

fasta_file = 'example.fasta'
result = group_fasta_headers(fasta_file)
for group, headers in result.items():
    print(f"Group: {group}")
    for header in headers:
        print(header)
    print()