python - Seaborn 因子绘制自定义错误栏而不是自举

Question

我想在 seaborn 中绘制因子图，但手动提供误差线而不是让 seaborn 计算它们。

我有一个大致如下所示的 pandas 数据框:

     model output feature  mean   std
0    first    two       a  9.00  2.00
1    first    one       b  0.00  0.00
2    first    one       c  0.00  0.00
3    first    two       d  0.60  0.05
...
77   third   four       a  0.30  0.02
78   third   four       b  0.30  0.02
79   third   four       c  0.10  0.01

我正在输出一个看起来像这样的图:

我正在使用这个 seaborn 命令来生成绘图:

g = sns.factorplot(data=pltdf, x='feature', y='mean', kind='bar',
                   col='output', col_wrap=2, sharey=False, hue='model')
g.set_xticklabels(rotation=90)

但是，我不知道如何让 seaborn 使用“std”列作为错误栏。不幸的是，重新计算相关数据框的输出将非常耗时。

这和这个问题有点相似: Plotting errors bars from dataframe using Seaborn FacetGrid

除了我不知道如何让它与 matplotlib.pyplot.bar 函数一起工作。

有没有办法使用 seaborn factorplot 或 FacetGrid 结合 matplotlib 来做到这一点？

Answer 1

在 python 3.8.12、pandas 1.3.4、matplotlib 3.4.3、seaborn 0.11 中测试。 2

你可以做类似的事情

import seaborn as sns
import matplotlib.pyplot as plt

tips = sns.load_dataset("tips")

tip_sumstats = (tips.groupby(["day", "sex", "smoker"])
                     .total_bill
                     .agg(["mean", 'sem'])
                     .reset_index())

def errplot(x, y, yerr, **kwargs):
    ax = plt.gca()
    data = kwargs.pop("data")
    data.plot(x=x, y=y, yerr=yerr, kind="bar", ax=ax, **kwargs)

g = sns.FacetGrid(tip_sumstats, col="sex", row="smoker")
g.map_dataframe(errplot, "day", "mean", "sem")