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

Question

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

例如：

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

Answer 1

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

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

Answer 2

我总是为此使用 os 模块，并且非常适合我。

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

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

Answer 3

这是我找到的快速答案。

import os

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