import pyedflib
import numpy as np
import matplotlib.pyplot as plt
from scipy import signal
import pylab

f = pyedflib.EdfReader("n7.edf")
n = f.signals_in_file
signal_labels = f.getSignalLabels()
sigbufs = np.zeros((n, f.getNSamples()[0]))
for i in np.arange(n):
     sigbufs[i, :] = f.readSignal(i)
print (sigbufs)
print(signal_labels)

sygnal = []
sygnal = sigbufs[0]
fs = 10000


for i in range(0,len(sigbufs)):
        print(sigbufs[i])
        sygnal[0:90000]=sigbufs[i][10000:100000]
        fs = f.getSampleFrequency(i)
        f, t, Sxx = signal.spectrogram(sygnal, fs)
        plt.pcolormesh(t, f, Sxx)
        plt.ylabel('Frequency [Hz]')
        plt.xlabel('Time [sec]')
        plt.show()