This commit is contained in:
François Boulogne 2025-09-11 18:10:10 +02:00
parent 19cf916db9
commit cdc5d7a9b9
2 changed files with 30 additions and 26 deletions

View file

@ -1 +1 @@
__version__ = '0.1.18'
__version__ = '0.1.19'

View file

@ -107,7 +107,7 @@ def filter_signal_by_modes(time, signal, num_modes=1, bandwidth_factor=0.1, requ
return np.array(filtered_signals), dominant_freqs, time_filtered
def plot_filtered_modes(t, e, e_filtered, e_frequencies, e_time, n_modes):
def plot_filtered_modes(t, e, e_filtered, e_frequencies, e_time, n_modes, output=None):
fig, ax = plt.subplots(nrows=n_modes+1, ncols=2, figsize=(12, 4 * n_modes), gridspec_kw={'width_ratios': [3, 1]})
@ -130,6 +130,8 @@ def plot_filtered_modes(t, e, e_filtered, e_frequencies, e_time, n_modes):
ax[i+1, 1].set_xlim(left=0.5 * e_time.mean(), right=0.5*e_time.mean() + 10/e_frequencies[i])
plt.tight_layout()
plt.show()
if output:
plt.savefig(output)
@ -598,3 +600,5 @@ def plot_phases(e_time, e_filtered, e_frequencies, s_filtered, n_modes, callback
plt.tight_layout();
plt.show();
if output:
plt.savefig(output)