python利用PyPDF2提取pdf部分頁面

參考:https://www.jianshu.com/p/17ff11efaba7

https://www.jianshu.com/p/82485e3e46e1

from PyPDF2 import PdfFileReader, PdfFileWriter

def extract_pdf(file, page_begin=1, page_end=5):
    pdf = PdfFileReader(file)
    writer = PdfFileWriter()

    pages_num = page_end - page_begin + 1

    for i in range(pages_num):
        writer.addPage(pdf.getPage(i + page_begin - 1))

    file_split = 'copy2'+file
    with open(file_split, 'wb') as file_out:
        writer.write(file_out)


if __name__ == '__main__':
    page_begin = 2   # 起始頁碼(包括)
    page_end = 5     # 結束頁碼(包括)
    file_name = 'PhaseDNNpdeFinal.pdf'  # 要拆分的文件名稱
    extract_pdf(file_name, page_begin, page_end)