import pandas as pd
def findlines(filename):
s_starts=[]
t_starts=[]
with open(filename) as file:
for linenum, linetxt in enumerate(file,1):
if "Study phase started at" in linetxt:
s_starts.append(linenum)
elif "Test phase started at" in linetxt:
t_starts.append(linenum)
return (s_starts, t_starts)
fname='MST_128210.txt'
s_starts,t_starts=findlines(fname)
n_stim_per=64
study_len = n_stim_per*2
test_len = n_stim_per*3
study_data=pd.read_csv(fname,skiprows=s_starts[0],header=0,nrows=study_len)
test_data=pd.read_csv(fname,skiprows=t_starts[0],header=0,nrows=test_len)