百度360必应搜狗淘宝本站头条
当前位置:网站首页 > 技术资源 > 正文

三步教你用Elasticsearch+PyMuPDF实现PDF大文件秒搜!

off999 2025-06-30 19:19 4 浏览 0 评论

面对100页以上的大型PDF文件时,阅读和搜索往往效率低下。传统关系型数据库在处理此类数据时容易遇到性能瓶颈,而 Elasticsearch 凭借其强大的全文检索和分布式架构,成为理想解决方案。通过 PyMuPDF 提取PDF文本内容,结合Elasticsearch的倒排索引技术,能够快速定位和检索信息,即使面对海量数据也能保持高效查询。其分布式特性支持多节点扩展,进一步提升性能和可靠性,完美解决传统数据库的性能瓶颈问题。

以下是详细的开发方案:

1. 系统设计

  • 功能需求

(1)支持上传 PDF 文件,并从中提取文本。
(2)支持分词操作以及全文检索。
(3)支持将匹配内容以高亮形式显示。
(4)支持分页展示搜索结果

  • 技术栈
    • PyMuPDF:用于解析PDF文件。
    • jieba:用于中文分词。
    • Elasticsearch:用于存储和检索文本数据。
    • Flask:用于构建Web界面和API。

2. 实现步骤

2.1 安装依赖

Python

pip install pymupdf flask jieba elasticsearch

2.2 启动Elasticsearch

(1)下载并安装Elasticsearch

(2)从 Elasticsearch官网(
https://www.elastic.co/cn/elasticsearch) 下载并解压

(3)运行 bin/elasticsearch 启动服务

(4)确保Elasticsearch运行在 http://localhost:9200

2.3 解析PDF并存入Elasticsearch

使用PyMuPDF提取PDF文本,并将文本数据存入Elasticsearch

Python

import fitz  # PyMuPDF
from elasticsearch import Elasticsearch
import jieba

# 连接Elasticsearch
es = Elasticsearch("http://localhost:9200")

def extract_and_store_pdf(filepath):
    # 提取文件名
    filename = filepath.split('/')[-1]

    # 解析PDF
    doc = fitz.open(filepath)
    for page_num in range(len(doc)):
        page = doc.load_page(page_num)
        text = page.get_text()

        # 对文本进行分词
        tokens = jieba.lcut(text)

        # 存入Elasticsearch
        doc_data = {
            "filename": filename,
            "page": page_num + 1,
            "text": text,
            "tokens": tokens
        }
        es.index(index="pdf_documents", body=doc_data)

# 示例:解析并存储PDF
extract_and_store_pdf('example.pdf')

2.4 实现分词搜索

使用Elasticsearch的全文检索功能实现分词搜索:

Python

def search_keyword(keyword):
    # 对关键词进行分词
    keyword_tokens = jieba.lcut(keyword)

    # 构建Elasticsearch查询
    query = {
        "query": {
            "match": {
                "tokens": " ".join(keyword_tokens)  # 将分词结果组合成查询字符串
            }
        },
        "highlight": {
            "fields": {
                "text": {}  # 高亮显示匹配内容
            }
        }
    }

    # 执行查询
    response = es.search(index="pdf_documents", body=query)
    return response['hits']['hits']

# 示例:搜索关键词
results = search_keyword('Python编程')
for hit in results:
    print(f"文件: {hit['_source']['filename']}, 页码: {hit['_source']['page']}")
    print(f"内容: {hit['highlight']['text'][0]}")

2.5 构建Web界面

使用Flask构建一个简单的Web界面,支持文件上传和搜索:

Python

from flask import Flask, render_template, request, redirect, url_for
import os

app = Flask(__name__)
UPLOAD_FOLDER = 'uploads'
app.config['UPLOAD_FOLDER'] = UPLOAD_FOLDER

# 首页:上传文件
@app.route('/', methods=['GET', 'POST'])
def index():
    if request.method == 'POST':
        file = request.files['file']
        if file:
            filepath = os.path.join(app.config['UPLOAD_FOLDER'], file.filename)
            file.save(filepath)
            extract_and_store_pdf(filepath)
            return redirect(url_for('index'))
    return render_template('index.html')

# 搜索页面
@app.route('/search', methods=['GET'])
def search():
    keyword = request.args.get('keyword')
    if keyword:
        results = search_keyword(keyword)
        return render_template('search.html', results=results, keyword=keyword)
    return render_template('search.html')

if __name__ == '__main__':
    if not os.path.exists(UPLOAD_FOLDER):
        os.makedirs(UPLOAD_FOLDER)
    app.run(debug=True)

2.6 创建HTML模板

在 templates 目录下创建以下HTML文件:

index.html(文件上传页面):

html



search.html(搜索页面):

html



3. 运行系统

(1) 启动Elasticsearch:

bash

bin/elasticsearch

(2) 启动Flask应用:

bash

python app.py

(3) 访问 http://127.0.0.1:5000,上传PDF文件。

(4) 访问
http://127.0.0.1:5000/search,输入关键词搜索。

4. 功能扩展

  • 分页显示:在Elasticsearch查询中添加 from 和 size 参数,实现分页。
  • 停用词过滤:在分词时过滤常见停用词(如“的”、“是”)。
  • 词性标注:结合词性标注(如名词、动词)提升搜索精度。
  • 性能优化:使用Elasticsearch的分布式特性,支持大规模数据检索。
  • 语义化搜索:通过理解用户意图和上下文,返回相关结果,而非简单关键词匹配。

通过以上步骤,你可以构建一个高性能的PDF文件搜索系统!

创作不易,您的建议,是我们进步的动力!欢迎留言及加关注,帮助我们做得更好!

相关推荐

python爬取电子课本,送给居家上课的孩子们

在这个全民抗疫的日子,中小学生们也开启了居家上网课的生活。很多没借到书的孩子,不得不在网上看电子课本,有的电子课本是老师发的网络链接,每次打开网页去看,既费流量,也不方便。今天我们就利用python的...

高效办公!Python 批量生成PDF文档是如何做到的?

前言:日常办公中,经常会使用PDF文档,难免需要对PDF文档进行编辑,有时候PDF文档中的大部分内容都是一样的,只是发送对象不同。这种模板套用的场景下,使用Python进行自动化就尤为方便,用最短的时...

如何用Python将PDF完整的转成Word?

PDF文件完整的转为Word,转换后格式排版不会乱,图片等信息完整显示不丢失。这个很简单,有很多方法都可以实现。方法一:Python利用Python将PDF文件转换为Word,有许多库可以帮你实现这一...

使用Python拆分、合并PDF(python合并多个pdf)

知识点使用Python操作PDF!主要内容有:1、PDF拆分;2、PDF合并。在工作中,难免会和PDF打交道,所以掌握一点处理PDF的技能非常有必要,本文将介绍几个常用的功能。PDF拆分很多时候,获取...

10分钟实现PDF转Word神器!看DeepSeek如何用Python解放打工人

开篇痛点每个被PDF折磨过的职场人都懂——领导发来的扫描件要修改,手动抄到Word需要2小时;网上下载的报告想复制数据,却变成乱码…今天我们用Python+DeepSeek,10分钟打造一个智能转换工...

《Python知识手册》,高清全彩pdf版开放下载

Python编程还不懂?今天我要把我参与编写的这套《Python知识手册》免费分享出来,看完文末有惊喜哦。...

利用python进行数据分析,PDF文档给你答案

本书详细介绍利用Python进行操作、处理、清洗和规整数据等方面的具体细节和基本要点。虽然本书的标题是“数据分析”,重点却是Python编程、库,以及用于数据分析的工具。兄弟,毫无套路!PDF版无偿获...

OCRmypdf:一款可以让扫描PDF文件变得可搜索、可复制!

简介在日常工作中,我们经常会接触到各种PDF文件,其中不少是扫描版文档。处理这些扫描PDF时,尽管内容看似完整,但往往无法直接复制或搜索其中的文本。尤其是在需要对大量文档进行文本分析、存档或后期编辑时...

高效的OCR处理工具!让扫描PDF文件变得可搜索、可复制!

在工作中,我们常常遇到各种各样的PDF文件,其中不乏一些扫描版的文档。而在处理扫描的PDF文件时,虽然文件内容看似完整,但你却无法复制、搜索其中的文本。特别是对大量文档需要进行文本分析、存档、或者...

三步教你用Elasticsearch+PyMuPDF实现PDF大文件秒搜!

面对100页以上的大型PDF文件时,阅读和搜索往往效率低下。传统关系型数据库在处理此类数据时容易遇到性能瓶颈,而Elasticsearch凭借其强大的全文检索和分布式架构,成为理想解决方案。通过...

用 Python 去除 PDF 水印,你学会吗?

今天介绍下用Python去除PDF(图片)的水印。思路很简单,代码也很简洁。首先来考虑Python如何去除图片的水印,然后再将思路复用到PDF上面。这张图片是前几天整理《数据结构和算法...

扫描PDF档案效率提升300%!OCRmyPDF:告别无法搜索的PDF噩梦,这款26K Star的开源神器让文本识别轻松上手!

要在PDF中搜索某个关键词,结果发现啥也找不到?这种情况大多数人都遇到过吧,特别是处理扫描文档或图片PDF时。就在前几天,我还在为这事抓狂呢!后来无意中发现了OCRmyPDF这个宝藏项目...简直就...

Python自动化办公之PDF版本发票识别并提取关键信息教程(上篇)

大家好,我是皮皮。一、前言前几天在Python白银交流群【上海新年人】问了一个Python自动化办公发票数据处理的问题,一起来看看吧。二、实现过程这个问题在实际工作中还是非常常见的,实用性和通用性都比...

PDF解锁神器:用PyMuPDF与pdfplumber告别手动提取

前言大家好,今天咱们来聊聊如何用Python中的PyMuPDF和pdfplumber库,轻松提取PDF文件里的文本和元数据。你是否曾经在处理一个复杂的PDF文件时,感到信息难以触及,提取过程让人抓狂?...

《Python知识手册》,高清pdf免费获取

今天我要把我参与编写的这套《Python知识手册》免费分享出来,真正弘扬Python开源精神!手册的部分页面如下:获取方式:...

取消回复欢迎 发表评论: