如何将多个Seaborn图保存到单个pdf文件中

时间:2018-06-10 11:21:03

标签: python plot seaborn

所以我试图将我在for循环中创建的多个图保存到单个pdf文件中。我已经搜索了SO并拼凑了一些似乎有用的代码,除了它没有保存它创建的pdf但没有任何内容的数字。

以下是重现它的代码:

import numpy as np
import matplotlib.pyplot as plt
import seaborn as sns
import pandas as pd

dftest = pd.DataFrame(np.random.randint(low=0, high=10, size=(5, 5)),
                    columns=['a', 'b', 'c', 'd', 'e']) 

from matplotlib.backends.backend_pdf import PdfPages

with PdfPages('count.pdf') as pdf_pages:
    df1 = dftest.select_dtypes([np.int, np.float, np.object])
    for i, col in enumerate(df1.columns):
        plt.figure(i)
        countplot = sns.countplot(x=col, data=df1)
        pdf_pages.savefig(countplot.fig)

1 个答案:

答案 0 :(得分:0)

保存plt.figure对我有用

with PdfPages('count.pdf') as pdf_pages:
    df1 = dftest.select_dtypes([np.int, np.float, np.object])
    for i, col in enumerate(df1.columns):
        figu = plt.figure(i)
        countplot = sns.countplot(x=col, data=df1)
        pdf_pages.savefig(figu)