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
|
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]})
|
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])
|
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.tight_layout()
|
||||||
plt.show()
|
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.tight_layout();
|
||||||
plt.show();
|
plt.show();
|
||||||
|
if output:
|
||||||
|
plt.savefig(output)
|
||||||
|
|
Loading…
Add table
Add a link
Reference in a new issue