Advertisement
MagicWinnie

Untitled

Dec 4th, 2018
177
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
Python 0.61 KB | None | 0 0
  1. import matplotlib.pyplot as mpl
  2. import numpy as np
  3. from scipy import signal
  4. f_sig = list(map(float, input().split(" ")))
  5. f_sig = list(map(float, f_sig))
  6. f_sig = np.fft.fft(f_sig)
  7. b, a = signal.butter(3, 0.05)
  8. zi = signal.lfilter_zi(b, a)
  9. zi = zi * int(f_sig[0])
  10. av_sig, _ = signal.lfilter(b, a,f_sig,-1,zi)
  11. zi=zi*int(av_sig[0])
  12. av_sig2, _ = signal.lfilter(b,a,av_sig,-1,zi)
  13. av_sig2 = list(map(float, av_sig2))
  14. for i in range(len(av_sig2)):
  15.       if (av_sig2[i] < 0):
  16.             av_sig2[i] = 0 - abs(av_sig2[i])
  17.       else:
  18.             av_sig2[i] = abs(av_sig2[i])
  19. mpl.plot(av_sig2)
  20. mpl.show()
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement