fileinput模塊提供處理一個或多個文本文件的功能,可以通過使用for循環來讀取一個或多個文本文件的所有行。它的工作方式和readlines很類似,不同點在于它不是將全部的行讀到列表中而是創建了一個xreadlines對象。
下面是fileinput模塊中的常用函數:
input() #返回能夠用于for循環遍歷的對象 filename() #返回當前文件的名稱 lineno() #返回當前已經讀取的行的數量(或者序號) filelineno() #返回當前讀取的行的行號 isfirstline() #檢查當前行是否是文件的第一行
創建測試文件test.txt:
# cat > test.txt << EOF
Hello,Python
www.jfrwli.cn
This is a test file
EOF
利用fileinput實現文件內容替換,如:file_input.p(注意文件名,別寫成fileinput.py)
1
2
3
4
5
|
#!/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.jfrwli.cn This is a test file # cat test.txt_bak Hello,Python www.jfrwli.cn 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
1
2
3
4
5
6
7
8
|
>>> import fileinput >>> for line in fileinput. input ( 'test.txt' ): ... if fileinput.isfirstline(): ... print line, ... else : ... break ... Hello,LinuxEye |