Fehlerwurst

graphs.py

@@ -4,7 +4,8 @@ import numpy as np
 from evaluation import TotalResults
 
 
-def get_error_vs_frac_plot(totalResults: TotalResults, attributes: list = [], methods: list = []) -> Figure:
+def get_error_vs_frac_plot(totalResults: TotalResults, attributes: list = [], methods: list = [],
+                           standarddevs=True, fill=True) -> Figure:
     if len(attributes) == 0 and len(methods) != 0:
         attributes = [[]]*len(methods)
     elif len(methods) == 0 and len(attributes) != 0:
@@ -31,9 +32,15 @@ def get_error_vs_frac_plot(totalResults: TotalResults, attributes: list = [], me
         for methodLabel in errorPerFraction.keys():
             errorDict: dict = errorPerFraction[methodLabel]
             fractions: list = list(errorDict.keys())
-            errors: list = [errorDict[fraction][0] for fraction in fractions]
-            stdevs: list = [errorDict[fraction][1] for fraction in fractions]
-            ax.errorbar(fractions, errors, stdevs, label=methodLabel, marker='s', capsize=5)
+            errors: np.ndarray = np.array([errorDict[fraction][0] for fraction in fractions])
+            stdevs: np.ndarray = np.array([errorDict[fraction][1] for fraction in fractions])
+            if not standarddevs:
+                ax.plot(fractions, errors, label=methodLabel, marker='s')
+            else:
+                line = ax.errorbar(fractions, errors, stdevs, label=methodLabel, marker='s', capsize=5)
+                if fill:
+                    color = line[0].get_color()
+                    ax.fill_between(fractions, errors-stdevs, errors+stdevs, alpha=0.2, facecolor=color)
 
         title: str = ''
         if len(attrs) > 0: