python 获取子目录指定文件路径

原创

mob64ca12e60047 2023-11-16 17:58:08 ©著作权

文章标签 子目录文件路径 Python 文章分类 Python 后端开发

©著作权归作者所有：来自51CTO博客作者mob64ca12e60047的原创作品，请联系作者获取转载授权，否则将追究法律责任

Python获取子目录指定文件路径

在Python中，我们经常需要操作文件和目录。有时候，我们需要获取指定目录下的所有子目录中满足一定条件的文件的路径。本文将介绍如何使用Python来实现这个功能，并提供相应的代码示例。

获取子目录下的所有文件

首先，我们需要获取指定目录下的所有文件。Python的os模块提供了一些函数来实现这个功能。其中，os.walk()函数可以遍历指定目录及其子目录，并返回每个目录下的子目录和文件。

下面是一个简单的代码示例，展示了如何使用os.walk()函数来获取指定目录下的所有文件的路径：

import os

def get_files(directory):
    file_paths = []
    for root, dirs, files in os.walk(directory):
        for file in files:
            file_path = os.path.join(root, file)
            file_paths.append(file_path)
    return file_paths

# 获取当前目录下的所有文件路径
current_directory = os.getcwd()
files = get_files(current_directory)
print(files)

在上述代码中，get_files()函数接收一个目录路径作为参数，并返回该目录下的所有文件路径。os.walk()函数返回一个生成器，它会依次遍历每个目录，并返回该目录下的子目录和文件。通过遍历返回的文件列表，我们可以获取每个文件的路径，并将其添加到file_paths列表中。

在代码示例中，我们使用了os.path.join()函数来拼接目录路径和文件名，以获取完整的文件路径。最后，我们打印出获取到的所有文件路径。

指定文件名获取路径

在上述代码示例中，我们获取了指定目录下的所有文件的路径。但有时候，我们只想获取满足特定条件的文件的路径。例如，我们只想获取所有以.txt结尾的文件的路径。

下面是一个修改后的代码示例，展示了如何根据文件名来获取指定目录下满足条件的文件的路径：

import os

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

# 获取当前目录下以".txt"结尾的所有文件路径
current_directory = os.getcwd()
files = get_files_with_extension(current_directory, ".txt")
print(files)

在上述代码中，我们添加了一个额外的参数extension，用于指定文件名的后缀。通过if file.endswith(extension)条件判断，我们只将满足条件的文件路径添加到file_paths列表中。