如何使用python查找文件夹下

如何使用Python查找文件夹下的文件

答：使用Python查找文件夹下的文件可以通过以下几种方法：os模块、glob模块、pathlib模块。其中，os模块的os.listdir()方法和os.walk()方法非常常用，可以轻松地列出文件夹中的所有文件和子文件夹。glob模块提供了文件名模式匹配功能，可以通过通配符查找文件。pathlib模块是Python 3.4引入的模块，用于处理文件系统路径，提供了更加直观和易用的接口。

详细描述：os模块的os.walk()方法可以递归地遍历目录树，生成文件名，它返回一个三元组(root, dirs, files)，其中root是所指目录的路径名，dirs是root下的子目录列表，files是root下的文件列表。通过os.walk()方法，可以非常方便地获取目标文件夹下的所有文件及其路径。

一、OS模块

1、os.listdir()

os.listdir()方法用于列出指定文件夹下的所有文件和子文件夹，可以通过循环遍历获取每个文件和文件夹的名称。

import os
def list_files(directory):
    for filename in os.listdir(directory):
        print(filename)
directory = '/path/to/directory'
list_files(directory)

上述代码通过os.listdir()方法列出指定路径下的所有文件和子文件夹，并逐一打印出它们的名称。

2、os.walk()

os.walk()方法用于递归地遍历目录树，生成文件名。它返回一个三元组(root, dirs, files)，其中root是所指目录的路径名，dirs是root下的子目录列表，files是root下的文件列表。

import os
def list_files_recursive(directory):
    for root, dirs, files in os.walk(directory):
        for filename in files:
            print(os.path.join(root, filename))
directory = '/path/to/directory'
list_files_recursive(directory)

该代码通过os.walk()方法递归遍历指定路径下的所有文件，并打印出它们的完整路径。

二、GLOB模块

1、glob.glob()

glob模块提供了文件名模式匹配功能，可以通过通配符查找文件。glob.glob()方法返回所有匹配的文件路径列表。

import glob
def find_files(pattern):
    return glob.glob(pattern)
pattern = '/path/to/directory/*.txt'
files = find_files(pattern)
for file in files:
    print(file)

上述代码通过glob.glob()方法查找指定路径下所有匹配模式的文件（例如，所有.txt文件），并打印出它们的路径。

2、glob.iglob()

glob.iglob()方法与glob.glob()类似，但返回的是一个生成器对象，可以逐一获取匹配的文件路径。

import glob
def find_files(pattern):
    return glob.iglob(pattern)
pattern = '/path/to/directory/*.txt'
files = find_files(pattern)
for file in files:
    print(file)

该代码通过glob.iglob()方法查找指定路径下所有匹配模式的文件（例如，所有.txt文件），并逐一打印出它们的路径。

三、PATHLIB模块

1、pathlib.Path.glob()

pathlib模块提供了更加直观和易用的接口，用于处理文件系统路径。pathlib.Path.glob()方法用于查找匹配模式的文件。

from pathlib import Path
def find_files(directory, pattern):
    return Path(directory).glob(pattern)
directory = '/path/to/directory'
pattern = '*.txt'
files = find_files(directory, pattern)
for file in files:
    print(file)

上述代码通过pathlib.Path.glob()方法查找指定路径下所有匹配模式的文件（例如，所有.txt文件），并打印出它们的路径。

2、pathlib.Path.rglob()

pathlib.Path.rglob()方法用于递归查找匹配模式的文件。

from pathlib import Path
def find_files_recursive(directory, pattern):
    return Path(directory).rglob(pattern)
directory = '/path/to/directory'
pattern = '*.txt'
files = find_files_recursive(directory, pattern)
for file in files:
    print(file)

该代码通过pathlib.Path.rglob()方法递归查找指定路径下所有匹配模式的文件（例如，所有.txt文件），并打印出它们的路径。

四、综合应用

在实际应用中，我们可能需要综合使用以上方法来实现更复杂的文件查找功能。例如，我们可以结合使用os.walk()和glob模块来查找特定扩展名的文件，并将其路径存储在列表中。

import os
import glob
def find_specific_files(directory, extension):
    files = []
    for root, dirs, files in os.walk(directory):
        pattern = os.path.join(root, f'*.{extension}')
        files.extend(glob.glob(pattern))
    return files
directory = '/path/to/directory'
extension = 'txt'
files = find_specific_files(directory, extension)
for file in files:
    print(file)

上述代码通过os.walk()方法递归遍历指定路径下的所有文件夹，并结合glob.glob()方法查找特定扩展名的文件（例如，所有.txt文件），最终打印出它们的路径。

五、处理大数据量文件查找

当需要处理大量文件时，可以使用生成器和迭代器来节省内存。例如，可以结合使用os.walk()和yield语句来实现高效的文件查找。

import os
def find_files_generator(directory):
    for root, dirs, files in os.walk(directory):
        for filename in files:
            yield os.path.join(root, filename)
directory = '/path/to/directory'
files = find_files_generator(directory)
for file in files:
    print(file)

该代码通过os.walk()方法递归遍历指定路径下的所有文件夹，并使用yield语句逐一返回文件的路径，从而实现高效的文件查找。

六、查找文件并执行操作

在查找到文件后，我们可能需要对其执行某些操作。以下示例展示了如何查找指定扩展名的文件，并对其进行处理（例如，读取文件内容）。

import os
def process_files(directory, extension):
    for root, dirs, files in os.walk(directory):
        for filename in files:
            if filename.endswith(extension):
                file_path = os.path.join(root, filename)
                with open(file_path, 'r') as file:
                    content = file.read()
                    print(f'Processing file: {file_path}')
                    print(content)
directory = '/path/to/directory'
extension = '.txt'
process_files(directory, extension)