Python提取多个pdf首页合并输出

pip install PyPDF2

import PyPDF2, os

# import sys
#
# if not sys.warnoptions:
#     import warnings
#     warnings.simplefilter("ignore")

#输入包含所有pdf文件的文件夹
dir = "c:/ceRNA"
#改变路径到该文件夹
os.chdir(dir)
#创建一个PdfFileWriter对象，后面用来保存提取的首页
pdfWriter = PyPDF2.PdfFileWriter()

#遍历文件夹中的每一个文件
for file in os.listdir(dir):
    #如果以.pdf为后缀
    if file.endswith(".pdf"):
        #打开这个pdf文件，以二进制的方式读取
        pdfFileObj = open(file, 'rb')
        #创建一个PdfFileReader对象，来读取pdf内容
        pdfReader = PyPDF2.PdfFileReader(pdfFileObj)
        #获取首页
        pageObj = pdfReader.getPage(0)
        #添加到pdfWriter中
        pdfWriter.addPage(pageObj)

#新建一个pdf文件，用来保存所有的首页，以二进制的方式来写
pdfOutput = open('all_first_page.pdf', 'wb')
#将pdfWriter的内容写到文件中
pdfWriter.write(pdfOutput)
#关闭输出文件
pdfOutput.close()

# import sys
#
# if not sys.warnoptions:
#     import warnings
#     warnings.simplefilter("ignore")