Python - 尝试创建一个简单的反垃圾邮件过滤器

Question

我必须为毕业论文编写一个简单的垃圾邮件过滤器。我的目的是打开并阅读两个不同的列表，一个是垃圾邮件，一个是火腿。名称（cartella）就像英文中的文件夹。问题是当我打印垃圾邮件时，它只打印数组spam[0] 的第一个元素，当我打印长度时它打印1，但它应该打印1500。问题出在函数中。我需要帮助。

这是我的代码：

import nltk 
import os

def importalista(cartella):
    lista=[]
    listafile=os.listdir(cartella)
    for file in listafile:
        f=open(cartella + file, 'r+')
        lista.append(f.read())
        f.close()

        return lista

spam = importalista('C:\\enron1\\spam\\')
ham = importalista('C:\\enron1\\ham\\')
print(len(spam))

Answer 1

你在 for 循环中有 return 语句。它将在第一次迭代后退出该函数。如果您按如下方式更改代码，那将是正确的。 导入 nltk 导入操作系统

def importalista(cartella):
    lista=[]
    listafile=os.listdir(cartella)
    for file in listafile:
        f=open(cartella + file, 'r+')
        lista.append(f.read())
        f.close()

    return lista

spam = importalista('C:\\enron1\\spam\\')
ham = importalista('C:\\enron1\\ham\\')
print(len(spam))