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

使用Python遍历当前目录下所有文件名有多种方法，包括使用os模块、glob模块和pathlib模块。

在这些方法中，os模块、glob模块、pathlib模块是最常用的。在本文中，我们将详细介绍这三种方法，并深入探讨每种方法的优缺点和使用场景。

一、使用os模块

os模块是Python标准库的一部分，提供了与操作系统进行交互的各种函数。os模块中的os.listdir()和os.walk()函数可以用于遍历目录。

1.1、使用os.listdir()

os.listdir()函数返回指定目录中的所有文件和目录的名称。

import os

def list_files(directory):
    for file_name in os.listdir(directory):
        print(file_name)
调用函数，遍历当前目录
list_files(".")

优点： 简单直接，适用于小规模目录遍历。

缺点： 不能递归遍历子目录。

1.2、使用os.walk()

os.walk()函数生成目录树下的所有文件名，向上或者向下遍历目录树。

import os

def walk_files(directory):
    for root, dirs, files in os.walk(directory):
        for file_name in files:
            print(os.path.join(root, file_name))
调用函数，递归遍历当前目录及其子目录
walk_files(".")

优点： 能递归遍历所有子目录，适用于复杂的目录结构。

缺点： 生成器形式可能对初学者不太友好，需要理解yield的用法。

二、使用glob模块

glob模块提供了一个文件名模式匹配的工具，可以使用通配符匹配文件路径。

2.1、使用glob.glob()

glob.glob()函数返回所有匹配特定模式的文件名，支持通配符。

import glob

def glob_files(pattern):
    for file_name in glob.glob(pattern):
        print(file_name)
调用函数，遍历当前目录下的所有文件
glob_files("./*")

优点： 支持通配符，简单易用。

缺点： 不能递归遍历子目录（除非使用通配符）。

2.2、使用glob.iglob()

glob.iglob()函数与glob.glob()类似，但返回一个迭代器，效率更高。

import glob

def iglob_files(pattern):
    for file_name in glob.iglob(pattern):
        print(file_name)
调用函数，遍历当前目录下的所有文件
iglob_files("./*")

优点： 更高效，适用于大规模文件遍历。

缺点： 需要了解迭代器的概念。

三、使用pathlib模块

pathlib模块是Python 3.4引入的一个面向对象的文件系统路径操作库。

3.1、使用Path.iterdir()

Path.iterdir()方法返回当前目录下所有文件和目录的Path对象。

from pathlib import Path

def iterdir_files(directory):
    path = Path(directory)
    for file in path.iterdir():
        print(file)
调用函数，遍历当前目录
iterdir_files(".")

优点： 面向对象，代码简洁，易于理解。

缺点： 不能递归遍历子目录。

3.2、使用Path.rglob()

Path.rglob()方法递归地遍历目录，支持通配符。

from pathlib import Path

def rglob_files(pattern):
    path = Path(".")
    for file in path.rglob(pattern):
        print(file)
调用函数，递归遍历当前目录及其子目录下的所有文件
rglob_files("*")

优点： 面向对象，支持递归遍历，代码简洁。

缺点： 需要Python 3.4及以上版本。

四、比较与总结

在实际应用中，根据需求选择合适的方法非常重要。

• os模块：适用于基本的文件遍历需求，尤其是需要递归遍历子目录时。
• glob模块：适用于需要文件名模式匹配的场景，简单高效。
• pathlib模块：适用于现代Python项目，代码简洁，面向对象。

实际应用场景

1. 小规模目录遍历：使用os.listdir()或glob.glob()。
2. 递归遍历复杂目录结构：使用os.walk()或Path.rglob()。
3. 文件名模式匹配：使用glob模块。
4. 现代Python项目：优先考虑使用pathlib模块。

选择合适的方法可以提高代码的可读性和执行效率，希望本文对你在Python中遍历目录有所帮助。如果你需要更高级的项目管理功能，推荐使用研发项目管理系统PingCode和通用项目管理软件Worktile，它们可以帮助你更好地组织和管理项目文件。

相关问答FAQs：

1. 如何使用Python遍历当前目录下的所有文件名？
Python提供了os模块，可以用来处理文件和目录。可以使用os模块中的listdir函数来获取当前目录下的所有文件名。以下是一个示例代码：

import os

def get_all_filenames():
    files = os.listdir()  # 获取当前目录下的所有文件和目录名
    filenames = [file for file in files if os.path.isfile(file)]  # 过滤出文件名
    return filenames

# 调用函数获取所有文件名
all_filenames = get_all_filenames()
print(all_filenames)

2. 如何使用Python遍历当前目录及其子目录下的所有文件名？
如果要遍历当前目录及其子目录下的所有文件，可以使用os模块中的walk函数。以下是一个示例代码：

import os

def get_all_filenames():
    all_filenames = []
    for root, dirs, files in os.walk("."):  # 遍历当前目录及其子目录
        for file in files:
            all_filenames.append(os.path.join(root, file))  # 获取完整的文件路径
    return all_filenames

# 调用函数获取所有文件名
all_filenames = get_all_filenames()
print(all_filenames)

3. 如何使用Python遍历当前目录下指定文件类型的文件名？
如果只想获取当前目录下指定文件类型的文件名，可以使用glob模块。以下是一个示例代码：

import glob

def get_filenames_by_extension(extension):
    filenames = glob.glob(f"*.{extension}")  # 获取指定文件类型的文件名
    return filenames

# 调用函数获取所有.txt文件名
txt_filenames = get_filenames_by_extension("txt")
print(txt_filenames)

希望以上内容能够帮助到您！如果还有其他问题，请随时提问。