用python解决一个文件的搜索与收集问题

Question

先搜索一个文件夹内的所有txt文件，然后在其中找出符合要求的字符串，最后把字符串存储到另一个文件中。

Answer 1

import os, re
#1 列出当前目录下的所有以txt结尾的文件
path = '.'
txtfiles = [item for item in os.listdir(path) if item.endswith('.txt')]
#2 用正则表达式查找文件
pattern = 'abc$' #假定这里是找以abc结尾的行
prog = re.compile(pattern) #编译正则表达式
outp = open('outfile.txt', 'w') # 输出文件名outfile.txt
for filep in txtfiles:
for line in open(filep, 'r').xreadlines()
if prog.match( line.strip() ) != None: # 如果匹配规则，就写入指定的文件
outp.write(line)