Python matplotlib : Save to pdf in multiple pages

Question

我正在尝试执行以下操作:

我使用 matplotlib 创建了一个图形，其中包含几个子图。更具体地说，2x4 子图

输出非常适合在屏幕上显示，但不适合将其保存为 pdf。

如果我只使用 save_fig，它会打印一个单页 pdf 文档，带有 2x4 网格。

我想做的是重新安排我的子图，比方说一个 2x4 网格(选择哪个子图去哪里，这很好，但不是必需的)并将其打印成包含 4 个子图的 2 页 pdf每个。 (为了能够适应A4页面大小)

这可能吗？

先谢谢你!

Answer 1

因为我的工作需要类似的东西，所以我努力根据显示媒介将绘图分组到图形中的过程自动化。起初我的想法是每个图只做一次，然后将子图添加到要保存在 pdf 中的图形中，但遗憾的是，根据 this answer 中的评论，这是不可能的，所以一切都需要重新绘制。该代码显示了如何使用 PdfPages 实现自动化的一般思路:

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


def niter(iterable, n):
    """
    Function that returns an n-element iterator, i.e.
    sub-lists of a list that are max. n elements long.
    """
    pos = 0
    while pos < len(iterable):
        yield iterable[pos:pos+n]
        pos += n


def plot_funcs(x, functions, funcnames, max_col, max_row):
    """
    Function that plots all given functions over the given x-range,
    max_col*max_row at a time, creating all needed figures while doing
    so.
    """

    ##amount of functions  to put in one plot    
    N = max_col*max_row

    ##created figures go here
    figs = []

    ##plotted-on axes go here
    used_axes = []

    ##looping through functions N at a time:
    for funcs, names in zip(niter(functions, N), niter(funcnames,N)):

        ##figure and subplots
        fig, axes = plt.subplots(max_col, max_row)

        ##plotting functions
        for name,func,ax in zip(names, funcs, axes.reshape(-1)):
            ax.plot(x, func(x))
            ax.set_title(name)
            used_axes.append(ax)

        ##removing empty axes:
        for ax in axes.reshape(-1):
            if ax not in used_axes:
                ax.remove()

        fig.tight_layout()
        figs.append(fig)

    return figs

##some functions to display
functions = [
    lambda x: x, lambda x: 1-x, lambda x: x*x, lambda x: 1/x, #4
    np.exp, np.sqrt, np.log, np.sin, np.cos,                  #5
    ]
funcnames = ['x','1-x', 'x$^2$', '1/x', 'exp', 'sqrt', 'log', 'sin','cos']

##layout for display on the screen
disp_max_col = 3
disp_max_row = 2

##layout for pdf
pdf_max_col = 2
pdf_max_row = 4

##displaying on the screen:
x = np.linspace(0,1,100)
figs = plot_funcs(x, functions, funcnames, disp_max_row, disp_max_col)
plt.show()


##saving to pdf if user wants to:
answer = input('Do you want to save the figures to pdf?')
if answer in ('y', 'Y', 'yes', ''):

    ##change number of subplots
    N = disp_max_col*disp_max_row
    figs = plot_funcs(x, functions, funcnames, pdf_max_row, pdf_max_col)

    ##from https://matplotlib.org/examples/pylab_examples/multipage_pdf.html
    with PdfPages('multipage_pdf.pdf') as pdf:
        for fig in figs:
            plt.figure(fig.number)
            pdf.savefig()

核心函数 plot_funcs 采用 max_col 和 max_row 关键字，然后创建具有相应子图数量的图形。然后循环遍历给定的要绘制的函数列表，每个函数都在其自己的子图中。未使用的子图被删除。最后返回所有图形的列表。

在我的示例中，我有 9 个不同的函数，我首先以 2x3 布局在屏幕上显示它们(总共制作两个图形，一个有 6 个子图，一个有 3 个子图)。如果用户满意，绘图将以 2x4 布局重做(同样是两幅图，但这次一幅有 8 个子图，一幅有 1 个子图)，然后保存到名为 multipage_pdf.pdf 的文件中，以下 example in the documentation .

在 python 3.5 上测试