在使用python 编写自动化脚本时需要对特定数据进行处理,那么如何在某文件夹下查找特定数据呢?
代码如下:
import os
path = os.getcwd()
files = os.listdir(path)
print files
for file in files:
logfile = open(file,"r")
lines = logfile.readlines()
for line in lines:
if line.find("111") != -1:
print "111 exist in file:" + file
elif line.find("222") != -1:
print "222 exist in file:" + file
运行结果:
可以看到,test.py 脚本中的数据也被读取了,这对于数据分析来说显然是不合理的,那么就需要对需处理的文件进行判断,即仅检查txt文件:
import os
path = os.getcwd()
files = os.listdir(path)
print files
for file in files:
if file.split(".")[1] == "txt":
logfile = open(file,"r")
lines = logfile.readlines()
for line in lines:
if line.find("111") != -1:
print "111 exist in file:" + file
elif line.find("222") != -1:
print "222 exist in file:" + file