Source code for SALib.plotting.bar


__all__ = ['plot']

# magic string indicating DF columns holding conf bound values
CONF_COLUMN = '_conf'


[docs]def plot(Si_df, ax=None): """Create bar chart of results. Parameters ---------- * Si_df: pd.DataFrame, of sensitivity results Returns ---------- * ax : matplotlib axes object Examples ---------- >>> from SALib.plotting.bar import plot as barplot >>> from SALib.test_functions import Ishigami >>> >>> # See README for example problem specification >>> >>> X = saltelli.sample(problem, 512) >>> Y = Ishigami.evaluate(X) >>> Si = sobol.analyze(problem, Y, print_to_console=False) >>> total, first, second = Si.to_df() >>> barplot(total) """ conf_cols = Si_df.columns.str.contains(CONF_COLUMN) confs = Si_df.loc[:, conf_cols] confs.columns = [c.replace(CONF_COLUMN, '') for c in confs.columns] Sis = Si_df.loc[:, ~conf_cols] ax = Sis.plot(kind='bar', yerr=confs, ax=ax) return ax