Python fileinput模块使用介绍
fileinput模块提供处理一个或多个文本文件的功能,可以通过使用for循环来读取一个或多个文本文件的所有行。它的工作方式和readlines很类似,不同点在于它不是将全部的行读到列表中而是创建了一个xreadlines对象。
下面是fileinput模块中的常用函数:
input()#返回能够用于for循环遍历的对象 filename()#返回当前文件的名称 lineno()#返回当前已经读取的行的数量(或者序号) filelineno()#返回当前读取的行的行号 isfirstline()#检查当前行是否是文件的第一行
创建测试文件test.txt:
#cat>test.txt<<EOF Hello,Python www.nhooo.com Thisisatestfile EOF
利用fileinput实现文件内容替换,如:file_input.p(注意文件名,别写成fileinput.py)
#!/usr/bin/envpython
importfileinput
forlineinfileinput.input('test.txt',backup='_bak',inplace=1):
printline.replace('Python','LinuxEye'),
fileinput.close()
inplace=1:标准输出会被重定向到打开文件;backup='_bak',:替换文件内容之前备份后缀以_bak结尾;另外,在调用fileinput.input()之后记得要fileinput.close()。
执行结果如下:
#pythonfile_input.py#执行file_input.py #lstest.txt* test.txttest.txt_bak #cattest.txt Hello,LinuxEye www.nhooo.com Thisisatestfile #cattest.txt_bak Hello,Python www.nhooo.com Thisisatestfile
其他测试:
>>>importfileinput
>>>forlineinfileinput.input('test.txt'):
...printfileinput.filename(),fileinput.lineno(),fileinput.filelineno()
...
test.txt11
test.txt22
test.txt33
>>>importfileinput
>>>forlineinfileinput.input('test.txt'):
...iffileinput.isfirstline():
...printline,
...else:
...break
...
Hello,LinuxEye