利用PyPDF2模块快速拆分PDF文档
利用PyPDF2模块快速拆分PDF文档,
自动查找py文件所在的目录里的pdf文件,并将拆分后的文件放到当前目录中。
from PyPDF2 import PdfFileReader, PdfFileWriter import os pdf_dir=[] def get_filepath(): docunames = os.listdir() for docuname in docunames: if os.path.splitext(docuname)[1] == '.pdf': pdf_dir.append(docuname) def current_path(): current_path = os.path.abspath(__file__) return os.path.abspath(os.path.dirname(current_path) + os.path.sep + ".") # 将PDF文件每页分割为一个单独pdf的文件,并pdf保存至当前目录中 def pdf_split_1(pdf_input, path_output): fname = os.path.splitext(os.path.basename(pdf_input))[0] # 获取pdf文件名,去掉后缀名 pdf = PdfFileReader(pdf_input) for page in range(pdf.getNumPages()): pdf_writer = PdfFileWriter() pdf_writer.addPage(pdf.getPage(page)) output_filename = path_output + r'\{}-{}.pdf'.format(fname, page+1) with open(output_filename, 'wb') as out: pdf_writer.write(out) # print('生成文件:{}'.format(output_filename)) if __name__ == "__main__": get_filepath() pdffile = os.path.join(current_path(),"".join(pdf_dir)) pdf_split_1(pdffile, current_path())