python – 仅选择PDF pypdf2的第一页

2019年7月28日 379次阅读

我试图剥离多个PDF文件的第一页并合并为一个文件. (我每天收到150个PDF文件,第一页是我需要的发票,以下三到12页只是我不需要的备份)所以输入的是150个不同大小的PDF文件,我想要的输出是1个PDF文件,仅包含150个文件中每个文件的第一页.

我似乎已经做的是合并所有页面,除了第一页(这是我唯一需要的页面).

import PyPDF2, os

pdfFiles = []
for filename in os.listdir('.'):
    if filename.endswith('.pdf'):
        pdfFiles.append(filename)
pdfFiles.sort(key=str.lower)
pdfWriter = PyPDF2.PdfFileWriter()

for filename in pdfFiles:
    pdfFileObj = open(filename, 'rb')
    pdfReader = PyPDF2.PdfFileReader(pdfFileObj)

for pageNum in range(1 , pdfReader.numPages):
    pageObj = pdfReader.getPage(pageNum)
    pdfWriter.addPage(pageObj)


pdfOutput = open('CombinedFirstPages.pdf', 'wb')
pdfWriter.write(pdfOutput)
pdfOutput.close()

最佳答案 试试这个：

import PyPDF2, os


your_target_folder = ""

pdf_files = []

for dirpath, _, filenames in os.walk(your_target_folder):

    for items in filenames:

        file_full_path = os.path.abspath(os.path.join(dirpath, items))

        if file_full_path.lower().endswith(".pdf"):
            pdf_files.append(file_full_path)

        else:
            pass


pdf_files.sort(key=str.lower)
pdfWriter = PyPDF2.PdfFileWriter()


for files_address in pdf_files:
    pdfFileObj = open(files_address, 'rb')
    pdfReader = PyPDF2.PdfFileReader(pdfFileObj)

    pageObj = pdfReader.getPage(0)
    pdfWriter.addPage(pageObj)


with open("CombinedFirstPages.pdf", "wb") as output:
    pdfWriter.write(output)

祝好运 ..

点赞