diff --git a/mysignal/__init__.py b/mysignal/__init__.py index 1c98a23..850505a 100644 --- a/mysignal/__init__.py +++ b/mysignal/__init__.py @@ -1 +1 @@ -__version__ = '0.1.9' +__version__ = '0.1.10' diff --git a/mysignal/phasefreq.py b/mysignal/phasefreq.py index 7b9c5ee..dffacda 100644 --- a/mysignal/phasefreq.py +++ b/mysignal/phasefreq.py @@ -574,19 +574,24 @@ def analyze_signal_wavelet(time, e_signal, s_signal, freq_rtol=0.01): def plot_phases(e_time, e_filtered, s_filtered, n_modes, callback=analyze_signal_wavelet): - fig, ax = plt.subplots(nrows=n_modes, figsize=(12, 6)) + fig, ax = plt.subplots(nrows=n_modes, ncols=2, figsize=(12, 6), gridspec_kw={'width_ratios': [3, 1]}) for mod in range(n_modes): res = callback(e_time, e_filtered[mod], s_filtered[mod], freq_rtol=0.3) - ax[mod].set_title(f'Freq: {res['freq_e']:.3f}, Phase: {res['phase']:.3f}, Delay: {res['delay']:.3f}' ) - ax[mod].plot(e_time, e_filtered[mod], label='e') - ax[mod].plot(e_time, s_filtered[mod], label='s') - ax[mod].set_xlim(left=20, right=230) - - for a in ax: + ax[mod, 0].set_title(f'Freq: {res['freq_e']:.3f}, Phase: {res['phase']:.3f}, Delay: {res['delay']:.3f}' ) + ax[mod, 0].plot(e_time, e_filtered[mod], label='e') + ax[mod, 0].plot(e_time, s_filtered[mod], label='s') + #ax[mod].set_xlim(left=20, right=230) + + + ax[mod, 1].plot(e_time, e_filtered[mod], label='e') + ax[mod, 1].plot(e_time, s_filtered[mod], label='s') + ax[i+1, 1].set_xlim(left=0.5 * e_time.mean(), right=0.5*e_time.mean() + 10/e_frequencies[i]) + + for a in ax[:, 0]: a.legend() plt.tight_layout();