使用regex将txt文件拆分为多个新文件

import re

def split_file(filename, pattern):
    # 读取原始txt文件内容
    with open(filename, 'r') as file:
        content = file.read()

    # 使用正则表达式匹配需要拆分的内容
    matches = re.findall(pattern, content, re.DOTALL)

    # 根据匹配结果将原始内容拆分为多个部分
    parts = re.split(pattern, content, re.DOTALL)

    # 将拆分后的内容分别保存到新的txt文件中
    for i, match in enumerate(matches):
        new_filename = f"{filename}_{i+1}.txt"
        with open(new_filename, 'w') as file:
            file.write(match)

    # 输出拆分后的文件数量
    print(f"成功拆分为 {len(matches)} 个文件。")

# 调用函数进行拆分
split_file('example.txt', r'===\s*Section\s*\d+\s*===')