Python：fileinput操作文件

fileinput模块提供了一个以行模型循环处理一个或多个文件内容的功能，它实现了对文件中行的“懒惰”迭代，读取时不需要把文件内容放入内存，这样可以提高程序的效率。

fileinput模块中常用的函数有：

• fileinput.input()：返回能够用于迭代一个或多个文件中所有行的对象。
• fileinput.filename()：返回当前文件的名称。
• fileinput.lineno()：返回当前读取的行的数量（总得行数）。
• fileinput.isfirstline()：返回当前行，判断是否是文件的第一行，是当前文件的第一行时返回True，反之返回False。
• fileinput.filelineno()：返回当前读取行在文件中的行数（哪个文件夹中的第几行）。

fileinput支持上下文管理器，可以使用with语句来进行操作，而不是在使用后手工关闭对象f.close()。

import fileinput


def demo_fileinput():
    with fileinput.input(['fpa.txt', 'fpb.txt']) as lines:
        for line in lines:
            print("总第%d行," % fileinput.lineno(),
                  "文件%s中第%d行：" % (fileinput.filename(), fileinput.filelineno()))
            print(line.strip())


if __name__ == '__main__':
    demo_fileinput()

输出结果：

总第1行, 文件fpa.txt中第1行：
fpa-1
总第2行, 文件fpa.txt中第2行：
fpa-2
总第3行, 文件fpa.txt中第3行：
fpa-3
总第4行, 文件fpb.txt中第1行：
fpb-1
总第5行, 文件fpb.txt中第2行：
fpb-2
总第6行, 文件fpb.txt中第3行：
fpb-3