Guest User

Untitled

a guest
Nov 23rd, 2017
92
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
text 1.16 KB | None | 0 0
  1. def butter_bandpass_filter(data, lowcut, highcut, fs, order=2):
  2. nyq = 0.5 * fs
  3. low = lowcut /nyq
  4. high = highcut/nyq
  5. b, a = butter(order, [low, high], btype='band')
  6. #print(b,a)
  7. y = lfilter(b, a, data)
  8. return y
  9.  
  10. for i in np.arange(n):
  11. alpha1 = butter_bandpass_filter(fft1[i, :], 8.1, 12.0, 256)
  12. beta1 = butter_bandpass_filter(fft1[i, :], 16.0, 36.0, 256)
  13. gamma1 = butter_bandpass_filter(fft1[i, :], 36.1, 80, 256)
  14. delta1 = butter_bandpass_filter(fft1[i, :], 0.0, 4.0, 256)
  15. sigma1 = butter_bandpass_filter(fft1[i, :], 12.1, 16.0, 256)
  16. theta1 = butter_bandpass_filter(fft1[i, :], 4.1, 8.0, 256)
  17. sumalpha1 = sum(abs(alpha1))
  18. sumbeta1 = sum(abs(beta1))
  19. sumgamma1 = sum(abs(gamma1))
  20. sumdelta1 = sum(abs(delta1))
  21. sumsigma1 = sum(abs(sigma1))
  22. sumtheta1 = sum(abs(theta1))
  23. objects = [sumalpha1, sumbeta1, sumgamma1, sumdelta1, sumsigma1, sumtheta1]
  24. N = len(objects)
  25. ra = range(N)
  26. plt.title(signal_labels[i])
  27. plt.autoscale
  28. somestuffneeded = np.arange(6)
  29. ticks = ['alpha','beta','gamma','delta','sigma','theta']
  30. plt.xticks(somestuffneeded, ticks)
  31. plt.bar(ra, objects)
  32. plt.show()
Add Comment
Please, Sign In to add comment