使用python-docx移除Word文档末尾内容的代码无法正常运行

代码中的核心问题： #

1. 未定义的file_path变量：函数remove_end里直接调用document.save(file_path)，但file_path既不是函数参数也未在内部定义，运行时会直接抛出NameError，这是代码无法运行的首要原因。
2. 单词数量判断逻辑错误：原代码用len(paragraph.text.split()) <=2匹配段落，这会把包含2个单词的段落也纳入判断，不符合你"仅含单个单词"的需求，应该改为严格判断单词数量等于1。
3. 冗余的无效判断：if paragraph not in document.paragraphs:完全没必要——循环本身就是遍历document.paragraphs的元素，这个判断永远为假，属于多余代码。
4. 低效的索引获取方式：通过document.paragraphs.index(paragraph)获取索引会再次遍历段落列表，效率低下，不如直接用enumerate同时获取索引和段落。

修复后的代码： #

import os
from docx import Document

def remove_end(document, file_path):
    # 用集合存储目标单词，查询效率更高
    target_words = {'references', 'acknowledgements', 'note', 'notes'}
    target_paragraph_idx = None

    # 遍历段落，同时获取索引和内容
    for idx, para in enumerate(document.paragraphs):
        cleaned_text = para.text.strip()
        # 先确认是单个单词，再检查是否在目标列表中（不区分大小写）
        if len(cleaned_text.split()) == 1:
            if cleaned_text.lower() in target_words:
                target_paragraph_idx = idx
                break

    # 找到目标段落则删除后续内容并保存
    if target_paragraph_idx is not None:
        del document.paragraphs[target_paragraph_idx + 1:]
        document.save(file_path)

额外优化说明： #

• 将目标单词改为集合类型，比列表的成员查询速度更快，尤其是单词数量较多时。
• 先判断单词数量再做大小写转换和匹配，逻辑更贴合需求，减少不必要的计算。
• 先记录目标索引再执行删除操作，避免遍历过程中修改段落集合可能引发的异常。

内容的提问来源于stack exchange，提问作者Leila