From 4adfbb6a544d14cbf81cec987b9b9aa7607ac43f Mon Sep 17 00:00:00 2001 From: ftong Date: Fri, 6 Jun 2025 17:09:48 +0200 Subject: [PATCH] add 'g' as unit of colorbar for PGA and SA --- src/seismic_hazard_forecasting.py | 6 +++++- 1 file changed, 5 insertions(+), 1 deletion(-) diff --git a/src/seismic_hazard_forecasting.py b/src/seismic_hazard_forecasting.py index 79ee05b..e1ad7ac 100644 --- a/src/seismic_hazard_forecasting.py +++ b/src/seismic_hazard_forecasting.py @@ -545,6 +545,10 @@ verbose: {verbose}") gradient = np.vstack((gradient, gradient)).T gradient = np.tile(gradient, (1, width)) + colorbar_title = products[j] + if "PGA" in colorbar_title or "SA" in colorbar_title: + colorbar_title = colorbar_title + " (g)" + fig, ax = plt.subplots(figsize=((width + 40) / 100.0, (height + 20) / 100.0), dpi=100) # Increase fig size for labels ax.imshow(gradient, aspect='auto', cmap=cmap.reversed(), @@ -554,7 +558,7 @@ verbose: {verbose}") tick_positions = np.linspace(vmin, vmax_hd, num_ticks) ax.set_yticks(tick_positions) ax.set_yticklabels([f"{tick:.2f}" for tick in tick_positions]) # format tick labels - ax.set_title(products[j], pad=15) + ax.set_title(colorbar_title, loc='right', pad=15) fig.subplots_adjust(left=0.25, right=0.75, bottom=0.05, top=0.95) # Adjust Layout fig.savefig("colorbar_" + str(j) + ".svg", bbox_inches='tight') plt.close(fig)