kopia lustrzana https://github.com/gabrielegilardi/SignalFilters
49 wiersze
1000 B
Python
49 wiersze
1000 B
Python
"""
|
|
Filters for time series.
|
|
|
|
Copyright (c) 2020 Gabriele Gilardi
|
|
|
|
ToDo:
|
|
- use NaN/input values for points not filtered?
|
|
- return idx?
|
|
- util to test filter (impulse, utils)
|
|
- warning in filter when wrong order?
|
|
- use self.a and self.b
|
|
- remove a and b from plots
|
|
- in comments write what filters do
|
|
- is necessary to copy X for Y untouched?
|
|
"""
|
|
|
|
import sys
|
|
import numpy as np
|
|
import filters as flt
|
|
import utils as utl
|
|
|
|
# Read data to filter
|
|
if len(sys.argv) != 2:
|
|
print("Usage: python test.py <data_file>")
|
|
sys.exit(1)
|
|
data_file = sys.argv[1] + '.csv'
|
|
|
|
# Read data from a csv file
|
|
data = np.loadtxt(data_file, delimiter=',')
|
|
n_samples = data.shape[0]
|
|
data = data.reshape(n_samples, -1)
|
|
|
|
spx = flt.Filter(data)
|
|
|
|
# args = (spx.X, spx.SMA(N=5), spx.EMA(alpha=0.7))
|
|
# utl.plot_signals(args)
|
|
|
|
# alpha = 0.8
|
|
# bb = np.array([alpha])
|
|
# aa = np.array([1.0, alpha - 1.0])
|
|
|
|
|
|
res, bb, aa = spx.GaussHigh(N=3, P=10)
|
|
print(bb)
|
|
print(aa)
|
|
|
|
utl.plot_frequency_response(bb, aa)
|
|
utl.plot_lag_response(bb, aa)
|