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("文本文件行列表：\n",txtLines) #輸出文本行列表

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

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

輸出結果如下：

感謝你能夠認真閱讀完這篇文章，希望小編分享的“Python中如何使用正則表達式實現查找功能”這篇文章對大家有幫助，同時也希望大家多多支持億速云，關注億速云行業資訊頻道，更多相關知識等著你來學習!