正文

掌握Python目录遍历：轻松实现文件和文件夹搜索

/0 浏览量
0531

在Python中，目录遍历是一个非常有用的功能，它可以帮助我们轻松地访问和操作文件系统中的文件和文件夹。无论是进行文件搜索、文件统计，还是其他需要访问文件系统的任务，目录遍历都是必不可少的。下面，我将详细介绍如何在Python中实现目录遍历，并分享一些实用的技巧。

使用os模块进行目录遍历

Python的os模块提供了丰富的目录遍历功能。以下是一些常用的方法：

1. os.listdir()

os.listdir()函数可以列出指定目录下的所有文件和文件夹。它返回一个列表，包含目录下的所有项。

import os

# 列出当前目录下的所有文件和文件夹
files = os.listdir('.')
print(files)

2. os.walk()

os.walk()函数是一个更高级的目录遍历方法，它可以生成目录树中的文件名。它返回一个三元组（dirpath, dirnames, filenames），其中dirpath是当前正在遍历的目录路径，dirnames是该目录下的所有子目录名列表，filenames是该目录下的所有非目录文件名列表。

import os

# 遍历当前目录及其子目录下的所有文件
for dirpath, dirnames, filenames in os.walk('.'):
    for filename in filenames:
        print(os.path.join(dirpath, filename))

3. os.scandir()

os.scandir()函数是一个更现代的目录遍历方法，它返回一个迭代器，可以用来遍历目录中的所有项。与os.listdir()相比，os.scandir()提供了更多的信息，例如每个项的类型（文件或目录）。

import os

# 使用os.scandir()遍历当前目录及其子目录下的所有文件
with os.scandir('.') as entries:
    for entry in entries:
        if entry.is_file():
            print(entry.path)

实现文件搜索

使用上述方法，我们可以轻松地实现文件搜索。以下是一个简单的例子，演示如何使用os.walk()来搜索特定名称的文件。

import os

def search_files(directory, pattern):
    for dirpath, dirnames, filenames in os.walk(directory):
        for filename in filenames:
            if pattern in filename:
                print(os.path.join(dirpath, filename))

# 搜索当前目录及其子目录下所有包含"example"的文件
search_files('.', 'example')

总结

目录遍历是Python中一个非常有用的功能，可以帮助我们轻松地访问和操作文件系统。通过使用os模块中的os.listdir(), os.walk(), 和os.scandir()等方法，我们可以实现各种目录遍历任务。希望这篇文章能帮助你更好地掌握Python目录遍历技巧。

-- 展开阅读全文 --