Python fileinput模块使用介绍


            fileinput模块提供处理一个或多个文本文件的功能，可以通过使用for循环来读取一个或多个文本文件的所有行。它的工作方式和readlines很类似，不同点在于它不是将全部的行读到列表中而是创建了一个xreadlines对象。

下面是fileinput模块中的常用函数：



input()    #返回能够用于for循环遍历的对象

filename()  #返回当前文件的名称

lineno()   #返回当前已经读取的行的数量（或者序号）

filelineno() #返回当前读取的行的行号

isfirstline() #检查当前行是否是文件的第一行

创建测试文件test.txt：



# cat > test.txt << EOF

Hello,Python

www.zwyuanma.com

This is a test file

EOF

利用fileinput实现文件内容替换，如：file_input.p（注意文件名，别写成fileinput.py）



#!/usr/bin/env python

import fileinput

for line in fileinput.input('test.txt',backup='_bak',inplace=1):

  print line.replace('Python','LinuxEye'),

fileinput.close()

inplace=1：标准输出会被重定向到打开文件；backup='_bak',：替换文件内容之前备份后缀以_bak结尾；另外，在调用fileinput.input()之后记得要fileinput.close()。

执行结果如下：



# python file_input.py #执行file_input.py

# ls test.txt*

test.txt test.txt_bak

 

# cat test.txt

Hello,LinuxEye

www.zwyuanma.com

This is a test file

 

# cat test.txt_bak

Hello,Python

www.zwyuanma.com

This is a test file

其他测试：



>>> import fileinput

>>> for line in fileinput.input('test.txt'):

...   print fileinput.filename(),fileinput.lineno(),fileinput.filelineno()

...

test.txt 1 1

test.txt 2 2

test.txt 3 3



>>> import fileinput

>>> for line in fileinput.input('test.txt'):

...   if fileinput.isfirstline():

...     print line,

...   else:

...     break

...

Hello,LinuxEye