fileinput模块可以对一个或多个文件中的内容进行迭代、遍历等操作。
常用用法:
import fileinput
for line in fileinput.input():
process(line)
基本格式
fileinput.input([files[, inplace[, backup[, bufsize[, mode[, openhook]]]]]])
files: #文件的路径列表,默认是stdin方式,多文件['1.txt','2.txt',...]
inplace: #是否将标准输出的结果写回文件,默认不取代
backup: #备份文件的扩展名,只指定扩展名,如.bak。如果该文件的备份文件已存在,则会自动覆盖。
bufsize: #缓冲区大小,默认为0,如果文件很大,可以修改此参数,一般默认即可
mode: #读写模式,默认为只读
openhook: #该钩子用于控制打开的所有文件,比如说编码方式等;
默认格式
fileinput.input (files=None, inplace=False, backup='', bufsize=0, mode='r', openhook=None)
常用函数
fileinput.input() #返回能够用于for循环遍历的对象
fileinput.filename() #返回当前文件的名称
fileinput.lineno() #返回当前已经读取的行的数量(或者序号)
fileinput.filelineno() #返回当前读取的行的行号
fileinput.isfirstline() #检查当前行是否是文件的第一行
fileinput.isstdin() #判断最后一行是否从stdin中读取
fileinput.close() #关闭队列
使用示例:
例子1:读取一个文件所有行
import fileinput
for line in fileinput.input('data.txt'):
print line #sys.stdout.write(line)
例子2:对多文件操作,并原地修改内容
#对1.txt和2.txt读取并进行修改
import fileinput
def process(line):
return line.rstrip() + ' line'
for line in fileinput.input(['1.txt','2.txt'],inplace=1):
print process(line)
例子3:现文件内容替换,并将原文件作备份
import fileinput
for line in fileinput.input('data.txt',backup='.bak',inplace=1):
print line.rstrip().replace('Python','Perl')
参考:
以上内容仅是参考内容作出个性化整理和删改,如有疑问,可参看以下详细链接:
http://blog.youkuaiyun.com/u012377333/article/details/41956715