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

一件小事:程序员用AI写代码,处理PDF页码截取并提纯

itomcoil 2025-05-24 14:42 18 浏览

这真的是一件小事,但是让我不得不说一说,谁让WPS气着我这个程序员了呢?!

我买了个自助打印设备的会员,每个月有免费打印60张的额度,过期作废。这个月剩余30多张,马上月底了,我想打印一些书来看。

我有一本PDF格式的电子书,500多页,300多M。文件太大了,没法直接传递给打印机。于是我想从这500页中选择性地拆出30页,生成一个新PDF小文件去打印。结果,WPS要收费。

这个WPS用的时候要登录,登录完了还要会员。我作为程序员,很了解PDF的文件组成结构,这拆分PDF没嘛技术含量。

PDF的全称是Portable Document Format,也就是便携式文档格式。它是一种由Adobe公司开发的用于表示文档的文件格式。PDF文件由多个部分组成,每个部分都有其特定的作用和结构。

比如Body部分,是它的主体内容。这里面包含实际的文档内容,例如文本、图形、图像、表格等。PDF中的页面内容以对象的形式存储,每个对象包括其在文件中的位置和长度。

还有Pages页面部分,这里面通常包含多个页面,每一页都可以包含文本、图像和其他元素。每个页面在PDF文件中被视为一个独立的对象,其内容和格式由页面对象描述。

另外Fonts字体、Images图像等信息,都可以解析并读取到。

我打算自己写一个,而且现在更方便了,咱们用AI生成代码。

AI给出的代码如下:

import PyPDF2

def extract_pages(input_pdf, output_pdf, start_page, end_page):
    # 打开输入PDF文件
    with open(input_pdf, "rb") as input_file:
        reader = PyPDF2.PdfReader(input_file)
        writer = PyPDF2.PdfWriter()
        
        # 获取总页数
        total_pages = len(reader.pages)
        
        # 确保页码范围有效
        if start_page < 1 or end_page > total_pages or start_page > end_page:
            raise ValueError("Invalid page range")
        
        # 将指定范围的页添加到新的PDF文件中
        for i in range(start_page - 1, end_page):
            writer.add_page(reader.pages[i])
        
        # 保存新的PDF文件
        with open(output_pdf, "wb") as output_file:
            writer.write(output_file)

# 示例用法
input_pdf = "t.pdf"  # 输入PDF文件路径
output_pdf = "output.pdf"  # 输出PDF文件路径
start_page = 10  # 开始页码(从1开始)
end_page = 45  # 结束页码(包含)

extract_pages(input_pdf, output_pdf, start_page, end_page)

将代码拷贝到编译器中,并把要处理的pdf文件也放到同级目录,点击执行:

这里是处理t.pdf文件,从第12页开始,一直到第45页(包含这一页),然后将这30多页在同级目录保存为output.pdf

去了注释,不到30行代码,一问一答,一复制一运行,3分钟内完成。

终于,新生成的小文件变为30来页,而且大小为20M,变得轻量便捷。

但是,我又发现一个问题。其实这个电子书的制作者,额外对书本添加了广告信息,看上图底部的页脚位置内容就知道。我不想要这些信息,看起来烦。其实,书里正常的内容是红色虚线部分,这里是原书内容的扫描件。

于是,我想对内容进行提纯。我将这个想法告诉了AI。

AI又给我一段代码:

import fitz  # PyMuPDF
import os

def extract_images(pdf_path, output_folder):
    # 打开PDF文件
    pdf_document = fitz.open(pdf_path)
    
    # 确保输出文件夹存在
    os.makedirs(output_folder, exist_ok=True)
    
    # 遍历每一页
    for page_num in range(len(pdf_document)):
        page = pdf_document.load_page(page_num)
        images = page.get_images(full=True)
        
        # 遍历每一页中的所有图片
        for img_index, img in enumerate(images):
            xref = img[0]
            base_image = pdf_document.extract_image(xref)
            image_bytes = base_image["image"]
            image_ext = base_image["ext"]
            image_filename = f"page{page_num+1}_img{img_index+1}.{image_ext}"
            image_path = os.path.join(output_folder, image_filename)
            
            # 保存图片
            with open(image_path, "wb") as image_file:
                image_file.write(image_bytes)
    
    print(f"提取的图片保存在文件夹: {output_folder}")

# 示例用法
pdf_path = "output.pdf"  # 输入PDF文件路径
output_folder = "extracted_images"  # 输出图片文件夹

extract_images(pdf_path, output_folder)

同第一次页码截取一样,复制代码,点击运行。它执行的操作是将output.pdf文件里的图片提取出来,然后保存在同级的extracted_images下。

图片提取出来,而且还是高清画质,原来在PDF里面是被压缩了反而模糊。

现在是图片,怎么办?我想要的是PDF。没关系,去问AI。

又是一串代码:

from PIL import Image
import fitz  # PyMuPDF
import os

def images_to_pdf(image_folder, output_pdf):
    # 获取指定文件夹中的所有图片文件
    image_files = [f for f in os.listdir(image_folder) if f.lower().endswith(('jpeg'))]
    # 按照文件名排序
    image_files = sorted(image_files)
    # 用于临时保存单页PDF文件的列表
    pdf_pages = []
    # 定义A4纸的尺寸(210 x 297 mm)以像素表示(假设300 DPI)
    a4_width_mm = 210
    a4_height_mm = 297
    dpi = 300
    a4_width_px = int(a4_width_mm * dpi / 25.4)
    a4_height_px = int(a4_height_mm * dpi / 25.4)
    a4_size = (a4_width_px, a4_height_px)

    for image_file in image_files:
        image_path = os.path.join(image_folder, image_file)
        image = Image.open(image_path)

        # 将图片调整为A4尺寸
        image = image.resize(a4_size, Image.LANCZOS)
        # 将图片转换为RGB模式,如果不是RGB模式
        if image.mode != 'RGB':
            image = image.convert('RGB')
        # 将图片保存为单页PDF
        pdf_path = os.path.join(image_folder, f"{os.path.splitext(image_file)[0]}.pdf")
        image.save(pdf_path, "PDF", resolution=dpi)
        pdf_pages.append(pdf_path)

    # 创建一个空的PDF文档
    pdf_document = fitz.open()
    # 将所有单页PDF合并到一个PDF文档中
    for pdf_page in pdf_pages:
        temp_pdf = fitz.open(pdf_page)
        pdf_document.insert_pdf(temp_pdf)
        temp_pdf.close()
    # 保存合并后的PDF文档
    pdf_document.save(output_pdf)
    pdf_document.close()
    # 删除临时保存的单页PDF文件
    for pdf_page in pdf_pages:
        os.remove(pdf_page)
# 示例用法
image_folder = "extracted_images"  # 存放图片的文件夹
output_pdf = "combined.pdf"  # 输出PDF文件路径
images_to_pdf(image_folder, output_pdf)

OK,获取到了一个纯净无广告而且超清晰版本的小PDF。

从开始有想法,到问AI要代码并处理完文档,甚至到写完这篇教程文章。总共花费不到2个小时。而这效果,却比花钱买会员要强得多。

我之所说这件事,想表达两点。

第一,信息差太重要了。这么简单的操作,真的有厂商拿来收钱。我们不管哪个行业,在花钱之前,先问问业内的人有没有必要花。我从没买过软件类的会员,有需求一般都是自己做。

第二,AI的应用越来越普及,其实不只是我可以这么操作。这种问AI要代码,然后放到编译器点击运行的操作,一旦熟悉了,其实和编辑word差不多。现在,很多人都可以像这样,自己去解决生活中遇到的问题。

我是IT男的一人企业,“专业实用化,永远不做大”是我的宗旨。

相关推荐

最强聚类模型,层次聚类 !!_层次聚类的优缺点

哈喽,我是小白~咱们今天聊聊层次聚类,这种聚类方法在后面的使用,也是非常频繁的~首先,聚类很好理解,聚类(Clustering)就是把一堆“东西”自动分组。这些“东西”可以是人、...

python决策树用于分类和回归问题实际应用案例

决策树(DecisionTrees)通过树状结构进行决策,在每个节点上根据特征进行分支。用于分类和回归问题。实际应用案例:预测一个顾客是否会流失。决策树是一种基于树状结构的机器学习算法,用于解决分类...

Python教程(四十五):推荐系统-个性化推荐算法

今日目标o理解推荐系统的基本概念和类型o掌握协同过滤算法(用户和物品)o学会基于内容的推荐方法o了解矩阵分解和深度学习推荐o掌握推荐系统评估和优化技术推荐系统概述推荐系统是信息过滤系统,用于...

简单学Python——NumPy库7——排序和去重

NumPy数组排序主要用sort方法,sort方法只能将数值按升充排列(可以用[::-1]的切片方式实现降序排序),并且不改变原数组。例如:importnumpyasnpa=np.array(...

PyTorch实战:TorchVision目标检测模型微调完

PyTorch实战:TorchVision目标检测模型微调完整教程一、什么是微调(Finetuning)?微调(Finetuning)是指在已经预训练好的模型基础上,使用自己的数据对模型进行进一步训练...

C4.5算法解释_简述c4.5算法的基本思想

C4.5算法是ID3算法的改进版,它在特征选择上采用了信息增益比来解决ID3算法对取值较多的特征有偏好的问题。C4.5算法也是一种用于决策树构建的算法,它同样基于信息熵的概念。C4.5算法的步骤如下:...

Python中的数据聚类及可视化分析实践

探索如何通过聚类分析揭露糖尿病预测数据集的特征!我们将运用Python的强力工具,深入挖掘数据,以直观的可视化揭示不同特征间的关系。一同探索聚类分析在糖尿病预测中的实践!所有这些可视化都可以通过数据操...

用Python来统计大乐透号码的概率分布

用Python来统计大乐透号码的概率分布,可以按照以下步骤进行:导入所需的库:使用Python中的numpy库生成数字序列,使用matplotlib库生成概率分布图。读取大乐透历史数据:从网络上找到大...

python:支持向量机监督学习算法用于二分类和多分类问题示例

监督学习-支持向量机(SVM)支持向量机(SupportVectorMachine,简称SVM)是一种常用的监督学习算法,用于解决分类和回归问题。SVM的目标是找到一个最优的超平面,将不同类别的...

25个例子学会Pandas Groupby 操作

groupby是Pandas在数据分析中最常用的函数之一。它用于根据给定列中的不同值对数据点(即行)进行分组,分组后的数据可以计算生成组的聚合值。如果我们有一个包含汽车品牌和价格信息的数据集,那么可以...

数据挖掘流程_数据挖掘流程主要有哪些步骤

数据挖掘流程1.了解需求,确认目标说一下几点思考方法:做什么?目的是什么?目标是什么?为什么要做?有什么价值和意义?如何去做?完整解决方案是什么?2.获取数据pandas读取数据pd.read.c...

使用Python寻找图像最常见的颜色_python 以图找图

如果我们知道图像或对象最常见的是哪种颜色,那么可以解决图像处理中的几个用例,例如在农业领域,我们可能需要确定水果的成熟度。我们可以简单地检查一下水果的颜色是否在预定的范围内,看看它是成熟的,腐烂的,还...

财务预算分析全网最佳实践:从每月分析到每天分析

原文链接如下:「链接」掌握本文的方法,你就掌握了企业预算精细化分析的能力,全网首发。数据模拟稍微有点问题,不要在意数据细节,先看下最终效果。在编制财务预算或业务预算的过程中,通常预算的所有数据都是按月...

常用数据工具去重方法_数据去重公式

在数据处理中,去除重复数据是确保数据质量和分析准确性的关键步骤。特别是在处理多列数据时,保留唯一值组合能够有效清理数据集,避免冗余信息对分析结果的干扰。不同的工具和编程语言提供了多种方法来实现多列去重...

Python教程(四十):PyTorch深度学习-动态计算图

今日目标o理解PyTorch的基本概念和动态计算图o掌握PyTorch张量操作和自动求导o学会构建神经网络模型o了解PyTorch的高级特性o掌握模型训练和部署PyTorch概述PyTorc...