优化Python脚本：高效灵活地从多文本文件中提取特定行-Python教程-PHP中文网

优化Python脚本：高效灵活地从多文本文件中提取特定行

碧海醫心

发布： 2025-12-05 11:08:49

原创

623人浏览过

优化Python脚本：高效灵活地从多文本文件中提取特定行

本教程旨在解决从目录中多个文本文件提取特定行时遇到的硬编码和效率问题。我们将介绍如何利用python的`pathlib`模块简化文件操作，并通过单次文件读取和条件判断来优化数据提取流程。此外，还将探讨如何使用正则表达式实现更灵活的模式匹配，并讨论健壮的数据处理策略，包括默认值设置和异常处理，以提升脚本的专业性和可维护性。

在处理大量文本文件时，我们经常需要从中提取特定的信息。一个常见的场景是，从一个包含多个日志或报告文件的目录中，根据特定的关键词或模式，抽取相关数据并汇总。然而，不当的实现方式可能导致脚本效率低下、难以维护，并且对文件内容的变化缺乏适应性。

原始脚本存在以下主要问题：

重复文件读取： 对于每个需要提取的信息，脚本都会重新打开并读取整个文件。这在处理大文件或大量文件时会造成显著的性能开销。
硬编码的字符串切片： 使用固定的索引进行字符串切片（如 linea[31:-5]）使得脚本对文件格式的变化非常敏感。一旦源文件的格式稍有调整，脚本就可能失效。
变量命名和结构： 变量命名不够直观，且数据提取逻辑分散，降低了代码的可读性。

为了克服这些限制，我们将介绍一种更高效、更灵活且更具可维护性的Python解决方案。

优化文件处理流程

核心优化在于避免重复的文件读取，并采用更智能的方式来定位和提取信息。我们将使用Python的pathlib模块来处理文件路径，这比传统的os模块提供了更面向对象的接口，使文件操作更加简洁和安全。

立即学习“Python免费学习笔记（深入）”；

改进后的代码结构

以下是优化后的Python脚本，它通过一次性遍历文件内容来提取所有所需信息，并利用pathlib进行文件路径管理：

from pathlib import Path
import re # 导入re模块用于正则表达式

def extract_lines(input_file_path: Path, output_handle):
    """
    从单个输入文件中提取特定信息，并写入输出句柄。

    Args:
        input_file_path (Path): 输入文件的Path对象。
        output_handle: 已打开的输出文件句柄。
    """
    # 初始化默认值，以防某些信息未找到
    lasinfo = 'No filename defined!'
    projcs = 'No DATUM defined!'
    pdens = 'No point density listed'
    pdensnum = ''

    with open(input_file_path, 'r') as file_lines:
        for line in file_lines:
            # 使用startswith进行初步筛选，提高效率
            if line.startswith('lasinfo'):
                # 使用正则表达式更精确地提取文件名
                match = re.search(r"report for '([^']+)'", line)
                if match:
                    lasinfo = match.group(1)
                else:
                    # 如果正则表达式匹配失败，可以回退到旧的切片逻辑或设置默认值
                    lasinfo = line.strip()[29:-2] 

            elif line.startswith('    PROJCS'):
                # 移除前后空白，再切片
                projcs = line.strip()[11:39] 

            elif line.startswith('point density'):
                # 使用正则表达式提取点密度数值，提高鲁棒性
                # := 是Python 3.8+ 的海象运算符，允许在表达式中赋值
                if (match := re.match(r'^point density: all returns ([\d.]+)', line)):
                    pdensnum = float(match.group(1)) # 提取为浮点数
                    pdens = line.strip() # 整个密度行
                else:
                    # 如果正则表达式匹配失败，可以回退到旧的切片逻辑
                    pdens = line.strip() 
                    pdensnum = pdens[27:31] 

    # 将所有提取到的信息组合成一行
    # 注意：如果pdensnum是float，需要转换回字符串
    lineout = ",".join([lasinfo, projcs, pdens, str(pdensnum)]) + "\n"
    output_handle.write(lineout)

def process_txt_files(directory_path: str, output_file_name: str):
    """
    遍历指定目录下的所有.txt文件，提取信息并写入一个汇总文件。

    Args:
        directory_path (str): 待处理文件所在的目录路径。
        output_file_name (str): 输出汇总文件的名称。
    """
    dir_path_obj = Path(directory_path)

    # 使用'w'模式清空或创建输出文件，然后使用该句柄进行后续写入
    with open(output_file_name, 'w') as output_handle:
        for file_path_obj in dir_path_obj.iterdir():
            # 确保只处理文件且文件扩展名为.txt
            if file_path_obj.is_file() and file_path_obj.suffix == ".txt":
                extract_lines(file_path_obj, output_handle)

if __name__ == '__main__':
    # 定义输入目录和输出文件
    directory_path = 'C:/Users/rinicholls/Richard/Gnarabup_LiDAR/LiDAR/Gnarabup_South_AVWS/reports'
    output_file = 'density.txt'

    # 执行文件处理
    process_txt_files(directory_path, output_file)

登录后复制

代码解析与改进点：

pathlib 的使用：
- from pathlib import Path 引入 Path 对象，提供更现代、面向对象的文件系统路径操作。
- Path(directory_path) 将字符串路径转换为 Path 对象，方便后续操作。
- dir_path_obj.iterdir() 迭代目录中的所有文件和子目录，返回 Path 对象。
- file_path_obj.is_file() 检查是否是文件，file_path_obj.suffix == ".txt" 检查文件扩展名。
- 这使得文件路径操作更直观、更安全，并减少了 os.path.join 的使用。
单次文件读取：
- 在 extract_lines 函数中，使用一个 with open(..., 'r') as file_lines: 语句打开文件。
- 通过 for line in file_lines: 循环，逐行读取文件内容。在一次遍历中，通过一系列 if/elif 条件判断来查找并提取所有目标信息。这显著提高了效率，避免了多次打开和读取同一文件。
默认值初始化：
- 在 extract_lines 函数开始时，为所有待提取的信息设置了默认值（如 'No filename defined!'）。这样，即使某些信息在文件中未找到，输出结果也能保持一致性，并明确指示缺失。
字符串处理优化：

Canva AI
Canva平台AI图片生成工具

1285

查看详情
- line.strip() 用于移除行首尾的空白字符，包括换行符，这在进行字符串切片或匹配前非常有用。
- str.startswith() 用于快速判断行是否以特定前缀开始，这比 in 操作符更高效，因为它不需要扫描整个字符串。

使用正则表达式增强灵活性

尽管 startswith 和简单的切片可以在某些情况下工作，但当需要从行中提取模式化的数据时，正则表达式（Regular Expressions, re模块）提供了无与伦比的灵活性和鲁棒性。

示例：使用正则表达式提取文件名和点密度

在上述代码中，我们已经初步集成了正则表达式来处理 lasinfo 和 point density 行。

提取文件名 (lasinfo 行):
```
match = re.search(r"report for '([^']+)'", line)
if match:
    lasinfo = match.group(1)
```
登录后复制
这里的 r"report for '([^']+)'" 是一个正则表达式：
- r 表示原始字符串，避免反斜杠转义问题。
- report for ' 匹配字面字符串。
- ([^']+) 是一个捕获组：
  - [^'] 匹配除单引号外的任何字符。
  - + 表示匹配一个或多个这样的字符。
  - 括号 () 将其标记为一个捕获组，我们可以通过 match.group(1) 来获取匹配到的内容（即文件名）。
提取点密度数值 (point density 行):
```
if (match := re.match(r'^point density: all returns ([\d.]+)', line)):
    pdensnum = float(match.group(1))
```
登录后复制
这里的 r'^point density: all returns ([\d.]+)' 解释如下：
- ^ 匹配行的开始。
- point density: all returns 匹配字面字符串。
- ([\d.]+) 是一个捕获组：
  - \d 匹配任何数字 (0-9)。
  - . 匹配字面点。
  - + 匹配一个或多个数字或点。这可以捕获像 "0.25" 这样的浮点数。
- float(match.group(1)) 将提取到的字符串转换为浮点数，这对于后续的数据分析非常有用。如果需要写入文件，记得将其转换回字符串。