diff --git a/mysignal/__init__.py b/mysignal/__init__.py index 13b7089..e6d0c4f 100644 --- a/mysignal/__init__.py +++ b/mysignal/__init__.py @@ -1 +1 @@ -__version__ = '0.1.11' +__version__ = '0.1.12' diff --git a/mysignal/phasefreq.py b/mysignal/phasefreq.py index 9876a6d..f0634cd 100644 --- a/mysignal/phasefreq.py +++ b/mysignal/phasefreq.py @@ -573,7 +573,7 @@ def analyze_signal_wavelet(time, e_signal, s_signal, freq_rtol=0.01): return res -def plot_phases(e_time, e_filtered, s_filtered, n_modes, callback=analyze_signal_wavelet): +def plot_phases(e_time, e_filtered, s_filtered, e_frequencies, n_modes, callback=analyze_signal_wavelet): fig, ax = plt.subplots(nrows=n_modes, ncols=2, figsize=(12, 6), gridspec_kw={'width_ratios': [3, 1]}) @@ -589,7 +589,7 @@ def plot_phases(e_time, e_filtered, s_filtered, n_modes, callback=analyze_signal ax[mod, 1].plot(e_time, e_filtered[mod], label='e') ax[mod, 1].plot(e_time, s_filtered[mod], label='s') - ax[mod, 1].set_xlim(left=0.5 * e_time.mean(), right=0.5*e_time.mean() + 10/e_frequencies[i]) + ax[mod, 1].set_xlim(left=0.5 * e_time.mean(), right=0.5*e_time.mean() + 10/e_frequencies[mod]) for a in ax[:, 0]: a.legend()