fileinput模块可以对一个或多个文件中的内容进行迭代、遍历等操作。该模块的input()函数有点类似文件readlines()方法,区别在于前者是一个迭代对象,需要用for循环迭代,后者是一次性读取所有行。用fileinput对文件进行循环遍历,格式化输出,查找、替换等操作,非常方便。
【默认格式】
fileinput.input (files='filename', inplace=False, backup='', bufsize=0, mode='r', openhook=None)
1 files: #文件的路径列表,默认是stdin方式,多文件['1.txt','2.txt',...] 2 inplace: #是否将标准输出的结果写回文件,默认不取代 3 backup: #备份文件的扩展名,只指定扩展名,如.bak。如果该文件的备份文件已存在,则会自动覆盖。 4 bufsize: #缓冲区大小,默认为0,如果文件很大,可以修改此参数,一般默认即可 5 mode: #读写模式,默认为只读 6 openhook: #该钩子用于控制打开的所有文件,比如说编码方式等;
【常用函数】
1 fileinput.input() #返回能够用于for循环遍历的对象 2 fileinput.filename() #返回当前文件的名称 3 fileinput.lineno() #返回当前已经读取的行的数量(或者序号) 4 fileinput.filelineno() #返回当前读取的行的行号 5 fileinput.isfirstline() #检查当前行是否是文件的第一行 6 fileinput.isstdin() #判断最后一行是否从stdin中读取 7 fileinput.close() #关闭队列
1、使用fileinput查看文件所有行
import fileinput
for line in fileinput.input('user.txt'):
print(line)
1、使用fileinput查看文件所有行
import fileinput
for line in fileinput.input('e:/test.txt'):
print(line)
2、使用fileinput修改文件并备份原文件
import fileinput
for line in fileinput.input('user.txt',backup='.bak',inplace=1):
line = line.replace('Rain','Jerry')
print(line)
3、使用fileinput输出当前行号和行内容
#cat user.txt Jack Rain Tom Suse #################################################### #cat f_replace.py #!/usr/bin/env python import fileinput for line in fileinput.input('user.txt'): lineno = fileinput.lineno() print(lineno,line) #################################################### #执行代码 #python f_replace.py #输出结果 1 Jack 2 Rain 3 Tom 4 Suse
4、使用fileinput判断是否为第一行
#cat user.txt Jack Rain Tom Suse #################################################### #cat f_replace.py #!/usr/bin/env python import fileinput for line in fileinput.input('user.txt'): if fileinput.isfirstline(): print(line) break #################################################### #执行代码 #python f_replace.py #输出结果 Jack