读取文件名并直接写入列表

Question

是否可以让Python查看文件夹并将所有文件名（带有特定扩展名）放入列表中？

e.g：

[filename1.txt, filename2.txt,...]

Answer 1

您可以使用glob模块轻松完成此操作：

import glob
filenames = glob.glob('<some_path>/*.<extension>')

Answer 2

这是我找到的快速答案。

import os

txt_files = filter(lambda x: x.endswith('.txt'), os.listdir('mydir'))

Answer 3

我总是为此使用os模块，并且对我来说效果很好。

import os
file_list = os.listdir(path)
print(file_list)

>>> ["file1.txt", "file2.txt", etc...]