如何为Seaborn catplot的每个子图设置不同调色板

问题背景 #

需要生成两个分类条形图，核心要求：

• 两个图仅x、hue、col参数互换
• 第二个图中，每个Surface子图（C/D/E/B）需使用第一个图对应Surface的色系
• 保留Seaborn自动计算误差棒的功能

原尝试代码出现了调色板警告、图例缺失、条形重叠等问题，以下是解决方案：

完整实现代码 #

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

# 模拟数据集（替换为你的真实数据）
data = pd.DataFrame({
    "config": [1,2,3]*20,
    "Surface": ["A","B","C","D","E"]*12,
    "Height": [2.8,7.8,12.8,22.8]*15,
    "Value": [10,12,15,8,9,11,14,7,10,13]*6
})

# ---------------------- 第一个图表 ----------------------
fig1 = sns.catplot(kind="bar", data=data, 
                   y="Value", x="config", hue="Surface", col="Height", col_wrap=2)
plt.subplots_adjust(right=0.75)

# 提取第一个图中每个Surface对应的主色
surface_color_map = {}
for patch in fig1.axes[0].patches:
    label = patch.get_label()
    if label not in surface_color_map:
        surface_color_map[label] = patch.get_facecolor()

# ---------------------- 第二个图表 ----------------------
filtered_data = data[data["Surface"] != "A"]
config_count = len(filtered_data["config"].unique())

# 为每个Surface生成对应主色的浅色调色板（匹配config的类别数量）
surface_palettes = {
    surf: sns.light_palette(surface_color_map[surf], n_colors=config_count)
    for surf in filtered_data["Surface"].unique()
}

# 自定义绘图函数：为每个子图分配专属调色板
def plot_colored_bars(x, y, hue, data, **kwargs):
    current_surface = data["Surface"].iloc[0]
    sns.barplot(x=x, y=y, hue=hue, data=data, palette=surface_palettes[current_surface], **kwargs)

# 创建FacetGrid并映射绘图函数
fig2 = sns.FacetGrid(filtered_data, col="Surface", col_order=["C", "D", "E", "B"], col_wrap=2)
fig2.map_dataframe(plot_colored_bars, x="Height", y="Value", hue="config")

# 调整图表样式
for ax, title in zip(fig2.axes, ["C", "D", "E", "B"]):
    ax.set_title(title)
fig2.set_all_xlabels("Height")
fig2.set_xticklabels([2.8,7.8,12.8,22.8])
fig2.add_legend(title="Config")
plt.subplots_adjust(right=0.74)

plt.show()

关键解决思路 #

1. 提取第一个图的颜色映射：从第一个图的条形补丁中获取每个Surface对应的主色，确保两个图的色系完全对应
2. 生成子图专属调色板：使用sns.light_palette基于主色生成多阶浅色调，匹配config的类别数量，保证同一子图内不同config的条形有区分度
3. 用FacetGrid灵活控制子图：通过map_dataframe为每个子图分配专属调色板，既避免了直接使用catplot时的调色板冲突，又保留了hue参数自动计算误差棒的功能
4. 修复图例问题：通过add_legend手动添加图例，确保config的类别标签正确显示

原错误原因解析 #

• 未正确关联hue与palette参数，触发Seaborn的版本兼容警告
• 直接混用catplot的全局调色板与子图专属调色板，导致图例标签冲突、条形重叠
• 未手动添加图例，导致config的类别说明缺失

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