From ce4cd882c2ab27ab2e26aa6d1cb68e08332e5692 Mon Sep 17 00:00:00 2001 From: podlesny <podlesny@zedat.fu-berlin.de> Date: Sun, 28 Mar 2021 23:56:45 +0200 Subject: [PATCH] now plots colorbar as legend --- data/tools/support/slip_rates.py | 10 ++++++---- 1 file changed, 6 insertions(+), 4 deletions(-) diff --git a/data/tools/support/slip_rates.py b/data/tools/support/slip_rates.py index d122199d..1002e0aa 100644 --- a/data/tools/support/slip_rates.py +++ b/data/tools/support/slip_rates.py @@ -2,13 +2,13 @@ import numpy as np import matplotlib.pyplot as plt import matplotlib.cm as cm -TANGENTIAL_COORDS = 1 +SORT_COORD = 0 def local_to_global(coords, SORT_COORD=0): return coords[:, SORT_COORD].argsort() -def slip_rates(h5file, body_ID, FINAL_TIME, patch=[], interval=[], SORT_COORD=0): - body = 'body' + str(body_ID) # 'frictionalBoundary' 'body' + str(body_ID) +def slip_rates(h5file, body_ID, FINAL_TIME, patch=[], interval=[], TANGENTIAL_COORDS=1): + body = 'body' + str(body_ID) # 'frictionalBoundary' 'body' + str(body_ID) coords = np.array(h5file[body + '/coordinates']) if len(patch) == 0: @@ -37,10 +37,12 @@ def slip_rates(h5file, body_ID, FINAL_TIME, patch=[], interval=[], SORT_COORD=0) # plot ax_slip = fig.add_subplot(1, 1, 1) - ax_slip.imshow(v_tx, interpolation='nearest', cmap=cm.Greys_r, aspect='auto', origin='lower', extent=[0, patch_size, time[0], time[-1]]) + pos = ax_slip.imshow(v_tx, interpolation='nearest', cmap=cm.Greys_r, aspect='auto', origin='lower', extent=[0, patch_size, time[0], time[-1]]) #ax_slip.set_ylim([time[-1], time[0]]) ax_slip.set_ylabel('time [s]') ax_slip.set_xlabel('horizontal coordinate [m]') + fig.colorbar(pos, ax=ax_slip) + #ax_slip.set_yscale('log') # ax_slip.set_ylim([1e-6,1e-2]) #------------------------- -- GitLab