使用python输出指定格式文本的内容

在测试过程中经常需要对程序运行后产生的结果文件进行分析，一两个文件直接打开看结果，但是几十个文件一个一个打开效率就太低，所以为了提高效率才想到用python来解决这个问题，结果文件有好几种类型的，但是我们只想读取其中一个格式的文件。于是下面就有了下面的代码

只读取data.txt里面指定的内容 读取txt里面的第一行

我这里只是个示范，具体的用时候可以使用正侧表达式来配置

代码如下

#coding=utf-8
'''
在很多文件下面只读取txt类型的文件，然后提取相关的数据

'''
import os
path = "./data" #当前路径,可以使绝对路径
files = os.listdir(path) #列出path目录下面的所有文件名，并且返回一个list

for file in files:
        file_ext = file.split('.')[1] #使用split分割文件名和后缀名 分割成一个list['xxxx','txt']，获取后缀
        ext = ['txt'] #期望读取的文件后缀
        if file_ext not in ext: #如果后缀名不在ext里面，则跳过本次循环，开始下次循环
                continue
        else:
                f = open(path + "/" + file,'r') 
                for s in f.readlines()[0:1]: #读取第一行的内容，可以实际情况来使用啥方法
                    print s,



f.close()

以上代码希望能给大家提供点思路，当然大家可以把结果写入到txt中保存起来。