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

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...]