Source code for SALib.util.results

import pandas as pd  # type: ignore
from SALib.plotting.bar import plot as barplot

[docs]class ResultDict(dict): '''Dictionary holding analysis results. Conversion methods (e.g. to Pandas DataFrames) to be attached as necessary by each implementing method ''' def __init__(self, *args, **kwargs): super(ResultDict, self).__init__(*args, **kwargs)
[docs] def to_df(self): '''Convert dict structure into Pandas DataFrame.''' return pd.DataFrame({k: v for k, v in self.items() if k != 'names'}, index=self['names'])
[docs] def plot(self, ax=None): '''Create bar chart of results''' Si_df = self.to_df() if isinstance(Si_df, (list, tuple)): import matplotlib.pyplot as plt # type: ignore if ax is None: fig, ax = plt.subplots(1, len(Si_df)) for idx, f in enumerate(Si_df): barplot(f, ax=ax[idx]) axes = ax else: axes = barplot(Si_df, ax=ax) return axes