upd
This commit is contained in:
parent
cb0dfbdde9
commit
5b47050819
2 changed files with 8 additions and 15 deletions
|
|
@ -1 +1 @@
|
||||||
__version__ = '0.1.29'
|
__version__ = '0.1.30'
|
||||||
|
|
|
||||||
|
|
@ -597,26 +597,16 @@ def get_phases(e_time, e_filtered, s_filtered, n_modes, callback=analyze_signal_
|
||||||
for mod in range(n_modes)
|
for mod in range(n_modes)
|
||||||
)
|
)
|
||||||
|
|
||||||
return pd.DataFrame(results)
|
df = pd.DataFrame(results)
|
||||||
|
df['puls'] = 2 * np.pi * df['freq']
|
||||||
# res = []
|
return df
|
||||||
#
|
|
||||||
# for mod in range(n_modes):
|
|
||||||
# output = callback(e_time, e_filtered[mod], s_filtered[mod], freq_rtol=0.3)
|
|
||||||
# res.append({
|
|
||||||
# 'mode': mod,
|
|
||||||
# 'freq': output['freq_e'],
|
|
||||||
# 'phase': output['phase'],
|
|
||||||
# 'delay': output['delay']
|
|
||||||
# })
|
|
||||||
# return pd.DataFrame(res)
|
|
||||||
|
|
||||||
|
|
||||||
def plot_phases(e_time, e_filtered, e_frequencies, s_filtered, n_modes, df, output=None):
|
def plot_phases(e_time, e_filtered, e_frequencies, s_filtered, n_modes, df, output=None):
|
||||||
"""
|
"""
|
||||||
"""
|
"""
|
||||||
|
|
||||||
fig, ax = plt.subplots(nrows=n_modes, ncols=2, figsize=(12, 2 * n_modes), gridspec_kw={'width_ratios': [3, 1]})
|
fig, ax = plt.subplots(nrows=n_modes, ncols=3, figsize=(12, 2 * n_modes), gridspec_kw={'width_ratios': [2, 1, 1]})
|
||||||
|
|
||||||
for mod in range(n_modes):
|
for mod in range(n_modes):
|
||||||
freq = df[df['mode'] == mod]['freq'].iloc[0]
|
freq = df[df['mode'] == mod]['freq'].iloc[0]
|
||||||
|
|
@ -633,6 +623,9 @@ def plot_phases(e_time, e_filtered, e_frequencies, s_filtered, n_modes, df, outp
|
||||||
ax[mod, 1].plot(e_time, s_filtered[mod], label='s')
|
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[mod])
|
ax[mod, 1].set_xlim(left=0.5 * e_time.mean(), right=0.5*e_time.mean() + 10/e_frequencies[mod])
|
||||||
|
|
||||||
|
ax[mod, 2].plot(e_time, e_filtered[mod] / e_filtered[mod].max(), label='e')
|
||||||
|
ax[mod, 2].plot(e_time, s_filtered[mod] / s_filtered[mod].max(), label='s')
|
||||||
|
ax[mod, 2].set_xlim(left=0.5 * e_time.mean(), right=0.5*e_time.mean() + 2/e_frequencies[mod])
|
||||||
for a in ax[:, 0]:
|
for a in ax[:, 0]:
|
||||||
a.legend()
|
a.legend()
|
||||||
|
|
||||||
|
|
|
||||||
Loading…
Add table
Add a link
Reference in a new issue