如何在文本文档中第n个单词之后创建新行？

def insert_new_line_after_nth_word(file_path, n, new_line):
    # 打开文本文档并读取内容
    with open(file_path, 'r') as file:
        content = file.read()

    # 分割文本内容为单词列表
    words = content.split()

    # 在第n个单词之后插入新行
    words.insert(n, new_line)

    # 重新构建文本内容
    new_content = ' '.join(words)

    # 写入文本文档
    with open(file_path, 'w') as file:
        file.write(new_content)

file_path = 'path/to/your/text/document.txt'
n = 5  # 在第5个单词之后创建新行
new_line = '\nThis is a new line.'

insert_new_line_after_nth_word(file_path, n, new_line)