diff --git a/eventio/scripts/plot_hists.py b/eventio/scripts/plot_hists.py index 2c9e7f24..1ccbe0fe 100644 --- a/eventio/scripts/plot_hists.py +++ b/eventio/scripts/plot_hists.py @@ -55,6 +55,7 @@ def main(): non_zero_y, = np.where(marginal_y != 0) plt.ylim(y_bins[non_zero_y[0]], y_bins[non_zero_y[-1] + 1]) + plt.colorbar(label='Number of Events') else: centers = 0.5 * (x_bins[:-1] + x_bins[1:])