Python fileinput模块：逐行读取多个文件

川长思鸟来 2023-10-08 17:47 5阅读 0赞

![0fd33084823a28634943f2516f56ccff.png][]

前面章节中，我们学会了使用 open() 和 read()（或者 readline()、readlines() ）组合，来读取单个文件中的数据。但在某些场景中，可能需要读取多个文件的数据，这种情况下，再使用这个组合，显然就不合适了。

庆幸的是，Python 提供了 fileinput 模块，通过该模块中的 input() 函数，我们能同时打开指定的多个文件，还可以逐个读取这些文件中的内容。

fileinput 模块中 input() 该函数的语法格式如下：

> fileinput.input（files="filename1, filename2, ...", inplace=False, backup='', bufsize=0, mode='r', openhook=None）

此函数会返回一个 FileInput 对象，它可以理解为是将多个指定文件合并之后的文件对象。其中，各个参数的含义如下：

*  files：多个文件的路径列表；

*  inplace：用于指定是否将标准输出的结果写回到文件，此参数默认值为 False；

*  backup：用于指定备份文件的扩展名；

*  bufsize：指定缓冲区的大小，默认为 0；

*  mode：打开文件的格式，默认为 r（只读格式）；

*  openhook：控制文件的打开方式，例如编码格式等。

> 注意，和 open() 函数不同，input() 函数不能指定打开文件的编码格式，这意味着使用该函数读取的所有文件，除非以二进制方式进行读取，否则该文件编码格式都必须和当前操作系统默认的编码格式相同，不然 Python 解释器可能会提示 UnicodeDecodeError 错误。

和 open() 函数返回单个的文件对象不同，fileinput 对象无需调用类似 read()、readline()、readlines() 这样的函数，直接通过 for 循环即可按次序读取多个文件中的数据。

值得一提的是，fileinput 模块还提供了很多使用的函数（如表 1 所示），通过调用这些函数，可以帮我们更快地实现想要的功能。

表 1 fileinput 模块常用函数

<table style=""> 
 <tbody> 
 <tr> 
 <td style="width:369px;vertical-align:top;height:51px;">函数名</td> 
 <td style="width:369px;vertical-align:top;height:51px;">功能描述</td> 
 </tr> 
 <tr> 
 <td style="width:369px;vertical-align:top;height:51px;">fileinput.filename()</td> 
 <td style="width:369px;vertical-align:top;height:51px;">返回当前正在读取的文件名称。</td> 
 </tr> 
 <tr> 
 <td style="width:369px;vertical-align:top;height:51px;">fileinput.fileno()</td> 
 <td style="width:369px;vertical-align:top;height:51px;">返回当前正在读取文件的文件描述符。</td> 
 </tr> 
 <tr> 
 <td style="width:369px;vertical-align:top;height:51px;">fileinput.lineno()</td> 
 <td style="width:369px;vertical-align:top;height:51px;">返回当前读取了多少行。</td> 
 </tr> 
 <tr> 
 <td style="width:369px;vertical-align:top;height:51px;">fileinput.filelineno()</td> 
 <td style="width:369px;vertical-align:top;height:51px;">返回当前正在读取的内容位于当前文件中的行号。</td> 
 </tr> 
 <tr> 
 <td style="width:369px;vertical-align:top;height:51px;">fileinput.isfirstline()</td> 
 <td style="width:369px;vertical-align:top;height:51px;">判断当前读取的内容在当前文件中是否位于第 1 行。</td> 
 </tr> 
 <tr> 
 <td style="width:369px;vertical-align:top;height:51px;">fileinput.nextfile()</td> 
 <td style="width:369px;vertical-align:top;height:51px;">关闭当前正在读取的文件，并开始读取下一个文件。</td> 
 </tr> 
 <tr> 
 <td style="width:369px;vertical-align:top;height:51px;">fileinput.close()</td> 
 <td style="width:369px;vertical-align:top;height:51px;">关闭 FileInput 对象。</td> 
 </tr> 
 </tbody> 
</table>

文件描述符是一个文件的代号，其值为一个整数。后续章节将会介绍关于文件描述符的操作。

讲了这么多，接下来举个例子。假设使用 input() 读取 2 个文件，分别为 my\_file.txt 和 file.txt，它们位于同一目录，且各自包含的内容如下所示：

> \#file.txt  
> Python教程  
> [http://csdn.net/python/][http_csdn.net_python]  
>   
> \#my\_file.txt  
> Linux教程  
> [http://csdn.net/linux\_tutorial/][http_csdn.net_linux_tutorial]

下面程序演示了如何使用 input() 函数依次读取这 2 个文件：

import fileinput
    #使用for循环遍历 fileinput 对象
    for line in fileinput.input(files=('my_file.txt','file.txt')):
    # 输出读取到的内容
    print(line)
    # 关闭文件流
    fileinput.close()

> 在使用 fileinput 模块中的 input() 函数之前，一定要先引入 fileinput 模块。

程序执行结果为：

> Linux教程  
>   
> [http://csdn.net/linux\_tutorial/][http_csdn.net_linux_tutorial]  
> Python教程  
>   
> [http://csdn.net/python/][http_csdn.net_python]

显然，读取文件内容的次序，取决于 input() 函数中文件名的先后次序。

[0fd33084823a28634943f2516f56ccff.png]: https://img-blog.csdnimg.cn/img_convert/0fd33084823a28634943f2516f56ccff.png
[http_csdn.net_python]: http://csdn.net/python/
[http_csdn.net_linux_tutorial]: http://csdn.net/linux_tutorial/