TS如何从目录中提取所有指定扩展名的文件

提取目录中指定扩展名的文件攻略

要从目录中提取所有指定扩展名的文件，可以按照以下步骤进行操作：

1. 导入必要的库和模块：

import os

1. 定义一个函数来提取目录中指定扩展名的文件：

def extract_files_with_extension(directory, extension):
    file_list = []
    for root, dirs, files in os.walk(directory):
        for file in files:
            if file.endswith(extension):
                file_list.append(os.path.join(root, file))
    return file_list

1. 调用函数并传入目录路径和指定的扩展名：

directory = '/path/to/directory'
extension = '.txt'
files = extract_files_with_extension(directory, extension)

这样，files 列表将包含目录中所有以 .txt 扩展名结尾的文件的完整路径。

示例说明

假设我们有一个名为 documents 的目录，其中包含以下文件：

• file1.txt
• file2.txt
• file3.docx
• file4.txt

我们想要提取所有以 .txt 扩展名结尾的文件。

调用函数：

directory = 'documents'
extension = '.txt'
files = extract_files_with_extension(directory, extension)

返回的 files 列表将包含以下文件的完整路径：

• documents/file1.txt
• documents/file2.txt
• documents/file4.txt

这些文件是目录 documents 中以 .txt 扩展名结尾的文件。

另外，如果我们想要提取所有以 .docx 扩展名结尾的文件，只需将 extension 参数设置为 .docx：

directory = 'documents'
extension = '.docx'
files = extract_files_with_extension(directory, extension)

返回的 files 列表将包含以下文件的完整路径：

• documents/file3.docx

这是目录 documents 中唯一以 .docx 扩展名结尾的文件。