Python技术求助：标点处理功能无法整合进spam_indicator函数

我来帮你梳理下代码里的几个关键错误，然后给出修复后的版本：

原代码的核心问题 #

1. 处理标点的顺序错误：你先把文本用split()拆成了单词列表，然后试图对列表调用replace()——但列表没有这个方法，这会直接抛出AttributeError。应该先对原始完整文本移除标点，再拆分单词。
2. 提前终止函数的错误return：代码里的return word完全是多余的，而且会让函数在执行到这一行时直接退出，后面的统计逻辑根本不会运行。
3. 潜在的零除风险：如果处理后的文本没有有效单词（比如全是标点），w会是0，这时s/w会触发ZeroDivisionError，需要加个判断。

修复后的完整代码 #

import string

# 假设SPAM_WORDS是你已经定义好的垃圾词集合
SPAM_WORDS = {"free", "win", "cash", "prize"}

def spam_indicator(text):
    # 第一步：移除所有标点符号
    cleaned_text = text
    for char in string.punctuation:
        cleaned_text = cleaned_text.replace(char, '')
    
    # 第二步：拆分单词并转为小写
    words = cleaned_text.lower().split()
    # 去重，统计唯一单词数
    unique_words = list(set(words))
    w = len(unique_words)
    
    # 处理无有效单词的情况，避免零除
    if w == 0:
        return 0.0
    
    # 统计垃圾词数量
    s = 0
    for word in unique_words:
        if word in SPAM_WORDS:
            s += 1
    
    # 返回保留两位小数的结果
    return round(s / w, 2)

代码说明 #

• 先对原始文本做标点清理，再拆分单词，避免列表操作的错误。
• 用set()快速去重，比手动遍历判断更简洁高效。
• 增加了w == 0的判断，防止除以零的错误。
• 用round()代替字符串格式化，返回浮点数更符合函数的数值返回预期。

你可以测试下这个版本，不管文本里有多少标点，都能正确识别单词并计算垃圾词占比啦。

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