research/mysignal
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

add option dominant freqs

Browse source
This commit is contained in:
François Boulogne 2026-01-06 11:03:58 +01:00
parent d007702712
commit 6f449512d1
2 changed files with 20 additions and 18 deletions
Download patch file Download diff file Expand all files Collapse all files

2
mysignal/__init__.py
View file

@ -1 +1 @@
__version__ = '0.1.23'
__version__ = '0.1.24'

4
mysignal/phasefreq.py
View file

@ -12,7 +12,7 @@ from scipy.stats import linregress
from scipy.optimize import curve_fit
def filter_signal_by_modes(time, signal, num_modes=1, bandwidth_factor=0.1, required_cycles=100):
def filter_signal_by_modes(time, signal, num_modes=1, bandwidth_factor=0.1, required_cycles=100, dominant_freqs=None):
"""
Filtre un signal pour extraire ses composantes fréquentielles dominantes.
@ -20,6 +20,7 @@ def filter_signal_by_modes(time, signal, num_modes=1, bandwidth_factor=0.1, requ
time (array-like): Array contenant les données temporelles.
signal (array-like): Array contenant les données du signal.
num_modes (int): Nombre de modes fréquentiels à extraire.
dominant_freqs: les frequences des modes, if None, computed internally
Returns:
tuple: (filtered_signals, frequencies, time_filtered)
@ -38,6 +39,7 @@ def filter_signal_by_modes(time, signal, num_modes=1, bandwidth_factor=0.1, requ
# Supprimer la composante DC
signal_clean = signal - np.mean(signal)
if dominant_freqs is None:
# Calculer la FFT
n = len(time)
fft_signal = fft(signal_clean)