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


    前面章节中,我们学会了使用 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 模块常用函数
    函数名功能描述
    fileinput.filename()返回当前正在读取的文件名称。
    fileinput.fileno()返回当前正在读取文件的文件描述符。
    fileinput.lineno()返回当前读取了多少行。
    fileinput.filelineno()返回当前正在读取的内容位于当前文件中的行号。
    fileinput.isfirstline()判断当前读取的内容在当前文件中是否位于第 1 行。
    fileinput.nextfile()关闭当前正在读取的文件,并开始读取下一个文件。
    fileinput.close()关闭 FileInput 对象。

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


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

    #file.txt
    Python教程
    http://c.biancheng.net/python/

    #my_file.txt
    Linux教程
    http://c.biancheng.net/linux_tutorial/

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

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

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

    程序执行结果为:

    Linux教程

    http://c.biancheng.net/linux_tutorial/
    Python教程

    http://c.biancheng.net/python/

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

  • 相关阅读:
    【数据结构】循环队列的实现
    安卓期末大作业——售票APP源码和设计报告
    8年测试老鸟总结,Python自动化测试实现思路(细致)
    每日一题:【LeetCode】258. 各位相加
    php不解压zip,直接读取zip内容
    并查集(Union-Find)
    如何通过python爬股票接口获取证券交易日?
    alb和clb区别
    基于java+SpringBoot+VUE+Mysql+微信小程序物业管理系统
    Ambari升级Atlas1.1.0到2.2.0
  • 原文地址:https://blog.csdn.net/LLC25802580/article/details/123817317