正则表达式查找：Python编程快速上手-让繁琐工作自动化练习题

题目

编写一个程序，打开文件夹中所有的.txt 文件，查找匹配用户提供的正则表达 式的所有行。结果应该打印到屏幕上。

代码

import re
import os

path='yourpath'

# 用户输入输入内容，放入一个compile对象
regex = input("Enter your regex: ")
regex = re.compile(regex)

#listdir获取路径下的文件列表，判断以.txt结尾
for file in os.listdir(path):
    if file.endswith('.txt'):
    	#打开文件，读取一行，正则判断是否包含关键字
        with open(path+'\\'+file,'r') as f:
            filelines = f.readlines()
            for fileline in filelines:
                if regex.search(fileline):
                    print(file+'---->'+fileline)
                    print('\n')