Python 绘制 PDF 多图

Question

我试图将多个图形绘制到一个 PDF 中，但它生成了损坏的 PDF。

import sys, os, pandas as pd, numpy as np, matplotlib.pyplot as plt
import pylab, scipy.stats as stats, calendar
from matplotlib.backends.backend_pdf import PdfPages
def plotLineGraph(df):
    fig = plt.figure()
    ax = fig.add_subplot(211)
    df.plot(ax=ax)
    ax = plt.subplot(111)
    return fig

pp = PdfPages(r'C:\myPLOT.pdf')
for ii in [1,2,3,4]:
    df = pd.DataFrame({'x': np.rand(500), 'y':np.rand(500)})
    pp.savefig(plotGraph(df))
pp.close()

我应该如何修改上述代码，以便在一个 PDF 文件中成功获取 4 个绘图。

Answer 1

您的代码中有几个拼写错误。

1. 该函数名为 plotLineGraph 而不是 plotGraph
2. numpy 没有 rand，而是 numpy.random.rand。
3. 您可以通过 ax = plt.subplot(111) 创建多余的坐标区。删除该行。

最后考虑一下，在某些Windows系统中不允许直接写入C:\。总共这将给出:

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

def plotLineGraph(df):
    fig = plt.figure()
    ax = fig.add_subplot(211)
    df.plot(ax=ax)
    return fig

pp = PdfPages(r'D:\myPLOT.pdf')
for ii in [1,2,3,4]:
    df = pd.DataFrame({'x': np.random.rand(500), 'y':np.random.rand(500)})
    pp.savefig(plotLineGraph(df))
pp.close()