如何将平均线和中线添加到海运地块

问题描述

有没有办法将平均值和中位数与Seborn的displot相加？

penguins = sns.load_dataset("penguins")
g = sns.displot(
    data=penguins, x='body_mass_g',
    col='species',  
    facet_kws=dict(sharey=False, sharex=False)
)

基于Add mean and variability to seaborn FacetGrid distplots，我看到我可以定义FacetGrid并映射函数。我可以将自定义函数传递给displot吗？

尝试直接使用displot的原因是绘图开箱即用，无需调整刻度标签大小、轴标签大小等，并且在视觉上与我正在绘制的其他绘图一致。

def specs(x, **kwargs):
    ax = sns.histplot(x=x)
    ax.axvline(x.mean(), color='k', lw=2)
    ax.axvline(x.median(), color='k', ls='--', lw=2)

g = sns.FacetGrid(data=penguins, col='species')
g.map(specs,'body_mass_g' )

---

解决方案

• 不推荐直接使用FacetGrid。相反，可以使用其他图形级方法，如seaborn.displot
  • seaborn.FacetGrid.map使用图形级方法。
  • seaborn: Building structured multi-plot grids
• 测试于python 3.8.11、pandas 1.3.2、matplotlib 3.4.3、seaborn 0.11.2

选项%1

• 使用plt.而不是ax。
  • 在OP中，vlines要为histplot创建ax，但这里的图形是在.map之前创建的。

penguins = sns.load_dataset("penguins")
g = sns.displot(
    data=penguins, x='body_mass_g',
    col='species',  
    facet_kws=dict(sharey=False, sharex=False)
)

def specs(x, **kwargs):
    plt.axvline(x.mean(), c='k', ls='-', lw=2.5)
    plt.axvline(x.median(), c='orange', ls='--', lw=2.5)

g.map(specs,'body_mass_g' )

选项2

• 此选项更详细，但更灵活，因为它允许从用于创建displot的数据源以外的数据源访问和添加信息。

import seaborn as sns
import pandas as pd

# load the data
pen = sns.load_dataset("penguins")

# groupby to get mean and median
pen_g = pen.groupby('species').body_mass_g.agg(['mean', 'median'])

g = sns.displot(
    data=pen, x='body_mass_g',
    col='species',  
    facet_kws=dict(sharey=False, sharex=False)
)
# extract and flatten the axes from the figure
axes = g.axes.flatten()

# iterate through each axes
for ax in axes:
    # extract the species name
    spec = ax.get_title().split(' = ')[1]
    
    # select the data for the species
    data = pen_g.loc[spec, :]
    
    # print data as needed or comment out
    print(data)
    
    # plot the lines
    ax.axvline(x=data['mean'], c='k', ls='-', lw=2.5)
    ax.axvline(x=data['median'], c='orange', ls='--', lw=2.5)

两个选项的输出

资源

• 另请参阅以下问题/答案，以了解将信息添加到海运FaceGrid的其他方式
  • Draw a line at specific position/annotate a Facetgrid in seaborn
  • Overlay a vertical line on seaborn scatterplot with multiple subplots
  • How to add additional plots to a seaborn FacetGrid and specify colors