如何按Pandas DataFrame列dtype分类生成图表并导出为每页2张的PDF

以下是可直接运行的函数，完全匹配需求：

import pandas as pd
import seaborn as sns
import matplotlib.pyplot as plt
from matplotlib.backends.backend_pdf import PdfPages

def data_explorer(data):
    # 筛选需要绘制的列
    obj_cols = [col for col in data.columns if data[col].dtype == 'object']
    float_cols = [col for col in data.columns if data[col].dtype == 'float64']
    all_plot_cols = obj_cols + float_cols  # 共8张图，顺序为countplot在前、boxplot在后
    
    with PdfPages('data_exploration.pdf') as pdf:
        # 每2张图为1页循环处理
        for i in range(0, len(all_plot_cols), 2):
            # 创建1行2列的子图布局
            fig, axes = plt.subplots(nrows=1, ncols=2, figsize=(12, 5))
            # 绘制第一个子图
            col1 = all_plot_cols[i]
            if col1 in obj_cols:
                sns.countplot(x=col1, data=data, ax=axes[0])
                axes[0].set_title(f'{col1} 计数分布', fontsize=12)
            else:
                sns.boxplot(y=data[col1], ax=axes[0])
                axes[0].set_title(f'{col1} 数值分布箱线图', fontsize=12)
            # 绘制第二个子图
            col2 = all_plot_cols[i+1]
            if col2 in obj_cols:
                sns.countplot(x=col2, data=data, ax=axes[1])
                axes[1].set_title(f'{col2} 计数分布', fontsize=12)
            else:
                sns.boxplot(y=data[col2], ax=axes[1])
                axes[1].set_title(f'{col2} 数值分布箱线图', fontsize=12)
            # 自动调整布局避免标签重叠
            plt.tight_layout()
            # 写入当前页到PDF
            pdf.savefig(fig)
            # 关闭画布释放内存
            plt.close(fig)
    print("PDF文件已生成：data_exploration.pdf")

# 调用示例
# data_explorer(df)

功能说明 #

• 自动跳过UID等int类型的列，无需额外判断
• 生成的PDF共4页，每页2张图：前2页是4张分类列countplot，后2页是4张数值列boxplot
• 自动添加图表标题，布局适配不会出现内容重叠
• 用PdfPages管理多页输出，不会出现文件覆盖问题

内容的提问来源于stack exchange，提问作者nasa313