Python：从 .txt 文件的目录创建文件列表

Question

我正在尝试从 .txt 文件的目录中加载语料库，然后创建一个文档列表。

我认为它会很简单，但是当我运行它时没有任何反应，我错过了什么吗？

import os.path
import re
import glob

 def load_data_from_dir(path):
file_list = glob.glob('/transcripts/*.txt')


# create document list:
documents_list = []
for filename in file_list:
    with open(filename, 'r', encoding='utf8') as f:
        text = f.read()
        f.close()
        documents_list.append(text)
print("Total Number of Documents:",len(documents_list))
return documents_list

Answer 1

确保正确缩进 python 中的代码。 另外，检查代码的路径。 要么给出绝对路径，要么给出相对路径。 不过，绝对路径可以完美运行。

import os.path
import re
import glob

def load_data_from_dir():
    file_list = glob.glob('./transcripts/*.txt')


    # create document list:
    documents_list = []
    for filename in file_list:
        with open(filename, 'r', encoding='utf8') as f:
            text = f.read()
            documents_list.append(text)
    print("Total Number of Documents:",len(documents_list))
    return documents_list

load_data_from_dir()

Answer 2

您的代码中有一些错误。

1. function ( load_data_from_dir ) 正文没有缩进。 缩进 function 正文中的所有行（直到您的 return 语句）
2. 使用with构造打开的文件不应显式关闭。 删除f.close()

3. 如果这是一个单独的模块并且您只运行 function。 然后什么都不会发生。 您需要确保将其称为 function。 所以添加

   if __name__ == '__main__': load_data_from_dir(...)