for left/right plots

Author: rachelstephlee

src/rachel_analysis_utils/analysis_utils.py

@@ -1,6 +1,7 @@
 import numpy as np
 import pandas as pd 
 from aind_dynamic_foraging_basic_analysis.metrics import trial_metrics
+import scipy.stats as stats
 
 
 

src/rachel_analysis_utils/nwb_utils.py

@@ -9,7 +9,14 @@
 from aind_dynamic_foraging_data_utils import code_ocean_utils as co_utils
 
 
-
+import copy 
+
+def split_nwb(nwb):
+    nwb_split = copy.deepcopy(nwb)
+    nwb_split.df_trials_left = nwb.df_trials.query('choice == 0.0')
+    nwb_split.df_trials_right = nwb.df_trials.query('choice == 1.0')
+    nwb_split.df_trials_ignore = nwb.df_trials.query('choice == 2.0')
+    return nwb_split
 
 
 class dummy_nwb: