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

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

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

这真的是一件小事,但是让我不得不说一说,谁让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男的一人企业,“专业实用化,永远不做大”是我的宗旨。

相关推荐

点过的网页会变色?没错,这玩意把你的浏览记录漏光了

提起隐私泄露这事儿,托尼其实早就麻了。。。平时网购、换手机号、注册各种账号之类的都会咔咔泄露,根本就防不住。但托尼真是没想到,浏览器里会有一个看起来完全人畜无害的功能,也在偷偷泄露我们的个人隐私,而且...

Axure教程:高保真数据可视化原型

本文将介绍如何制作Axure高保真数据可视化原型,供大家参考和学习。高保真数据可视化原型设计,称得上是Axure高阶水平。数据可视化在原型设计中是一个重要的分支,但是对于Axure使用者具有一定要求。...

Flutter web开发中禁用浏览器后退按钮

路由采用的go-router路由框架:finalrootNavigatorKey=GlobalKey<NavigatorState>();finalGoRouterrouter...

jQuery 控制属性和样式

标记的属性each()遍历元素:each(callback)方法主要用于对选择器进行遍历,它接受一个函数为参数,该函数接受一个参数,指代元素的序号。对于标记的属性而言,可以利用each()方法配合th...

微信小程序入门教程之二:页面样式

这个系列的上一篇教程,教大家写了一个最简单的Helloworld微信小程序。但是,那只是一个裸页面,并不好看。今天接着往下讲,如何为这个页面添加样式,使它看上去更美观,教大家写出实际可以使用的页...

如何在Windows11的任务栏中禁用和删除天气小部件图标?

微软该公司已在Windows11的任务栏中添加了一个天气小部件图标,作为小部件的入口点。这个功能与之前Win10上的新闻与资讯功能相同,但是有的用户不喜欢想要关闭,不知道如何操作,下面小编为大家带来...

CSS伪类选择器大全:提升网页交互与样式的神奇工具

CSS伪类选择器是前端开发中不可或缺的强大工具,它们允许我们根据元素的状态、位置或用户行为动态地应用样式。本文将全面介绍常用的伪类选择器,并通过代码示例展示其实际应用场景。一、基础交互伪类1.超链接...

7个Axure使用小技巧

编辑导读:对于Axure原型工具,很少有产品经过系统学习,一般都是直接上手,边摸索边学习,这直接导致很多快捷操作被忽视。笔者在日常工作中总结出以下小技巧,希望对各位有帮助。之前整理了2期Axure的...

JavaScript黑暗技巧:禁止浏览器点击“后退”按钮

浏览网页时,当从A页面点击跳转到B页面后,一般情况下,可以点击浏览器上的“后退”按钮返回A页面。如果进入B页面后,B页面想让访问者留下,禁止返回,是否可以实现呢?这简直是要控制浏览器的行为,虽然有些邪...

对齐PyTorch,一文详解OneFlow的DataLoader实现

撰文|赵露阳在最新的OneFlowv0.5.0版本中,我们增加了许多新特性,比如:新增动态图特性:OneFlow默认以动态图模式(eager)运行,与静态图模式(graph)相比,更容易搭建网...

Python计算机视觉编程 第一章 基本的图像操作和处理

以下是使用Python进行基本图像操作和处理的示例代码:使用PIL库加载图像:fromPILimportImageimage=Image.open("image.jpg"...

PyTorch 深度学习实战(31):可解释性AI与特征可视化

在上一篇文章中,我们探讨了模型压缩与量化部署技术。本文将深入可解释性AI与特征可视化领域,揭示深度学习模型的决策机制,帮助开发者理解和解释模型的内部工作原理。一、可解释性AI基础1.核心概念特征重要...

学习编程第177天 python编程 富文本框text控件的使用

今天学习的是刘金玉老师零基础Python教程第72期,主要内容是python编程富文本框text控件。一、知识点1.tag_config方法:利用某个别名作为标签,具体的对应标签的属性功能配置在后面参...

用Python讓電腦攝像頭實現掃二維碼

importsys#系統模組,用來存取命令列參數與系統功能importcv2#OpenCV,處理影像與相機操作importnumpyasnp#Numpy,用來處理數值與...

使用Transformer来做物体检测

作者:JacobBriones编译:ronghuaiyang导读这是一个Facebook的目标检测Transformer(DETR)的完整指南。介绍DEtectionTRansformer(D...