查找文件夹内文件中,含有特定词语的页

使用语言:python(不得不说代码写起来是方便了,可是跑起来效率十分感人。)

调用库:pdfFilePages对pdf文件进行处理,将pdf文件中的内容提取出来。然后使用正则表达式匹配文件中的汉字,输出存在该汉字的文件名和页码。

# -*- coding: utf-8 -*-
import pdfplumber
import re
import os

from pkg_resources import WorkingSet
def getFileName(filepath): #获取文件夹下面的所有文件
    file_list = []
    for root,dirs,files in os.walk(filepath):
        for filespath in files:
            file_list.append(os.path.join(root,filespath))
    return file_list
    

def getContent(pdfFile,pdfReader,page):
    # find = re.findall('前景(.*)', pdfFile.pages[page].extract_text())
    page = pdfReader.pages[page]   # 第page页的信息
    context = page.extract_text()
    pattern1 = re.compile(r'[\u4e00-\u9fa5]+')
    context1 = pattern1.findall(context)
    context2 = "".join(context1)
    pattern2 = re.compile(u'人民币')  #把人民币换成自己要匹配的字符串
    context3 = pattern2.findall(context2)
    if context3 != []:
        print(pdfFile,page)

def getResult(fileList):
    for fileName in fileList:
        pdfFile = open(fileName, 'rb')  # 读取pdf文件
        with pdfplumber.open(pdfFile) as pdfReader:
            for page in range(len(pdfReader.pages)):
                getContent(pdfFile,pdfReader,page)


if __name__ == '__main__':
    fileFolderName = "C:\\Users\\klsxs\\Desktop\\pdf测试文件" #把这个地方改成自己的文件夹的名字
    fileList = getFileName(fileFolderName)
    getResult(fileList)

使用库PyPDF2,过程是一样的,只不过识别pdf文件的准确率不太行。

# -*- coding: utf-8 -*-
import pdfplumber
import PyPDF2
import re
import os

from pkg_resources import WorkingSet
def getFileName(filepath): #获取文件夹下面的所有文件
    file_list = []
    for root,dirs,files in os.walk(filepath):
        for filespath in files:
            file_list.append(os.path.join(root,filespath))
    return file_list
    

def getContent(pdfFile,pdfReader,page):
    # find = re.findall('前景(.*)', pdfFile.pages[page].extract_text())
    page = pdfReader.pages[page]   # 第page页的信息
    context = page.extract_text()
    pattern1 = re.compile(r'[\u4e00-\u9fa5]+')
    context1 = pattern1.findall(context)
    context2 = "".join(context1)
    pattern2 = re.compile(u'人民币')  #把人民币换成自己要匹配的字符串
    context3 = pattern2.findall(context2)
    if context3 != []:
        print(pdfFile,page)

def getResult(fileList):
    for fileName in fileList:
        pdfFile = open(fileName, 'rb')  # 读取pdf文件
        pdfReader = PyPDF2.PdfFileReader(pdfFile)
        pdfFilePages = pdfReader.numPages #文件的页数
        with pdfplumber.open(pdfFile) as pdfReader:
            for page in range(pdfFilePages):
                getContent(pdfFile,pdfReader,page)


if __name__ == '__main__':
    fileFolderName = "C:\\Users\\klsxs\\Desktop\\pdf测试文件" #把这个地方改成自己的文件夹的名字
    fileList = getFileName(fileFolderName)
    getResult(fileList)

最终输出的结果如下所示:

 输出出现“人民币”一词的文件名和页码。

  • 3
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值