在測(cè)試過程中經(jīng)常需要對(duì)程序運(yùn)行后產(chǎn)生的結(jié)果文件進(jìn)行分析，一兩個(gè)文件直接打開看結(jié)果，但是幾十個(gè)文件一個(gè)一個(gè)打開效率就太低，所以為了提高效率才想到用python來解決這個(gè)問題，結(jié)果文件有好幾種類型的，但是我們只想讀取其中一個(gè)格式的文件。于是下面就有了下面的代碼

只讀取data.txt里面指定的內(nèi)容

讀取txt里面的第一行

我這里只是個(gè)示范，具體的用時(shí)候可以使用正側(cè)表達(dá)式來配置

代碼如下

#coding=utf-8
'''
在很多文件下面只讀取txt類型的文件，然后提取相關(guān)的數(shù)據(jù)

'''
import os
path = "./data" #當(dāng)前路徑,可以使絕對(duì)路徑
files = os.listdir(path) #列出path目錄下面的所有文件名，并且返回一個(gè)list

for file in files:
        file_ext = file.split('.')[1] #使用split分割文件名和后綴名 分割成一個(gè)list['xxxx','txt']，獲取后綴
        ext = ['txt'] #期望讀取的文件后綴
        if file_ext not in ext: #如果后綴名不在ext里面，則跳過本次循環(huán)，開始下次循環(huán)
                continue
        else:
                f = open(path + "/" + file,'r') 
                for s in f.readlines()[0:1]: #讀取第一行的內(nèi)容，可以實(shí)際情況來使用啥方法
                    print s,



f.close()

以上代碼希望能給大家提供點(diǎn)思路，當(dāng)然大家可以把結(jié)果寫入到txt中保存起來。