python 如何查找文件

在Python中查找文件的方法有多种，包括使用os库的walk函数、glob模块进行模式匹配、以及pathlib模块的rglob方法等。其中，os.walk是递归遍历目录的基础方法，而glob模块适合用于简单的文件模式匹配。pathlib是Python 3.4引入的面向对象的路径处理库，提供了更为简洁和现代的文件查找方法。以下将详细介绍其中的一种方法：os.walk。

使用os.walk查找文件时，可以遍历指定目录及其子目录，获取所有文件和文件夹的路径信息。这使得查找特定文件名或文件类型变得非常方便。具体来说，os.walk返回一个生成器对象，该对象会逐一生成一个三元组，包含当前目录路径、当前目录中的子目录列表以及当前目录中的文件列表。通过遍历这些生成的三元组，可以轻松实现对所有文件的查找和筛选。接下来，我们将深入探讨Python中查找文件的不同方法。

一、OS.WALK方法

os.walk是Python内置os模块中的一个方法，用于遍历目录树。在查找文件时，os.walk可以递归地遍历指定目录及其子目录，获取所有文件和目录的信息。

1.1 os.walk的基本用法

os.walk方法返回一个生成器对象，通过迭代该生成器，可以得到一个三元组(root, dirs, files)，其中root表示当前遍历到的目录路径，dirs是该目录下的子目录列表，files是该目录下的文件列表。

import os
for root, dirs, files in os.walk('/path/to/directory'):
    for file in files:
        if file.endswith('.txt'):
            print(os.path.join(root, file))

1.2 os.walk的优缺点

优点：

递归遍历： os.walk能够递归遍历目录树，适用于需要遍历多个子目录的情况。
灵活性高： 可以在遍历过程中自定义文件筛选条件，比如根据文件名后缀、文件大小等进行筛选。

缺点：

性能问题： 当目录树很大时，os.walk可能会导致性能问题，因为它需要遍历整个目录树。

二、GLOB模块

glob模块用于查找符合特定模式的文件路径。与os.walk不同，glob模块适合用于简单的文件模式匹配。

2.1 glob.glob的基本用法

glob.glob方法接受一个模式字符串，返回匹配该模式的文件路径列表。模式字符串可以包含通配符，比如'*'表示匹配任意数量的字符，'?'表示匹配单个字符。

import glob
查找当前目录下所有的txt文件
txt_files = glob.glob('*.txt')
for file in txt_files:
    print(file)

2.2 glob模块的优缺点

优点：

简单易用： glob模块使用简单，适合用于简单的文件匹配场景。
效率高： glob模块直接返回匹配的文件路径列表，避免了不必要的遍历。

缺点：

功能有限： glob模块不支持复杂的文件筛选条件，只能进行简单的模式匹配。

三、PATHLIB模块

pathlib模块是Python 3.4引入的面向对象的路径处理库，提供了更为简洁和现代的文件查找方法。

3.1 pathlib.Path的基本用法

Path类的rglob方法可以递归地查找匹配特定模式的文件。与glob模块类似，rglob方法接受一个模式字符串，返回匹配该模式的文件路径生成器。

from pathlib import Path
查找当前目录及子目录下所有的txt文件
for file in Path('/path/to/directory').rglob('*.txt'):
    print(file)

3.2 pathlib模块的优缺点

优点：

现代化： pathlib是Python 3.4引入的现代化路径处理库，提供了面向对象的接口。
简洁： pathlib简化了路径操作，使代码更加直观和易读。

缺点：

兼容性： pathlib仅在Python 3.4及以上版本可用，对于低版本的Python不适用。

四、OS.LISTDIR方法

os.listdir方法用于获取指定目录下的文件和子目录列表。与os.walk不同，os.listdir只获取当前目录下的文件和目录，不会递归遍历子目录。

4.1 os.listdir的基本用法

os.listdir方法返回指定目录下的文件和子目录名列表。

import os
获取当前目录下的文件和子目录列表
entries = os.listdir('/path/to/directory')
for entry in entries:
    print(entry)

4.2 os.listdir的优缺点

优点：

简单： os.listdir使用简单，适合用于获取当前目录下的文件和子目录列表。

缺点：

不支持递归： os.listdir不支持递归遍历子目录，适用于无需递归的简单场景。

五、总结

在Python中查找文件的方法多种多样，选择适合的方法取决于具体的需求和使用场景。对于需要递归遍历目录树的情况，os.walk是一个强大的工具；对于简单的文件模式匹配，glob模块和pathlib模块提供了简洁的解决方案；而对于仅需获取当前目录下文件和子目录列表的情况，os.listdir是最简单的方法。理解和掌握这些方法的优缺点，可以帮助我们在不同的场景中灵活选择合适的工具。