Mapping between behavior_session_id and regimen/stage

behavior_session_regimen_stage_df.py

import pandas as pd

from mtrain_api.utils import get_df

stages_df = get_df('stages').rename(columns={'id':'stage_id','name':'stage_name'}).drop(['script', 'script_md5', 'states'], axis=1)
states_df = get_df('states').rename(columns={'id':'state_id'})
subjects_df = get_df('subjects').rename(columns={'id':'state_id'})
subjects_df['state_id'] = subjects_df['state'].map(lambda x:x['id'])
subjects_df.drop(['state'], axis=1, inplace=True)
stages_states_df = pd.merge(stages_df, states_df, on='stage_id')
regimens_df = get_df('regimens').rename(columns={'id':'regimen_id','name':'regimen_name'}).drop(['states', 'active'], axis=1)
stages_states_regimens_df = pd.merge(stages_states_df, regimens_df, on='regimen_id')
behavior_session_df = get_df('behavior_sessions')

behavior_session_metadata_df =  pd.merge(stages_states_regimens_df, behavior_session_df, on='state_id') 

print behavior_session_metadata_df