Python fileinput模块：迭代遍历与高效操作实践

Python的fileinput模块是一个强大的工具，它允许程序员在一个或多个文件中进行迭代式处理，支持逐行读取并执行相应的操作。该模块的`input()`函数是其核心功能，它类似于内置的`open()`函数与`readlines()`方法的结合体。与`readlines()`不同，`input()`返回的是一个迭代器，每次循环只提供一行数据，这对于处理大型文件时尤为有利，因为它避免了一次性加载整个文件到内存中，从而节省了内存资源。 `fileinput.input()`函数的典型用法是在循环中逐行处理文件内容，例如格式化输出、查找和替换等操作。其基本格式如下： ```python import fileinput # 迭代文件并处理每一行 for line in fileinput.input(['file1.txt', 'file2.txt']): process(line) ``` 这个函数接受多个可选参数，如： - `files`：一个文件路径列表，如果不指定，将使用标准输入（stdin）处理。可以处理多个文件。 - `inplace`：布尔值，表示是否将处理结果写回原始文件，默认为False，意味着不覆盖原文件。 - `backup`：备份文件的扩展名，用于创建备份，如果备份文件已存在则会覆盖。 - `bufsize`：缓冲区大小，一般默认值即可，但如果处理大文件，可调整以提高性能。 - `mode`：文件打开模式，默认为读模式（'r'），其他模式（如'w'或'a'）可用于写入或追加。 - `openhook`：自定义函数，用于控制文件的打开方式，如设定特定的编码。 除了主函数外，fileinput模块还提供了一些辅助函数，如： - `fileinput.filename()`：获取当前正在处理的文件名。 - `fileinput.lineno()`：返回当前读取的行号，对于迭代器来说是动态变化的。 - `fileinput.filelineno()`：返回文件内部的行号，对于多文件处理时，每个文件有自己的行号计数。 - `fileinput.isfirstline()`：判断当前行是否是文件的开始。 - `fileinput.isstdin()`：检查当前是否来自标准输入。 - `fileinput.close()`：手动关闭迭代器。 一个常见的应用场景是逐行读取文件并执行特定操作，比如在例子01中： ```python >>> import fileinput >>> >>> # 读取文件'example.txt'的所有行 >>> for i, line in enumerate(fileinput.input('example.txt')): ... print(f'Line {i + 1}: {line.strip()}') ``` 在这个例子中，我们使用`enumerate()`来同时获取行号和行内容，方便进行后续处理。 Python的fileinput模块提供了一种灵活且高效的方式来处理多文件或大文件，通过迭代器形式逐行读取和处理，极大地简化了编程任务。掌握这个模块有助于在实际开发中实现文件处理的高效与便捷。