Python編程快速上手——正則表達(dá)式查找功能案例分析

更新時(shí)間：2020年02月28日 12:43:28 作者：授我以驢

這篇文章主要介紹了Python正則表達(dá)式查找功能,結(jié)合具體實(shí)例形式分析了Python基于正則表達(dá)式遍歷查找指定格式文件的相關(guān)操作技巧,需要的朋友可以參考下

本文實(shí)例講述了Python正則表達(dá)式查找功能。分享給大家供大家參考，具體如下：

題目如下：

編寫一個(gè)程序，打開文件夾中所有的.txt文件，查找匹配用戶提供的正則表達(dá)式的所有行。結(jié)果應(yīng)該打印到屏幕上。

思路如下:

程序需要做的事情如下：
遍歷文件夾得到所有.txt文件名
打開所有.txt文件，正則表達(dá)式進(jìn)行模式匹配
查找結(jié)果顯示到屏幕
代碼需要做的事情如下：
導(dǎo)入re,os模塊
定義正則表達(dá)式函數(shù)
函數(shù)內(nèi)進(jìn)行正則表達(dá)式匹配，并返回匹配所在行列表
for調(diào)用os.listdir(path)，生成.txt文件名列表
for循環(huán)打開所有.txt文件
用戶輸入需要查找的字符串
for循環(huán)遍歷函數(shù)返回結(jié)果
輸出結(jié)果到屏幕

代碼如下：

import os, re

def fileRex(inputStr,txtLines):
  txtRex = re.compile(r'{0}'.format(inputStr)) #正則表達(dá)式對(duì)象
  blockList = []
  i = 0
  for t in txtLines:
    try:
      mo1 = txtRex.search(t)
      mo1.group()
      #search()方法匹配成功生成match對(duì)象，group()返回匹配到的對(duì)象，匹配成功即表示這一行為需要查找結(jié)果，添加到列表
      i += 1 #計(jì)數(shù)
      blockList.append(t)
    except:
      continue
      #search()方法未匹配成功會(huì)報(bào)錯(cuò)，程序崩潰，因此需要except加上continue，進(jìn)入下一次循環(huán)
  print("匹配到的模式個(gè)數(shù)：",i)
  return blockList


nameList = [] #創(chuàng)建文件名列表

file = input("請(qǐng)輸入文件夾絕對(duì)路徑(如：C:\\Users\\Administrator\\Desktop\\exam):") #\\轉(zhuǎn)義字符相當(dāng)于 \
for fileName in os.listdir(file): #os.listdir()方法遍歷文件夾
  p = 0
  if fileName.endswith('.txt'): #enswith方法，判斷字符串結(jié)尾，檢查文件名后綴是不是.txt文本文件
    nameList.append(fileName) #添加到文件名列表
    p += 1
  else:
    continue
print("找到的文本文件個(gè)數(shù)： {0} ！文件有：{1}".format(p,nameList))
txtLines = [] #創(chuàng)建存放文本行的列表
for i in nameList:
  txtFile = open('{0}\\{1}'.format(file,i)) #open函數(shù)創(chuàng)建file對(duì)象
  txtLines += txtFile.readlines() #readlines返回列表，+ 號(hào)連接返回的列表
print("文本文件行列表：\n",txtLines) #輸出文本行列表

inputStr = input("請(qǐng)輸入要查找的字符串：")

for k in fileRex(inputStr,txtLines): #調(diào)用函數(shù)，循環(huán)遍歷返回的列表
  k = k.strip('\n') #去掉列表字符串中的空格
  print("含有字符 “{0}” 的文本行有：{1}".format(inputStr,k)) #輸出查找結(jié)果