fileinput模塊提供處理一個或多個文本文件的功能,可以通過使用for循環來讀取一個或多個文本文件的所有行。它的工作方式和readlines很類似,不同點在于它不是將全部的行讀到列表中而是創建了一個xreadlines對象。
下面是fileinput模塊中的常用函數:
input() #返回能夠用于for循環遍歷的對象filename() #返回當前文件的名稱lineno() #返回當前已經讀取的行的數量(或者序號)filelineno() #返回當前讀取的行的行號isfirstline() #檢查當前行是否是文件的第一行
創建測試文件test.txt:
# cat > test.txt << EOFHello,Pythonwww.jb51.netThis is a test fileEOF
利用fileinput實現文件內容替換,如:file_input.p(注意文件名,別寫成fileinput.py)
#!/usr/bin/env pythonimport fileinputfor 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.txtHello,LinuxEyewww.jb51.netThis is a test file # cat test.txt_bakHello,Pythonwww.jb51.netThis is a test file
其他測試:
>>> 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
新聞熱點
疑難解答