Seaborn factor plot custom error bars

2019-01-14 09:16发布

问题:

I'd like to plot a factorplot in seaborn but manually provide the error bars instead of having seaborn calculate them.

I have a pandas dataframe that looks roughly like this:

     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

and I'm outputting a plot that looks roughly like this:

I'm using this seaborn commands to generate the plot:

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)

However, I can't figure out how to have seaborn use the 'std' column as the error bars. Unfortunately, it would be quite time consuming to recompute the output for the data frame in question.

This is a little similar to this q: Plotting errors bars from dataframe using Seaborn FacetGrid

Except I can't figure out how to get it to work with the matplotlib.pyplot.bar function.

Is there a way to do this using seaborn factorplot or FacetGrid combined with matplotlib?

Thanks!

回答1:

You could do something like

import seaborn as sns
import matplotlib.pyplot as plt
from scipy.stats import sem
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")