从txt到两个不同的txt，拆分并打印\ of *n行之前和之后的单词

import re

def split_and_print_words(file_path, n):
    # 读取txt文件内容
    with open(file_path, 'r') as file:
        content = file.read()

    # 按行拆分文本内容
    lines = content.split('\n')

    # 获取*n行之前的文本内容
    before_lines = lines[:n]
    before_text = '\n'.join(before_lines)

    # 获取*n行之后的文本内容
    after_lines = lines[n:]
    after_text = '\n'.join(after_lines)

    # 提取单词
    word_pattern = r'\b\w+\b'
    before_words = re.findall(word_pattern, before_text)
    after_words = re.findall(word_pattern, after_text)

    # 打印并保存单词
    print("前{}行的单词：".format(n))
    for word in before_words:
        print(word)
    save_words_to_txt(before_words, "before_words.txt")

    print("后{}行的单词：".format(len(lines) - n))
    for word in after_words:
        print(word)
    save_words_to_txt(after_words, "after_words.txt")

def save_words_to_txt(words, file_path):
    # 将单词保存到txt文件
    with open(file_path, 'w') as file:
        file.write('\n'.join(words))

# 调用函数进行处理
split_and_print_words("input.txt", 5)