newP3_exampleBV.py

from utils import *
data_dir = '/Users/kylemathewson/data/'
exp = 'P3'
subs = ['001','002','004','005','006','007','008','010']
sessions = ['ActiveDry','ActiveWet','PassiveWet']

event_id = {'Target': 1, 'Standard': 2}
epochs = {} #dict
all_evokeds = {}
fig = plt.figure()

for i_session, session in enumerate(sessions):
	all_lists = {'Target':[], 'Standard':[]}
	for sub in subs:
		print('Loading data for subject ' + sub)
		#Load Data
		raw = LoadBVData(sub,session,data_dir,exp)
		#Pre-Process EEG Data
		epochs[session] = PreProcess(raw,event_id,
							emcp_epochs=True, rereference=True,
							plot_erp=False, rej_thresh_uV=1000, 
							epoch_time=(-.2,1), baseline=(-.2,0) )
		#create evoked dict by averaging all the epochs for this test subject
		all_lists['Target'].append(epochs[session]['Target'].average())
		all_lists['Standard'].append(epochs[session]['Standard'].average())
	#create the dict with condition keys and values are lists over subs	
	all_evokeds['Target'] = all_lists['Target']
	all_evokeds['Standard'] = all_lists['Standard']
	ax = plt.subplot(3,1,i_session+1)
	viz.plot_compare_evokeds(all_evokeds,picks=[6],axes=ax,
							title=session,show=False,ci=.95)
plt.show()