fileinput模块提供处理一个或多个文本文件的功能,可以通过使用for循环来读取一个或多个文本文件的所有行。它的工作方式和readlines很类似,不同点在于它不是将全部的行读到列表中而是创建了一个xreadlines对象。
下面是fileinput模块中的常用函数:
input() #返回能够用于for循环遍历的对象filename() #返回当前文件的名称lineno() #返回当前已经读取的行的数量(或者序号)filelineno() #返回当前读取的行的行号isfirstline() #检查当前行是否是文件的第一行
创建测试文件test.txt:
# cat > test.txt > import fileinput>>> for line in fileinput.input('test.txt'):... print fileinput.filename(),fileinput.lineno(),fileinput.filelineno()...test.txt 1 1test.txt 2 2test.txt 3 3
>>> import fileinput>>> for line in fileinput.input('test.txt'):... if fileinput.isfirstline():... print line,... else:... break...hello,linuxeye