python如何遍历当前目录下所有文件名

使用Python遍历当前目录下所有文件名的方法有很多，其中最常用的包括os模块、os.path模块和glob模块。 这些方法都能有效地获取当前目录下的文件名，并且每个方法都有其独特的优点和适用场景。本文将详细介绍这些方法，并探讨它们的具体应用和优缺点。

一、使用os模块遍历当前目录

1.1、os.listdir()方法

os.listdir() 是Python中用于列出指定目录下所有文件和子目录的方法。它返回一个包含目录中所有文件和子目录名称的列表。

import os
def list_files(directory):
    return os.listdir(directory)
示例
current_directory = os.getcwd()
files = list_files(current_directory)
print(files)

在这个示例中，os.getcwd()用于获取当前工作目录，os.listdir()则返回当前目录下所有文件和子目录的名称。这种方法简单易用，但它不会递归地遍历子目录中的文件。

1.2、os.walk()方法

os.walk() 是一个更强大的方法，它可以递归地遍历目录树，生成包含目录路径、目录中的子目录和目录中的文件的三元组。

import os
def walk_directory(directory):
    for dirpath, dirnames, filenames in os.walk(directory):
        for filename in filenames:
            print(os.path.join(dirpath, filename))
示例
current_directory = os.getcwd()
walk_directory(current_directory)

在这个示例中，os.walk() 会遍历当前目录及其所有子目录，打印出每个文件的完整路径。这种方法适用于需要递归遍历目录的场景。

二、使用os.path模块遍历当前目录

2.1、os.path.isfile()和os.path.isdir()方法

使用os.path模块中的方法可以更细粒度地控制文件和目录的遍历。

import os
def list_files(directory):
    for item in os.listdir(directory):
        if os.path.isfile(os.path.join(directory, item)):
            print(item)
示例
current_directory = os.getcwd()
list_files(current_directory)

在这个示例中，os.path.isfile() 用于检查路径是否为文件，os.path.isdir() 用于检查路径是否为目录。这种方法适用于需要区分文件和目录的场景。

三、使用glob模块遍历当前目录

3.1、glob.glob()方法

glob模块提供了一个方便的文件模式匹配工具，可以使用通配符模式来匹配目录中的文件。

import glob
def list_files(pattern):
    return glob.glob(pattern)
示例
current_directory = os.getcwd()
pattern = os.path.join(current_directory, '*')
files = list_files(pattern)
print(files)

在这个示例中，glob.glob() 使用通配符模式匹配当前目录下的所有文件和目录。这种方法简单易用，适用于需要模式匹配的场景。

3.2、递归匹配

glob模块还提供了递归匹配功能，可以使用通配符递归地匹配目录中的文件。

import glob
def list_files(pattern):
    return glob.glob(pattern, recursive=True)
示例
current_directory = os.getcwd()
pattern = os.path.join(current_directory, '', '*')
files = list_files(pattern)
print(files)

在这个示例中，glob.glob() 使用 <strong> 通配符递归地匹配当前目录及其所有子目录中的文件。这种方法适用于需要递归模式匹配的场景。

四、综合比较

4.1、性能比较

在性能方面，os.listdir() 和 os.path 方法通常比 os.walk() 和 glob 方法更快，因为前者不需要递归遍历目录。然而，如果需要递归遍历目录，则 os.walk() 和 glob 方法更为合适。

4.2、易用性比较

在易用性方面，os.listdir() 方法最为简单，但功能有限；os.walk() 方法功能强大，但使用稍显复杂；glob 方法提供了方便的模式匹配功能，但在某些情况下可能不如 os.walk() 灵活。

4.3、适用场景比较

os.listdir() 适用于简单的目录遍历，不需要递归的场景；os.walk() 适用于需要递归遍历目录的场景，尤其是需要访问目录路径、子目录和文件的情况；glob 适用于需要模式匹配的场景，特别是需要递归模式匹配的情况。

五、实际应用案例

5.1、文件统计

假设我们需要统计当前目录及其所有子目录中的文件数量，可以使用 os.walk() 方法：

import os
def count_files(directory):
    count = 0
    for dirpath, dirnames, filenames in os.walk(directory):
        count += len(filenames)
    return count
示例
current_directory = os.getcwd()
file_count = count_files(current_directory)
print(f"Total number of files: {file_count}")

5.2、文件过滤

假设我们需要过滤出当前目录及其所有子目录中的特定类型文件（例如 .txt 文件），可以使用 glob 方法：

import glob
def filter_files(pattern):
    return glob.glob(pattern, recursive=True)
示例
current_directory = os.getcwd()
pattern = os.path.join(current_directory, '', '*.txt')
txt_files = filter_files(pattern)
print(txt_files)

六、总结

Python 提供了多种方法用于遍历当前目录下的所有文件名，主要包括 os 模块、os.path 模块和 glob 模块。 其中，os.listdir() 方法简单易用，但不支持递归遍历；os.walk() 方法功能强大，适用于递归遍历目录的场景；glob 方法提供了方便的模式匹配功能，适用于需要模式匹配的场景。

根据实际需求选择合适的方法，可以显著提高代码的效率和可读性。在大多数情况下，os.walk() 和 glob 方法是处理复杂目录遍历任务的首选。