Seaborn子图x轴顺序异常及自定义图例添加求助

一、x轴排序问题修复 #

核心原因 #

Seaborn的barplot和pointplot默认会对分类类型的x轴自动排序，即便你的DataFrame本身顺序正确，只要x轴列是普通字符串类型，就会被按字母/数字逻辑重排。

可行解决方法 #

• 方法一：将x轴列转为有序分类类型
  提取DataFrame中x轴的原始顺序，把x轴列设置为有序分类，从根源上固定排序逻辑：

  import pandas as pd
  
  # 假设x轴分组列名为'bin'，提取两个数据集的原始顺序
  valid_x_order = df['graph_df_uni_valid']['bin'].tolist()
  oot_x_order = df['graph_df_uni_oot']['bin'].tolist()
  
  # 转为有序分类
  df['graph_df_uni_valid']['bin'] = pd.Categorical(
      df['graph_df_uni_valid']['bin'],
      categories=valid_x_order,
      ordered=True
  )
  df['graph_df_uni_oot']['bin'] = pd.Categorical(
      df['graph_df_uni_oot']['bin'],
      categories=oot_x_order,
      ordered=True
  )
  

  后续绘图无需额外设置order参数，x轴会严格遵循原始顺序。

• 方法二：绘图时精准指定order参数
  若不想修改数据类型，确保order参数传入的是x轴列的实际取值列表，而非DataFrame索引：

  # Test图示例
  sns.barplot(
      data=df['graph_df_uni_valid'],
      x='bin',
      y='volume',
      order=valid_x_order
  )
  sns.pointplot(
      data=df['graph_df_uni_valid'],
      x='bin',
      y='book_rate',
      color='red',
      order=valid_x_order
  )
  

二、自定义图例设置 #

由于barplot和pointplot的默认图例无法直接匹配需求，需要手动创建图例元素：

import matplotlib.pyplot as plt
import seaborn as sns
import matplotlib.patches as mpatches

# 创建1行2列的子图
fig, (ax_test, ax_oot) = plt.subplots(1, 2, figsize=(14, 6))

# 绘制Test组图表
sns.barplot(
    data=df['graph_df_uni_valid'],
    x='bin',
    y='volume',
    ax=ax_test,
    color='blue'
)
sns.pointplot(
    data=df['graph_df_uni_valid'],
    x='bin',
    y='book_rate',
    ax=ax_test,
    color='red',
    linestyle='-'
)
sns.pointplot(
    data=df['graph_df_uni_valid'],
    x='bin',
    y='score',
    ax=ax_test,
    color='green',
    linestyle='-'
)

# 绘制OOT组图表
sns.barplot(
    data=df['graph_df_uni_oot'],
    x='bin',
    y='volume',
    ax=ax_oot,
    color='blue'
)
sns.pointplot(
    data=df['graph_df_uni_oot'],
    x='bin',
    y='book_rate',
    ax=ax_oot,
    color='red',
    linestyle='-'
)
sns.pointplot(
    data=df['graph_df_uni_oot'],
    x='bin',
    y='score',
    ax=ax_oot,
    color='green',
    linestyle='-'
)

# 创建自定义图例元素
bar_legend = mpatches.Patch(color='blue', label='Volume')
red_line_legend = mpatches.Patch(color='red', label='Book rate')
green_line_legend = mpatches.Patch(color='green', label='Score')

# 添加图例到画布顶部中央
fig.legend(
    handles=[bar_legend, red_line_legend, green_line_legend],
    loc='upper center',
    bbox_to_anchor=(0.5, 1.08),
    ncol=3,
    fontsize=12
)

# 调整布局避免图例遮挡
plt.tight_layout()
plt.show()

版本适配说明 #

针对你使用的matplotlib 3.4.0和seaborn 0.10.0，上述代码完全兼容，无需额外调整参数。

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