diff --git a/modules/util/bar_plot.py b/modules/util/bar_plot.py index 85d98c5a688579722adb1928bbe9e0f8bd233d9d..dd982bd731810717c11eb167421a5ea36eec0892 100644 --- a/modules/util/bar_plot.py +++ b/modules/util/bar_plot.py @@ -1,11 +1,8 @@ import numpy as np import matplotlib.pyplot as plt - -# set width of bar -barWidth = 0.10 -def do_plot(group_names, group_values, value_labels, bar_colors, xlabel): +def do_plot(group_names, group_values, value_labels, bar_colors, xlabel, ylabel, barWidth=0.1): num_groups = len(group_names) num_bars = len(group_values) @@ -21,7 +18,8 @@ def do_plot(group_names, group_values, value_labels, bar_colors, xlabel): plt.xlabel(xlabel, fontweight='bold') plt.xticks([r + barWidth for r in range(num_groups)], group_names) - plt.ylabel('count', fontweight='bold') + if ylabel is not None: + plt.ylabel('count', fontweight='bold') # Create legend & Show graphic plt.legend()