這篇文章主要介紹了Python fileinput模塊使用實例,fileinput模塊可以遍歷文本文件的所有行,本文就給出它的使用代碼實例,需要的朋友可以參考下
fileinput模塊可以遍歷文本文件的所有行.它的工作方式和readlines很類似,不同點在於,它不是將全部的行讀到列表中而是創建了一個xreadlines對象.
下面是fileinput模塊中的常用函數
input() #它會返回能夠用於for循環遍歷的對象.
filename() #返回當前文件的名稱
lineno() #返回當前(累計)的行數
filelineno() #返回當前文件的行數
isfirstline() #檢查當前行是否是文件的第一行
代碼如下:
#!/bin/env python
#coding=utf-8
import fileinput
import sys
import glob
import string
''' 處理一個文本文件 '''
for line in fileinput.input("tab.sh"):
print line
'''處理多個文本文件 並輸出行號'''
for line in fileinput.input(glob.glob("*.sh")):
if fileinput.isfirstline():
print "------ reading %s ------n" % fileinput.filename()
print str(fileinput.lineno()) + " " + line
例子：
代碼如下:
[root@rac1 admin]# python ftest.py
abc_permission_collection
abc_user
abc_user_group_relationship
------ reading c.sh ------
1 c
2 cc
3 ccc
------ reading tab.sh ------
4 abc_permission_collection
5 abc_user
6 abc_user_group_relationship
[root@rac1 admin]#
