这篇文章主要介绍“Python怎么使用正则表达式实现查找功能”,在日常操作中,相信很多人在Python怎么使用正则表达式实现查找功能问题上存在疑惑,小编查阅了各式资料,整理出简单好用的操作方法,希望对大家解答”Python怎么使用正则表达式实现查找功能”的疑惑有所帮助!接下来,请跟着小编一起来学习吧!
题目如下:
编写一个程序,打开文件夹中所有的.txt文件,查找匹配用户提供的正则表达式的所有行。结果应该打印到屏幕上。
思路如下:
程序需要做的事情如下:
遍历文件夹得到所有.txt文件名
打开所有.txt文件,正则表达式进行模式匹配
查找结果显示到屏幕代码需要做的事情如下:
导入re,os模块
定义正则表达式函数
函数内进行正则表达式匹配,并返回匹配所在行列表
for调用os.listdir(path),生成.txt文件名列表
for循环打开所有.txt文件
用户输入需要查找的字符串
for循环遍历函数返回结果
输出结果到屏幕
代码如下:
import os, re def fileRex(inputStr,txtLines): txtRex = re.compile(r'{0}'.format(inputStr)) #正则表达式对象 blockList = [] i = 0 for t in txtLines: try: mo1 = txtRex.search(t) mo1.group() #search()方法匹配成功生成match对象,group()返回匹配到的对象,匹配成功即表示这一行为需要查找结果,添加到列表 i += 1 #计数 blockList.append(t) except: continue #search()方法未匹配成功会报错,程序崩溃,因此需要except加上continue,进入下一次循环 print("匹配到的模式个数:",i) return blockList nameList = [] #创建文件名列表 file = input("请输入文件夹绝对路径(如:C:\Users\Administrator\Desktop\exam):") #\转义字符相当于 for fileName in os.listdir(file): #os.listdir()方法遍历文件夹 p = 0 if fileName.endswith('.txt'): #enswith方法,判断字符串结尾,检查文件名后缀是不是.txt文本文件 nameList.append(fileName) #添加到文件名列表 p += 1 else: continue print("找到的文本文件个数: {0} !文件有:{1}".format(p,nameList)) txtLines = [] #创建存放文本行的列表 for i in nameList: txtFile = open('{0}\{1}'.format(file,i)) #open函数创建file对象 txtLines += txtFile.readlines() #readlines返回列表,+ 号连接返回的列表 print("文本文件行列表: ",txtLines) #输出文本行列表 inputStr = input("请输入要查找的字符串:") for k in fileRex(inputStr,txtLines): #调用函数,循环遍历返回的列表 k = k.strip(' ') #去掉列表字符串中的空格 print("含有字符 “{0}” 的文本行有:{1}".format(inputStr,k)) #输出查找结果
输出结果如下: