updated calculate integral in useful functions

This commit is contained in:
mbergmann 2024-10-23 11:10:04 +02:00
parent ac9ced6bcc
commit 702cbd08bb

View File

@ -75,9 +75,12 @@ def calculate_integral(freq, power, point, delta = 2.5):
The calculated integral around the point.
local_mean : float
The local mean value (adjacent integrals).
p_power : float
The local maxiumum power.
"""
indices = (freq >= point - delta) & (freq <= point + delta)
integral = np.trapz(power[indices], freq[indices])
p_power = np.max(power[indices])
left_indices = (freq >= point - 5 * delta) & (freq < point - delta)
right_indices = (freq > point + delta) & (freq <= point + 5 * delta)
@ -86,7 +89,7 @@ def calculate_integral(freq, power, point, delta = 2.5):
r_integral = np.trapz(power[right_indices], freq[right_indices])
local_mean = np.mean([l_integral, r_integral])
return integral, local_mean
return integral, local_mean, p_power
def extract_stim_data(stimulus):
'''