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

Question

所以我试图将我在 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)

Answer 1

保存 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)