如何实现Seaborn生成图表自动调整尺寸以解决内容重叠问题

核心实现逻辑为根据当前绘制列的分类数量动态计算画布尺寸，无需手动为不同列添加单独判断条件，可自动适配所有分类数量的场景。 #

具体实现步骤 #

• 提前计算当前列的唯一分类数，按「单个分类分配固定宽度」的规则计算画布总宽度，分类数量越多画布自动加宽
• 所有绘图方法统一传入动态计算的画布尺寸参数
• 配合plt.tight_layout()自动调整图表留白，避免标签被截断

适配后的完整代码 #

for col in dff.drop(target_col_name, axis=1).columns:
    n_unique = dff[col].nunique()
    # 动态计算画布宽度，每个分类分配0.8英寸宽度，系数可根据实际显示效果调整
    fig_width = max(6, n_unique * 0.8)
    fig_size = (fig_width, 6)
    if ((n_unique / len(dff[col])) < threshold):
        # 计数图传入动态尺寸
        plt.figure(figsize=fig_size)
        ax = sns.countplot(x=dff[col], hue= dff[target_col_name] )
        ax.set_xticklabels(ax.get_xticklabels(), rotation = 90)
        plt.tight_layout()
        plt.show()
        # 堆叠比例条形图传入动态尺寸
        pd.crosstab(
            index = dff[col], 
            columns = dff[target_col_name], 
            normalize = 'index'
        ).plot.bar(figsize=fig_size)
        plt.tight_layout()
        plt.show()
    elif (dff[col].dtype == 'int64' or dff[col].dtype == 'float64'):
        # 箱线图传入动态尺寸
        plt.figure(figsize=fig_size)
        sns.boxplot(x=dff[target_col_name], y=dff[col])
        plt.tight_layout()
        plt.show()

效果对比 #

正常显示效果：

未适配尺寸的重叠效果：
#

额外优化提示 #

• 动态宽度的系数可根据实际需求调整，若分类名称普遍较长可将0.8上调为1甚至更高，若分类名称较短可下调压缩画布宽度
• 若需要保存图片，可在plt.savefig()中添加bbox_inches='tight'参数，避免边缘标签被截断
• 若分类数量极多，可额外设置x轴标签横向排列，进一步压缩画布高度

内容的提问来源于stack exchange，提问作者M.Zubair Akram