Python 使用os.walk()获取特定文件的路径

在本文中，我们将介绍如何使用Python的os.walk()函数来获取特定文件的路径。os.walk()函数可以递归遍历指定目录及其子目录，并返回所有文件和文件夹的路径。我们可以利用这个函数来查找特定类型的文件。

os.walk()函数的基本用法

os.walk()函数可以接受一个参数，即要遍历的目录的路径。该函数返回一个迭代器对象，可以使用for循环来遍历这个迭代器对象，获取所有的文件和文件夹路径。

下面是os.walk()函数的基本用法示例：

import os

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

directory_path = "/path/to/directory"  # 替换为要遍历的目录的路径
file_extension = ".txt"  # 替换为要查找的文件类型的扩展名
file_paths = get_files_with_extension(file_extension, directory_path)
print(file_paths)

在上面的示例中，我们定义了一个名为get_files_with_extension的函数，用于获取指定目录下特定类型的文件路径。函数接受两个参数：文件类型的扩展名和要遍历的目录的路径。通过遍历os.walk()函数返回的迭代器对象，我们可以筛选出符合文件类型的文件，并将其路径存储在result列表中。最后，函数返回result列表。

示例和进阶用法

下面是一些示例，展示了如何根据不同的需求使用os.walk()函数来获取特定文件的路径。

1. 获取所有的文本文件路径

def get_text_files(directory):
    return get_files_with_extension(".txt", directory)

2. 获取特定前缀的文件路径

def get_files_with_prefix(prefix, directory):
    result = []
    for root, dirs, files in os.walk(directory):
        for file in files:
            if file.startswith(prefix):
                result.append(os.path.join(root, file))
    return result

3. 获取特定后缀的文件路径

def get_files_with_suffix(suffix, directory):
    result = []
    for root, dirs, files in os.walk(directory):
        for file in files:
            file_name, file_extension = os.path.splitext(file)
            if file_extension == suffix:
                result.append(os.path.join(root, file))
    return result

4. 获取指定大小的文件路径

def get_files_with_size(size, directory):
    result = []
    for root, dirs, files in os.walk(directory):
        for file in files:
            file_path = os.path.join(root, file)
            if os.path.getsize(file_path) == size:
                result.append(file_path)
    return result