Python实现文本去标点并转小写的代码问题求助

原代码的问题点 #

• 函数remove_punctuation错误使用self参数：这不是类方法，普通函数应使用常规参数名（如text）
• 循环遍历标点列表时逻辑错误：for i in punctuations中i是标点元素本身，而非索引，不需要再用punctuations[i]取值
• 使用split方法错误：split是用来分割字符串的，要去除标点应该用replace方法，且字符串不可变，需重新赋值保存结果
• remove_punctuation无返回值：调用后会得到None，无法获取处理后的文本
• 主入口定义错误：Python的主程序入口应该用if __name__ == "__main__":，而非定义__main__函数

修正后的代码 #

基础修正版 #

import string

# 可以直接用string.punctuation，比自定义更全面，若需要额外标点可追加
punctuations = string.punctuation + "...~@{}*"

def remove_punctuation(text):
    cleaned_text = text
    for punct in punctuations:
        cleaned_text = cleaned_text.replace(punct, "")
    return cleaned_text

def main():
    text = input("Enter a text: ")
    text_lower = text.lower()
    print("Here is your text in lower case:\n")
    print(text_lower)
    text_no_punct = remove_punctuation(text_lower)
    print("\nText without punctuation:")
    print(text_no_punct)

if __name__ == "__main__":
    main()

更高效的优化版（用str.translate） #

import string

def process_text(text):
    # 转小写
    text_lower = text.lower()
    # 创建标点映射表，用于删除所有标点
    translator = str.maketrans("", "", string.punctuation + "...~@{}*")
    # 去除标点
    cleaned_text = text_lower.translate(translator)
    return cleaned_text

if __name__ == "__main__":
    text = input("Enter a text: ")
    result = process_text(text)
    print("Processed text:\n", result)

关于是否用类实现 #

这个功能逻辑简单，用函数完全足够，代码更简洁。如果后续需要扩展更多相关功能（比如统计词频、情感分析等），可以考虑封装成类来组织代码，让结构更清晰。示例如下：

import string

class TextProcessor:
    def __init__(self, additional_punct=None):
        self.punctuations = string.punctuation
        if additional_punct:
            self.punctuations += additional_punct
    
    def to_lower(self, text):
        return text.lower()
    
    def remove_punctuation(self, text):
        translator = str.maketrans("", "", self.punctuations)
        return text.translate(translator)
    
    def process(self, text):
        text_lower = self.to_lower(text)
        return self.remove_punctuation(text_lower)

if __name__ == "__main__":
    processor = TextProcessor(additional_punct="...~@{}*")
    text = input("Enter a text: ")
    result = processor.process(text)
    print("Processed text:\n", result)

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