You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
96 lines
1.8 KiB
96 lines
1.8 KiB
# coding: utf-8 |
|
|
|
# In[1]: |
|
|
|
|
|
import os |
|
import sys |
|
import pyqtgraph as pg |
|
import numpy as np |
|
|
|
|
|
# In[7]: |
|
|
|
|
|
dir_path = os.path.abspath("") |
|
|
|
win = pg.GraphicsWindow() |
|
win.setBackground("w") |
|
p1 = win.addPlot( |
|
title="Pauser PSTH", |
|
row=0, |
|
col=0, |
|
labels={"bottom": "T (ms)", "left": "# of spikes"}, |
|
) |
|
p2 = win.addPlot( |
|
title="Buildup PSTH", |
|
row=1, |
|
col=0, |
|
labels={"bottom": "T (ms)", "left": "# of spikes"}, |
|
) |
|
p3 = win.addPlot( |
|
title="Wide Chopper PSTH", |
|
row=2, |
|
col=0, |
|
labels={"bottom": "T (ms)", "left": "# of spikes"}, |
|
) |
|
|
|
# In[ ]: |
|
|
|
bins = np.arange(0, 80, 0.5) |
|
|
|
PB_spike_data = [] |
|
with open("Ad081098_065_PauserBuildup_psth.txt", "r+") as df: |
|
for x in df: |
|
x = x.strip("\n").strip() |
|
x = float(x) * 1e-3 |
|
if x: |
|
PB_spike_data.append(x) |
|
histogram, binedges = np.histogram(PB_spike_data, bins) |
|
p1.plot( |
|
binedges, |
|
histogram, |
|
stepMode=True, |
|
fillBrush=(0, 0, 0, 255), |
|
brush=pg.mkBrush("k"), |
|
fillLevel=0, |
|
) |
|
B_spike_data = [] |
|
with open("Ad041599_062_Buildup_psth.txt", "r+") as df: |
|
for x in df: |
|
x = x.strip("\n").strip() |
|
x = float(x) * 1e-3 |
|
if x: |
|
B_spike_data.append(x) |
|
histogram, binedges = np.histogram(B_spike_data, bins) |
|
p2.plot( |
|
binedges, |
|
histogram, |
|
stepMode=True, |
|
fillBrush=(0, 0, 0, 255), |
|
brush=pg.mkBrush("k"), |
|
fillLevel=0, |
|
) |
|
C_spike_data = [] |
|
with open("Ad081998_199_WideChopper_psth.txt", "r+") as df: |
|
for x in df: |
|
x = x.strip("\n").strip() |
|
x = float(x) * 1e-3 |
|
if x: |
|
C_spike_data.append(x) |
|
histogram, binedges = np.histogram(C_spike_data, bins) |
|
p3.plot( |
|
binedges, |
|
histogram, |
|
stepMode=True, |
|
fillBrush=(0, 0, 0, 255), |
|
brush=pg.mkBrush("k"), |
|
fillLevel=0, |
|
) |
|
# In[ ]: |
|
|
|
|
|
win.show() |
|
print("finished") |
|
if sys.flags.interactive == 0: |
|
pg.QtGui.QApplication.exec_()
|
|
|