python如何选取多个目录

Python选取多个目录的方法有多种，包括使用os模块、glob模块、以及pathlib模块。os模块提供了操作系统相关的功能，glob模块可以用于查找符合特定规则的文件路径，而pathlib模块则是Python 3.4引入的面向对象的路径操作模块。通过这些模块，您可以方便地遍历多个目录、读取文件、或者进行批量处理。

一、OS模块

os模块是Python标准库的一部分，提供了多种与操作系统交互的方法。

1. 使用os.walk遍历目录

os.walk是一个用于遍历目录树的生成器，返回一个三元组 (dirpath, dirnames, filenames)。

import os
def list_files_in_directories(directories):
    all_files = []
    for directory in directories:
        for dirpath, dirnames, filenames in os.walk(directory):
            for filename in filenames:
                filepath = os.path.join(dirpath, filename)
                all_files.append(filepath)
    return all_files
directories = ['/path/to/dir1', '/path/to/dir2']
files = list_files_in_directories(directories)
print(files)

2. 使用os.listdir列出目录内容

os.listdir返回指定路径下的文件和目录名列表。

import os
def list_directories_content(directories):
    all_contents = {}
    for directory in directories:
        try:
            contents = os.listdir(directory)
            all_contents[directory] = contents
        except FileNotFoundError as e:
            print(f"Directory not found: {directory}")
    return all_contents
directories = ['/path/to/dir1', '/path/to/dir2']
contents = list_directories_content(directories)
print(contents)

二、GLOB模块

glob模块提供了一个简易的文件名模式匹配功能。

1. 使用glob.glob匹配文件路径

glob可以通过通配符来匹配目录中的文件。

import glob
def find_files_with_pattern(directories, pattern):
    matched_files = []
    for directory in directories:
        matched_files.extend(glob.glob(os.path.join(directory, pattern)))
    return matched_files
directories = ['/path/to/dir1', '/path/to/dir2']
pattern = '*.txt'
files = find_files_with_pattern(directories, pattern)
print(files)

三、PATHLIB模块

pathlib是Python 3.4引入的模块，提供了面向对象的文件系统路径操作。

1. 使用Path对象遍历目录

from pathlib import Path
def list_files_with_pathlib(directories):
    all_files = []
    for directory in directories:
        path = Path(directory)
        for file in path.rglob('*'):
            if file.is_file():
                all_files.append(file)
    return all_files
directories = ['/path/to/dir1', '/path/to/dir2']
files = list_files_with_pathlib(directories)
print(files)

2. 使用Path.glob匹配文件

from pathlib import Path
def match_files_with_pathlib(directories, pattern):
    matched_files = []
    for directory in directories:
        path = Path(directory)
        matched_files.extend(path.glob(pattern))
    return matched_files
directories = ['/path/to/dir1', '/path/to/dir2']
pattern = '*.txt'
files = match_files_with_pathlib(directories, pattern)
print(files)

四、批量处理文件

在处理多个目录中的文件时，可能需要对文件执行一些批量操作，如读取文件内容、修改文件等。

1. 读取文件内容

def read_files(files):
    file_contents = {}
    for file in files:
        with open(file, 'r') as f:
            file_contents[file] = f.read()
    return file_contents
files = list_files_with_pathlib(['/path/to/dir1', '/path/to/dir2'])
contents = read_files(files)
print(contents)

2. 修改文件内容

def modify_file_contents(files, old_str, new_str):
    for file in files:
        with open(file, 'r') as f:
            content = f.read()
        content = content.replace(old_str, new_str)
        with open(file, 'w') as f:
            f.write(content)
files = find_files_with_pattern(['/path/to/dir1', '/path/to/dir2'], '*.txt')
modify_file_contents(files, 'old', 'new')

五、处理文件异常

在批量处理文件时，可能会遇到文件读取错误、权限错误等异常情况，需要进行适当的异常处理。

1. 捕获文件读取错误

def read_files_with_error_handling(files):
    file_contents = {}
    for file in files:
        try:
            with open(file, 'r') as f:
                file_contents[file] = f.read()
        except FileNotFoundError:
            print(f"File not found: {file}")
        except PermissionError:
            print(f"Permission denied: {file}")
    return file_contents
files = list_files_with_pathlib(['/path/to/dir1', '/path/to/dir2'])
contents = read_files_with_error_handling(files)
print(contents)

六、总结

在Python中，有多种方法可以实现选取多个目录并进行文件操作。os模块适用于需要操作系统交互的场景，glob模块提供了简单的文件路径匹配功能，而pathlib模块通过面向对象的方式简化了路径操作。通过结合这些模块，您可以高效地实现批量目录和文件处理任务。在实际应用中，根据具体需求选择合适的方法和模块，可以提高代码的可读性和维护性。