upd
This commit is contained in:
parent
19cf916db9
commit
cdc5d7a9b9
2 changed files with 30 additions and 26 deletions
|
@ -1 +1 @@
|
|||
__version__ = '0.1.18'
|
||||
__version__ = '0.1.19'
|
||||
|
|
|
@ -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)
|
||||
|
|
Loading…
Add table
Add a link
Reference in a new issue