本文实例讲述了Python正则表达式查找功能。分享给大家供大家参考,具体如下:
题目如下:
- 编写一个程序,打开文件夹中所有的.txt文件,查找匹配用户提供的正则表达式的所有行。结果应该打印到屏幕上。
思路如下:
- 程序需要做的事情如下: 遍历文件夹得到所有.txt文件名 打开所有.txt文件,正则表达式进行模式匹配 查找结果显示到屏幕
- 代码需要做的事情如下: 导入re,os模块 定义正则表达式函数 函数内进行正则表达式匹配,并返回匹配所在行列表 for调用os.listdir(path),生成.txt文件名列表 for循环打开所有.txt文件 用户输入需要查找的字符串 for循环遍历函数返回结果 输出结果到屏幕
代码如下:
代码语言:javascript复制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:UsersAdministratorDesktopexam):") #转义字符相当于
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("文本文件行列表:n",txtLines) #输出文本行列表
inputStr = input("请输入要查找的字符串:")
for k in fileRex(inputStr,txtLines): #调用函数,循环遍历返回的列表
k = k.strip('n') #去掉列表字符串中的空格
print("含有字符 “{0}” 的文本行有:{1}".format(inputStr,k)) #输出查找结果
输出结果如下:
PS:这里再为大家提供2款非常方便的正则表达式工具供大家参考使用:
JavaScript正则表达式在线测试工具: http://tools.zalou.cn/regex/javascript
正则表达式在线生成工具: http://tools.zalou.cn/regex/create_reg
更多关于Python相关内容可查看本站专题:《Python正则表达式用法总结》、《Python数据结构与算法教程》、《Python函数使用技巧总结》、《Python字符串操作技巧汇总》、《Python入门与进阶经典教程》及《Python文件与目录操作技巧汇总》
希望本文所述对大家Python程序设计有所帮助。