np.savez(base_folder + 'behavioral_traces.npz', **res_bt) #%% with np.load(base_folder + 'behavioral_traces.npz') as ld: res_bt = dict(**ld) #%% pl.close() tm = res_bt['time'] f_rate_bh = old_div(1, np.median(np.diff(tm))) ISI = res_bt['trial_info'][0][3] - res_bt['trial_info'][0][2] eye_traces = np.array(res_bt['eyelid']) idx_CS_US = res_bt['idx_CS_US'] idx_US = res_bt['idx_US'] idx_CS = res_bt['idx_CS'] idx_ALL = np.sort(np.hstack([idx_CS_US, idx_US, idx_CS])) eye_traces, amplitudes_at_US, trig_CRs = gc.process_eyelid_traces( eye_traces, tm, idx_CS_US, idx_US, idx_CS, thresh_CR=.15, time_CR_on=-.1, time_US_on=.05) idxCSUSCR = trig_CRs['idxCSUSCR'] idxCSUSNOCR = trig_CRs['idxCSUSNOCR'] idxCSCR = trig_CRs['idxCSCR'] idxCSNOCR = trig_CRs['idxCSNOCR'] idxNOCR = trig_CRs['idxNOCR'] idxCR = trig_CRs['idxCR'] idxUS = trig_CRs['idxUS'] idxCSCSUS = np.concatenate([idx_CS, idx_CS_US]) pl.plot(tm, np.mean(eye_traces[idxCSUSCR], 0)) pl.plot(tm, np.mean(eye_traces[idxCSUSNOCR], 0)) pl.plot(tm, np.mean(eye_traces[idxCSCR], 0)) pl.plot(tm, np.mean(eye_traces[idxCSNOCR], 0)) pl.plot(tm, np.mean(eye_traces[idx_US], 0))
with np.load(base_folder+'all_triggers.npz') as at: triggers_img=at['triggers'] trigger_names_img=at['trigger_names'] with np.load(base_folder+'behavioral_traces.npz') as ld: res_bt = dict(**ld) tm=res_bt['time'] f_rate_bh=1/np.median(np.diff(tm)) ISI=res_bt['trial_info'][0][3]-res_bt['trial_info'][0][2] eye_traces=np.array(res_bt['eyelid']) idx_CS_US=res_bt['idx_CS_US'] idx_US=res_bt['idx_US'] idx_CS=res_bt['idx_CS'] idx_ALL=np.sort(np.hstack([idx_CS_US,idx_US,idx_CS])) eye_traces,amplitudes_at_US, trig_CRs=gc.process_eyelid_traces(eye_traces,tm,idx_CS_US,idx_US,idx_CS,thresh_CR=.15,time_CR_on=-.1,time_US_on=.05) idxCSUSCR = trig_CRs['idxCSUSCR'] idxCSUSNOCR = trig_CRs['idxCSUSNOCR'] idxCSCR = trig_CRs['idxCSCR'] idxCSNOCR = trig_CRs['idxCSNOCR'] idxNOCR = trig_CRs['idxNOCR'] idxCR = trig_CRs['idxCR'] idxUS = trig_CRs['idxUS'] idxCSCSUS=np.concatenate([idx_CS,idx_CS_US]) with open(base_folder+'traces.pk','r') as f: locals().update(pickle.load(f)) triggers_img=np.array(triggers_img) idx_expected_US= np.repeat( np.nanmedian(triggers_img[:,1]),len(triggers_img[:,1]))
learning_phase=0 print 'early' else: if day != session_now: session_id += 1 session_now=day chunk=re.search('_00[0-9][0-9][0-9]_',nm.split('/')[9]).group(0)[3:-1] idx_CS_US=np.where(tr_bh[:,-2]==2)[0] idx_US=np.where(tr_bh[:,-2]==1)[0] idx_CS=np.where(tr_bh[:,-2]==0)[0] idx_ALL=np.sort(np.hstack([idx_CS_US,idx_US,idx_CS])) eye_traces,amplitudes_at_US, trig_CRs=process_eyelid_traces(eye,tm,idx_CS_US,idx_US,idx_CS,thresh_CR=thresh_CR,time_CR_on=time_CR_on,time_US_on=time_US_on) idxCSUSCR = trig_CRs['idxCSUSCR'] idxCSUSNOCR = trig_CRs['idxCSUSNOCR'] idxCSCR = trig_CRs['idxCSCR'] idxCSNOCR = trig_CRs['idxCSNOCR'] idxNOCR = trig_CRs['idxNOCR'] idxCR = trig_CRs['idxCR'] idxUS = trig_CRs['idxUS'] idxCSCSUS=np.concatenate([idx_CS,idx_CS_US]) wheel_traces, movement_at_CS, trigs_mov = process_wheel_traces(np.array(whe),tm,thresh_MOV_iqr=thresh_MOV_iqr,time_CS_on=time_CS_on_MOV,time_US_on=time_US_on_MOV) print 'fraction with movement:' + str(len(trigs_mov['idxMOV'])*1./len(trigs_mov['idxNO_MOV'])) mn_idx_CS_US =np.intersect1d(idx_CS_US,trigs_mov['idxNO_MOV'])