如何修改Python词干提取代码以支持自定义输入输出文件路径

以下是满足需求的完整修改后代码：

import nltk
from nltk.stem import PorterStemmer

# 初始化词干提取器
stemmer = PorterStemmer()

# 获取用户输入的待处理文件路径
input_path = input("请输入待处理单词文本文件的完整路径：").strip()
# 获取用户输入的输出文件路径
output_path = input("请输入输出结果文件的完整路径：").strip()

# 读取待处理文件内容
with open(input_path, 'r', encoding='utf-8') as fp:
    tokens = fp.readlines()

# 统一写入结果到输出文件
with open(output_path, 'w', encoding='utf-8') as out_fp:
    for t in tokens:
        original_word = t.strip()
        # 跳过空行避免无效结果
        if not original_word:
            continue
        stem_result = stemmer.stem(original_word)
        # 原始单词和词干结果用制表符分隔，可自行调整分隔格式
        out_fp.write(f"{original_word}\t{stem_result}\n")

调整说明 #

• 清理了原代码冗余导入：删除了重复导入的PorterStemmer和未使用的LancasterStemmer，减少不必要的资源加载
• 实现自定义路径输入：通过input()函数分别提示用户输入待处理文件、输出文件的完整路径，适配不同用户的文件存储位置
• 优化文件写入逻辑：不再在循环中重复打开输出文件，改用上下文管理器一次性打开写入，避免资源泄漏同时提升运行效率
• 调整输出内容格式：每行同时保留原始单词和词干提取结果，默认用制表符分隔，可根据需求自行修改为空格、逗号等分隔符
• 新增异常兼容逻辑：自动跳过输入文件的空行，同时读写文件明确指定utf-8编码，避免不同系统下特殊字符乱码问题
• 依赖提示：首次运行如果提示nltk库缺失，执行pip install nltk安装依赖即可正常运行

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