plotnine中geom_boxplot自定义中位数线段颜色：如何计算x/xend与箱宽

不需要依赖类似gg_build的功能，直接手动计算x轴位置和箱宽即可，步骤如下：

• 获取分组的x轴映射位置：plotnine会把字符串类型的分组变量自动映射为连续的整数x坐标（第一个分组对应1，第二个对应2，以此类推）。你可以通过把分组列转为pandas.Categorical来获取这个映射关系：

  import pandas as pd
  
  # 假设你的数据框是df，分组列是'group'
  df['group'] = pd.Categorical(df['group'])
  # 获取每个分组对应的x中心位置（从1开始）
  group_x_positions = {group: i+1 for i, group in enumerate(df['group'].cat.categories)}
  

• 确定箱宽：geom_boxplot的默认箱宽是0.8，如果你在绘图时手动设置了width参数（比如geom_boxplot(width=0.6)），就用你设置的值；没设置的话直接用默认的0.8即可。

• 构建中位数线段的数据集：
  先计算每个分组的中位数，再结合上面的x位置和箱宽计算x、xend：

  import numpy as np
  
  # 计算每个分组的中位数
  median_df = df.groupby('group')['value'].median().reset_index(name='median')
  # 添加x中心位置
  median_df['x_center'] = median_df['group'].map(group_x_positions)
  # 设置箱宽（这里用默认的0.8，可根据实际调整）
  box_width = 0.8
  # 计算x和xend
  median_df['x'] = median_df['x_center'] - box_width/2
  median_df['xend'] = median_df['x_center'] + box_width/2
  # y和yend都是中位数
  median_df['y'] = median_df['median']
  median_df['yend'] = median_df['median']
  

• 绘图时叠加中位数线段：
  先画普通箱线图，再用geom_segment叠加自定义颜色的中位数线段，记得把线段的颜色设置为你想要的样式，同时可以调大线条宽度让它更明显：

  from plotnine import ggplot, aes, geom_boxplot, geom_segment, theme_minimal
  
  (ggplot(df, aes(x='group', y='value'))
   + geom_boxplot(width=box_width)  # 这里的width要和上面计算的一致
   + geom_segment(data=median_df, 
                  aes(x='x', xend='xend', y='y', yend='yend'),
                  color='red', size=1.5)  # 自定义颜色和粗细
   + theme_minimal()
  )
  

• 注意事项：

  • 如果你的分组是自定义顺序的，一定要确保pd.Categorical的categories顺序和plotnine绘图时的顺序一致，避免x位置错位。
  • 如果后续调整了箱线图的width参数，要同步修改box_width变量的值，保证线段和箱线宽度匹配。

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