Python高效遍历文件并根据后缀名进行筛选和处理337

在日常的编程工作中，我们经常需要处理大量的文件，而这些文件通常具有不同的后缀名，例如 `.txt`、`.csv`、`.jpg`、`.pdf` 等。有效地遍历文件系统并根据文件的后缀名进行筛选和处理，是提高工作效率的关键。Python 提供了丰富的库和功能，可以轻松实现这一目标。本文将详细介绍几种 Python 遍历文件并根据后缀名进行处理的方法，并对效率进行比较，帮助你选择最适合自己场景的方案。

方法一：使用 `` 和 ``

这是最基础也是最直接的方法。`` 可以列出指定目录下的所有文件和子目录名称，`` 可以将文件名拆分成文件名和扩展名两部分。通过循环遍历文件列表，并使用 `` 判断文件后缀名，我们可以实现对特定后缀名的文件进行处理。```python
import os
def process_files_by_extension(directory, extension):
"""
遍历指定目录，处理指定后缀名的文件。
Args:
directory: 目录路径。
extension: 文件后缀名（例如 '.txt'）。
"""
for filename in (directory):
base, ext = (filename)
if ext == extension:
filepath = (directory, filename)
# 在此处添加你想要对文件的处理逻辑
print(f"Processing file: {filepath}")
try:
with open(filepath, 'r', encoding='utf-8') as f: #处理文本文件示例
contents = ()
# 对文件内容进行处理
print(f"File content (truncated): {contents[:50]}...")
except UnicodeDecodeError:
print(f"Skipping non-text file: {filepath}")
except Exception as e:
print(f"Error processing file {filepath}: {e}")

# 使用示例
directory_to_process = "./my_files" # 替换成你的目录
target_extension = ".txt"
process_files_by_extension(directory_to_process, target_extension)
```

方法二：使用 `glob` 模块

`glob` 模块提供了一种更简洁的方式来匹配文件路径。它使用通配符模式来查找符合条件的文件。对于需要根据后缀名筛选文件的情况，`glob` 模块非常高效。```python
import glob
def process_files_with_glob(directory, extension):
"""
使用glob模块遍历指定目录，处理指定后缀名的文件。
Args:
directory: 目录路径。
extension: 文件后缀名（例如 '.txt'）。
"""
filepaths = ((directory, f"*{extension}"))
for filepath in filepaths:
# 在此处添加你想要对文件的处理逻辑
print(f"Processing file: {filepath}")
try:
with open(filepath, 'r', encoding='utf-8') as f:
contents = ()
# 对文件内容进行处理
print(f"File content (truncated): {contents[:50]}...")
except UnicodeDecodeError:
print(f"Skipping non-text file: {filepath}")
except Exception as e:
print(f"Error processing file {filepath}: {e}")
# 使用示例
directory_to_process = "./my_files"
target_extension = ".txt"
process_files_with_glob(directory_to_process, target_extension)
```

方法三：使用 `pathlib` 模块 (Python 3.4+)

`pathlib` 模块提供了一种面向对象的方式来操作文件和目录，它更加简洁易读，并且支持更多的功能。使用 `pathlib` 可以更优雅地处理文件路径和筛选。```python
import pathlib
def process_files_with_pathlib(directory, extension):
"""
使用pathlib模块遍历指定目录，处理指定后缀名的文件。
Args:
directory: 目录路径。
extension: 文件后缀名（例如 '.txt'）。
"""
path = (directory)
for file_path in (f"*{extension}"):
# 在此处添加你想要对文件的处理逻辑
print(f"Processing file: {file_path}")
try:
with open(file_path, 'r', encoding='utf-8') as f:
contents = ()
# 对文件内容进行处理
print(f"File content (truncated): {contents[:50]}...")
except UnicodeDecodeError:
print(f"Skipping non-text file: {file_path}")
except Exception as e:
print(f"Error processing file {file_path}: {e}")
# 使用示例
directory_to_process = "./my_files"
target_extension = ".txt"
process_files_with_pathlib(directory_to_process, target_extension)
```