Data

Tips for handling autopilot data!

Split Subject Data by Session

The Subject() data interface in autopilot can return trial data as a pandas DataFrame, which can then use its groupby() method to write separate .csvs for each session

from autopilot.core.subject import Subject

# open subject file
sub = Subject('subject_id', dir='/mnt/ion-nas/autopilot/data')

# get dataframe of subject trial data
df = sub.get_trial_data('all')

for session, group in df.groupby('session'):
    group.to_csv(f'/output/location/subject_id_{session}.csv')