使用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中保存起来。

最后编辑于
©著作权归作者所有,转载或内容合作请联系作者
平台声明:文章内容(如有图片或视频亦包括在内)由作者上传并发布,文章内容仅代表作者本人观点,简书系信息发布平台,仅提供信息存储服务。

推荐阅读更多精彩内容