Python获取目录文件列表

在进行文件操作时，有时我们需要获取目录中的所有文件列表。Python提供了多种方法来实现这一目标。本文将介绍如何使用Python获取目录文件列表，并提供了相应的代码示例。

os模块

Python的内置模块os提供了一系列函数用于操作文件和目录。其中，os.listdir()函数可以获取指定目录下的所有文件和子目录列表。

import os

def get_file_list(directory):
    file_list = os.listdir(directory)
    return file_list

print(get_file_list('/path/to/directory'))

上述代码中，get_file_list()函数接收一个目录路径作为参数，并返回该目录下的所有文件和子目录列表。os.listdir()函数可以接收一个路径作为参数，并返回该路径下的所有文件和子目录名称列表。

然而，os.listdir()函数仅返回文件和子目录的名称，而不包含完整的路径信息。如果需要获取完整的路径信息，可以使用os.path.join()函数拼接路径。

import os

def get_file_list(directory):
    file_list = []
    for file_name in os.listdir(directory):
        file_path = os.path.join(directory, file_name)
        file_list.append(file_path)
    return file_list

print(get_file_list('/path/to/directory'))

上述代码中，在遍历目录下的文件和子目录时，我们使用os.path.join()函数将目录路径和文件名拼接成完整的文件路径。然后，将完整的文件路径添加到file_list列表中。

glob模块

除了os模块，Python还提供了glob模块用于根据指定规则搜索文件。glob模块支持通配符匹配，可以方便地获取目录文件列表。

import glob

def get_file_list(directory):
    file_list = glob.glob(directory + '/*')
    return file_list

print(get_file_list('/path/to/directory'))

上述代码中，glob.glob()函数接收一个路径作为参数，并返回符合指定规则的所有文件列表。*通配符表示匹配任意字符，/表示目录分隔符。因此，glob.glob(directory + '/*')表示获取目录下的所有文件和子目录列表。

pathlib模块

Python 3.4及以上版本提供了pathlib模块，用于进行路径操作。pathlib模块提供了Path类，可以方便地获取目录文件列表。

from pathlib import Path

def get_file_list(directory):
    file_list = [str(file) for file in Path(directory).rglob('*')]
    return file_list

print(get_file_list('/path/to/directory'))

上述代码中，Path(directory)表示根据指定的路径创建一个Path对象。然后，使用rglob()方法搜索该路径及其子目录下的所有文件。*通配符表示匹配任意字符。最后，将搜索到的文件路径转换为字符串，并添加到file_list列表中。

总结

本文介绍了使用Python获取目录文件列表的几种方法，包括os模块、glob模块和pathlib模块。其中，os模块提供了os.listdir()函数用于获取目录下的文件和子目录列表；glob模块提供了glob.glob()函数用于根据指定规则搜索文件；pathlib模块提供了Path类用于进行路径操作。

根据具体的需求，选择合适的方法可以更加高效地获取目录文件列表。在实际应用中，可以根据文件的特定规则进行筛选和处理。

甘特图

gantt
    dateFormat  YYYY-MM-DD
    title 获取目录文件列表示例
    section 获取目录文件列表
    获取文件列表        :done, 2021-11-30, 1d
    获取完整路径        :done, 2021-11-30, 1d
    使用通配符匹配搜索文件 :done, 2021-11-30, 1d
    使用pathlib模块获取文件列表 :done, 2021-11-30, 1d

甘特图表示了获取目录文件列表的整个过程，并标识出各个步骤的完成时间。这有助于我们更好地理解和掌握代码的执行流程。