使用扩展名列出目录 Python 中的文件

在 Python 中，可以使用 os 模块中的 listdir() 函数列出指定目录中的所有文件和子目录。如果需要仅列出某种类型的文件，则可以使用 endswith() 方法筛选出指定扩展名的文件。

下面是一个示例代码片段，演示如何列出指定目录中以 .txt 扩展名结尾的文件：

import os

def get_files(directory, extension='.txt'):
    files = []
    for file in os.listdir(directory):
        if file.endswith(extension):
            files.append(file)
    return files

directory = '/path/to/directory'
txt_files = get_files(directory)
print(txt_files)

以上代码定义了一个名为 get_files 的函数，它接受两个参数：directory 表示要列出文件的目录，extension 表示要筛选的文件扩展名，默认为 .txt。

函数中使用了 os.listdir() 函数列出指定目录中的所有文件，然后使用 endswith() 方法筛选出指定扩展名的文件，并将其添加到 files 列表中。最后，函数返回该列表。

在代码的下一行，我们调用 get_files 函数，并将返回的列表赋值给 txt_files 变量。最后，我们使用 print() 函数输出该列表。

运行上述代码将输出目录中以 .txt 扩展名结尾的文件列表。如果想要列出其它类型的文件，只需要修改 extension 参数即可。

本文介绍了如何使用 Python 中的 os 模块列出指定目录中的文件，并演示了如何筛选特定类型的文件。希望本文能够帮助你更好地处理文件操作任务。