使用python

Question

是否可以将新页面插入多页pdf文件的任意位置？

在这个虚拟示例中，我正在创建一些pdf页面：

from matplotlib.backends.backend_pdf import PdfPages
import matplotlib.pyplot as plt

with PdfPages('dummy.pdf') as pdf:
    for i in range(5):
        plt.plot(1,1)
        pdf.savefig()
        plt.close()

现在我想绘制其他内容并将新图保存为pdf的第1页。

Answer 1

您可以使用模块PyPDF2。它使您可以合并或操作pdf文件。我尝试了一个简单的例子，创建了另一个只有1页的pdf文件，并在第一个文件的中间添加了这个页面。然后我将所有内容写入新的输出文件：

from matplotlib.backends.backend_pdf import PdfPages
import matplotlib.pyplot as plt
from PyPDF2 import PdfFileWriter, PdfFileReader


with PdfPages('dummy.pdf') as pdf:
    for i in range(5):
        plt.plot(1,1)
        pdf.savefig()
        plt.close()

#Create another pdf file
with PdfPages('dummy2.pdf') as pdf:
    plt.plot(range(10))
    pdf.savefig()
    plt.close()


infile = PdfFileReader('dummy.pdf', 'rb')
infile2 = PdfFileReader('dummy2.pdf', 'rb')
output = PdfFileWriter()

p2 = infile2.getPage(0)

for i in xrange(infile.getNumPages()):
    p = infile.getPage(i)
    output.addPage(p)
    if i == 3:
        output.addPage(p2)

with open('newfile.pdf', 'wb') as f:
   output.write(f)

也许有一种更聪明的方法可以做到这一点，但我希望这有助于开始。