read_data_pkl.py 512 B

12345678910111213141516171819
  1. import pandas as pd
  2. import matplotlib.pylab as plt
  3. filename = '../Data/interictal/01ref/pkl/activity_SU_4ms.pkl'
  4. data = pd.read_pickle(filename)
  5. print(data.columns)
  6. existing_binnings = data['binning'].tolist()
  7. for binning in existing_binnings:
  8. activity = data[data['binning'] == binning]['activity'].tolist()[0]
  9. plt.figure()
  10. plt.plot(activity)
  11. plt.xlabel('Time [4 ms]')
  12. plt.ylabel('Activity [# recorded spikes]')
  13. plt.title('Activity in the {} hemisphere'.format(binning))
  14. plt.show()