python - 获取 Windows 目录中的每个文件

Question

我在 Windows 7 中有一个文件夹，其中包含多个 .txt 文件。如何将所述目录中的每个文件作为列表获取？

Answer 1

您可以使用 os.listdir(".")列出当前目录的内容(“.”):

for name in os.listdir("."):
    if name.endswith(".txt"):
        print(name)

如果您希望将整个列表作为 Python 列表，请使用 列表推导:

a = [name for name in os.listdir(".") if name.endswith(".txt")]