快捷導(dǎo)航

拿來就用！Python批量合并PDF的示例代碼

更新時(shí)間：2020年08月10日 11:47:47 作者：劉早起

這篇文章主要介紹了Python批量合并PDF的示例代碼，幫助大家更好的理解和學(xué)習(xí)Python，感興趣的朋友可以了解下

大家好，今天分享一個(gè)實(shí)用的辦公腳本：將多個(gè)PDF合并為一個(gè)PDF，例如我手上現(xiàn)在有如下3個(gè)PDF分冊(cè)，需要整合成一個(gè)完整的PDF

如果換成你操作的話，是不是打開百度搜索：PDF合并，然后去第三方網(wǎng)站操作，可能會(huì)收費(fèi)不說還擔(dān)心文件泄漏，現(xiàn)在有請(qǐng)Python出場(chǎng)，簡(jiǎn)單快速，光速合并，拿走就用！

首先導(dǎo)入需要的庫和路徑設(shè)置

import os
from PyPDF2 import PdfFileReader, PdfFileWriter
if __name__ == '__main__':
  # 設(shè)置存放多個(gè)pdf文件的文件夾
  dir_path = r'C:\Scientific Research\Knowladge\Ophthalmology\Chinese Ophthalmology' 
  # 目標(biāo)文件的名字
  file_name = "中華眼科學(xué)（第3版）合并版.pdf"

接著獲取所有pdf文件的絕對(duì)路徑，這里需要利用os庫中的os.walk遍歷文件和os.path.join拼接路徑

for dirpath, dirs, files in os.walk(dir_path):
  print(dirpath)
  print(files)
# 結(jié)果返回當(dāng)前路徑、當(dāng)前路徑下文件夾，并以列表返回所有文件

建議直接將需要合并的pdf放在一個(gè)文件夾，這樣就無需再對(duì)文件后綴進(jìn)行判斷，包裝成函數(shù)后如下：

def GetFileName(dir_path):
  file_list = [os.path.join(dirpath, filesname) \
         for dirpath, dirs, files in os.walk(dir_path) \
         for filesname in files]
  return file_list

調(diào)用該函數(shù)的結(jié)果

現(xiàn)在建立合并PDF的函數(shù)

def MergePDF(dir_path, file_name):
  # 實(shí)例化寫入對(duì)象
  output = PdfFileWriter()
  outputPages = 0
  # 調(diào)用上一個(gè)函數(shù)獲取全部文件的絕對(duì)路徑
  file_list = GetFileName(dir_path)

  for pdf_file in file_list:
    print("文件：%s" % pdf_file.split('\\')[-1], end=' ')

    # 讀取PDF文件
    input = PdfFileReader(open(pdf_file, "rb"))
    # 獲得源PDF文件中頁面總數(shù)
    pageCount = input.getNumPages()
    outputPages += pageCount
    print("頁數(shù)：%d" % pageCount)
    # 分別將page添加到輸出output中
    for iPage in range(pageCount):
      output.addPage(input.getPage(iPage))
  print("\n合并后的總頁數(shù):%d" % outputPages)
  # 寫入到目標(biāo)PDF文件
  print("PDF文件正在合并，請(qǐng)稍等......")
  with open(os.path.join(dir_path, file_name), "wb") as outputfile:
    # 注意這里的寫法和正常的上下文文件寫入是相反的
    output.write(outputfile)
  print("PDF文件合并完成")

可以看到雖然待合并的PDF文件比較大，但是依舊快速的合并成功！

最后附上完整代碼，只需將代碼中PDF的路徑和文件名修改即可使用！

import os
from PyPDF2 import PdfFileReader, PdfFileWriter

def GetFileName(dir_path):
  file_list = [os.path.join(dirpath, filesname) \
         for dirpath, dirs, files in os.walk(dir_path) \
         for filesname in files]
  return file_list

def MergePDF(dir_path, file_name):
  output = PdfFileWriter()
  outputPages = 0
  file_list = GetFileName(dir_path)
  for pdf_file in file_list:
    print("文件：%s" % pdf_file.split('\\')[-1], end=' ')
    # 讀取PDF文件
    input = PdfFileReader(open(pdf_file, "rb"))
    # 獲得源PDF文件中頁面總數(shù)
    pageCount = input.getNumPages()
    outputPages += pageCount
    print("頁數(shù)：%d" % pageCount)
    # 分別將page添加到輸出output中
    for iPage in range(pageCount):
      output.addPage(input.getPage(iPage))
  print("\n合并后的總頁數(shù):%d" % outputPages)
  # 寫入到目標(biāo)PDF文件
  print("PDF文件正在合并，請(qǐng)稍等......")
  with open(os.path.join(dir_path, file_name), "wb") as outputfile:
    # 注意這里的寫法和正常的上下文文件寫入是相反的
    output.write(outputfile)
  print("PDF文件合并完成")

if __name__ == '__main__':
  # 設(shè)置存放多個(gè)pdf文件的文件夾
  dir_path = r'C:\Scientific Research\Knowladge\Ophthalmology\Chinese Ophthalmology'
  # 目標(biāo)文件的名字
  file_name = "中華眼科學(xué)（第3版）合并版.pdf"
  MergePDF(dir_path, file_name)

以上就是拿來就用！Python批量合并PDF的示例代碼的詳細(xì)內(nèi)容，更多關(guān)于Python合并PDF的資料請(qǐng)關(guān)注腳本之家其它相關(guān)文章！

您可能感興趣的文章: