trial_index,trial_type,reward_inference_query,choice_inference_query,reward_prediction_query,choice_prediction_query 0,1,0,0,0,0 1,2,0,0,0,0 2,3,0,0,1,0 3,1,0,0,1,0 4,3,0,1,0,0 5,2,1,0,0,0 6,1,0,0,0,1 7,3,0,0,0,0 8,2,0,0,1,0 9,3,0,0,0,0 10,2,0,0,0,0