如何在Seaborn factorplot中为每两组条形图添加间距

我使用Seaborn执行以下代码绘制了条形图：

ax = sns.factorplot(x="feat", y="acc", col="roi", hue="alpha", data=df_d_pt, kind="bar", dodge=True)

当前图表中，每个feat类别下不同alpha对应的条形图是均匀分散的。我希望实现：深蓝与浅蓝的条形图紧密相邻，深蓝与浅绿的条形图之间保留一定间距。

补充数据样式：数据集包含feat（特征类别）、acc（准确率）、roi（子图分类）、alpha（分组变量，对应不同颜色，其中两个alpha值属于一组，另一个属于另一组）。

解决方案 #

核心思路 #

Seaborn默认的dodge=True会让同一x类别下的hue条形均匀分配间距，无法区分子组。因此需要手动调整条形位置：先关闭自动避让，再根据自定义的子组规则，将同组条形平移靠近，异组条形之间留出间距。

具体实现代码 #

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

# 1. 绘制基础图表，关闭自动避让
g = sns.catplot(x="feat", y="acc", col="roi", hue="alpha", data=df_d_pt, kind="bar", dodge=False)

# 2. 定义alpha分组规则：将需要紧密相邻的alpha归为同一组
# 请根据你的实际alpha值修改此字典
alpha_groups = {'0.1': 0, '0.2': 0, '0.3': 1}
total_subgroups = len(set(alpha_groups.values()))
# 获取同一子组内的条形数量
subgroup_bar_count = sum(1 for v in alpha_groups.values() if v == 0)
# 单个条形宽度
bar_width = 0.2
# 子组之间的间距（可按需调整）
inter_subgroup_gap = 0.1

# 3. 遍历每个子图，调整条形位置
for ax in g.axes.flat:
    # 获取所有条形对象
    all_bars = ax.patches
    # 获取每个x轴刻度的中心位置
    xtick_positions = np.array([tick.get_position()[0] for tick in ax.get_xticklabels()])
    # 获取hue对应的alpha值顺序
    alpha_labels = [text.get_text() for text in g._legend.texts]
    
    # 处理每个feat类别下的条形
    for idx, xtick_center in enumerate(xtick_positions):
        # 提取当前feat下的所有条形
        feat_bars = all_bars[idx*len(alpha_labels) : (idx+1)*len(alpha_labels)]
        # 计算每个子组的中心位置
        subgroup_centers = np.linspace(
            xtick_center - (total_subgroups-1)*(subgroup_bar_count*bar_width + inter_subgroup_gap)/2,
            xtick_center + (total_subgroups-1)*(subgroup_bar_count*bar_width + inter_subgroup_gap)/2,
            total_subgroups
        )
        
        bar_index = 0
        # 调整每个子组内的条形位置
        for group_num in range(total_subgroups):
            # 获取当前组对应的alpha值
            group_alphas = [a for a in alpha_labels if alpha_groups[a] == group_num]
            # 计算子组内每个条形的位置
            intra_group_positions = np.linspace(
                subgroup_centers[group_num] - (subgroup_bar_count-1)*bar_width/2,
                subgroup_centers[group_num] + (subgroup_bar_count-1)*bar_width/2,
                subgroup_bar_count
            )
            # 设置条形的x位置
            for pos in intra_group_positions:
                feat_bars[bar_index].set_x(pos)
                bar_index += 1
    
    # 调整x轴范围，避免条形被截断
    x_min = xtick_positions.min() - (subgroup_bar_count*bar_width + inter_subgroup_gap)
    x_max = xtick_positions.max() + (subgroup_bar_count*bar_width + inter_subgroup_gap)
    ax.set_xlim(x_min, x_max)
    # 保持x轴刻度在原位置
    ax.set_xticks(xtick_positions)

plt.show()

关键说明 #

• 先设置dodge=False让同一feat下的条形初始重叠，便于手动调整位置
• alpha_groups需根据你的实际alpha取值修改，将需要紧密相邻的类别归为同一组
• 可通过调整bar_width和inter_subgroup_gap参数，控制条形宽度和子组间距
• 代码兼容多子图（col="roi"）的场景，会自动遍历所有子图进行调整

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