feat: add normalization func

This commit is contained in:
Skudalen 2021-06-24 09:40:30 +02:00
parent 72b2cff336
commit 9e60922875

View File

@ -104,20 +104,21 @@ def prep_df_for_trans(df:DataFrame):
return x, y return x, y
def normalize_wave(y_values): def normalize_wave(y_values):
return None y = np.int16((y_values / y_values.max()) * 32767)
return y
def transformed_df(df:DataFrame): def transformed_df(df:DataFrame):
x, y = prep_df_for_trans(df) x, y = prep_df_for_trans(df)
y_values = fft(y) y_values = fft(y)
return new_df return None
def plot_df(df:DataFrame): def plot_df(df:DataFrame):
lines = df.plot.line(x='timestamp') lines = df.plot.line(x='timestamp')
plt.show() plt.show()
handler = CSV_handler handler = Handler.CSV_handler
file = "/Exp20201205_2myo_hardTypePP/HaluskaMarek_20201207_1810/myoLeftEmg.csv" file = "/Exp20201205_2myo_hardTypePP/HaluskaMarek_20201207_1810/myoLeftEmg.csv"
df = handler.get_time_emg_table(file, 1) df = handler.get_time_emg_table(file, 1)
#plot_df(df) #plot_df(df)