model-fitting.ipynb

{
 "cells": [
  {
   "cell_type": "markdown",
   "metadata": {},
   "source": [
    "# Model fitting\n",
    "author: pam popp (pamop@nyu.edu)\n",
    "date: june 25, 2024"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 1,
   "metadata": {},
   "outputs": [],
   "source": [
    "# imports\n",
    "import numpy as np\n",
    "import pandas as pd\n",
    "import os\n",
    "\n",
    "# display all columns of dataframes\n",
    "pd.set_option('display.max_columns', None)\n",
    "pd.set_option('display.max_rows', 60)\n",
    "\n",
    "# # Prints out a game trial by trial in ascii\n",
    "from utils import ascii_map as amap\n",
    "\n",
    "# seed RNG\n",
    "np.random.seed(0)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 6,
   "metadata": {},
   "outputs": [
    {
     "data": {
      "text/html": [
       "<div>\n",
       "<style scoped>\n",
       "    .dataframe tbody tr th:only-of-type {\n",
       "        vertical-align: middle;\n",
       "    }\n",
       "\n",
       "    .dataframe tbody tr th {\n",
       "        vertical-align: top;\n",
       "    }\n",
       "\n",
       "    .dataframe thead th {\n",
       "        text-align: right;\n",
       "    }\n",
       "</style>\n",
       "<table border=\"1\" class=\"dataframe\">\n",
       "  <thead>\n",
       "    <tr style=\"text-align: right;\">\n",
       "      <th></th>\n",
       "      <th>subjid</th>\n",
       "      <th>session</th>\n",
       "      <th>trialNum</th>\n",
       "      <th>gameNum</th>\n",
       "      <th>costCond</th>\n",
       "      <th>resourceCond</th>\n",
       "      <th>visibilityCond</th>\n",
       "      <th>redFirst</th>\n",
       "      <th>counterbalance</th>\n",
       "      <th>objectLayer</th>\n",
       "      <th>eventName</th>\n",
       "      <th>turnCount</th>\n",
       "      <th>agent</th>\n",
       "      <th>target</th>\n",
       "      <th>turnStartTimestamp</th>\n",
       "      <th>responseTime</th>\n",
       "      <th>decisionMadeTimestamp</th>\n",
       "      <th>dataSavedTimestamp</th>\n",
       "      <th>redXloc</th>\n",
       "      <th>redYloc</th>\n",
       "      <th>purpleXloc</th>\n",
       "      <th>purpleYloc</th>\n",
       "      <th>farmItems</th>\n",
       "      <th>farmBox</th>\n",
       "      <th>purpleBackpack</th>\n",
       "      <th>redBackpack</th>\n",
       "      <th>gameover</th>\n",
       "      <th>legalMoves</th>\n",
       "      <th>purpleBackpackSize</th>\n",
       "      <th>purpleEnergy</th>\n",
       "      <th>purpleScore</th>\n",
       "      <th>purplePoints</th>\n",
       "      <th>purplePointsCumulative</th>\n",
       "      <th>redBackpackSize</th>\n",
       "      <th>redEnergy</th>\n",
       "      <th>redScore</th>\n",
       "      <th>redPoints</th>\n",
       "      <th>redPointsCumulative</th>\n",
       "      <th>lastTrial</th>\n",
       "      <th>energy_expense</th>\n",
       "      <th>targetColor</th>\n",
       "      <th>targetCat</th>\n",
       "      <th>actionCat</th>\n",
       "      <th>other_expense</th>\n",
       "      <th>ownBPsize</th>\n",
       "      <th>step_expense</th>\n",
       "      <th>step_other_expense</th>\n",
       "      <th>step_own_expense</th>\n",
       "      <th>step_box_expense</th>\n",
       "      <th>helping_event</th>\n",
       "      <th>helped</th>\n",
       "      <th>meFirst</th>\n",
       "      <th>steps</th>\n",
       "      <th>helpful_steps</th>\n",
       "      <th>player_helped_lasttrial</th>\n",
       "      <th>partner_helped_lasttrial</th>\n",
       "      <th>player_helpingevents_lasttrial</th>\n",
       "      <th>partner_helpingevents_lasttrial</th>\n",
       "      <th>player_helpfulsteps_lasttrial</th>\n",
       "      <th>partner_helpfulsteps_lasttrial</th>\n",
       "      <th>otherBPsize</th>\n",
       "      <th>ownScore</th>\n",
       "      <th>otherScore</th>\n",
       "      <th>ownEnergy</th>\n",
       "      <th>otherEnergy</th>\n",
       "      <th>ownPoints</th>\n",
       "      <th>otherPoints</th>\n",
       "      <th>ownPointsCumulative</th>\n",
       "      <th>otherPointsCumulative</th>\n",
       "      <th>ownBackpack</th>\n",
       "      <th>otherBackpack</th>\n",
       "      <th>nOwnVegFarm</th>\n",
       "      <th>nOtherVegFarm</th>\n",
       "      <th>nOwnVegBox</th>\n",
       "      <th>nOtherVegBox</th>\n",
       "      <th>nOwnVegInOwnBP</th>\n",
       "      <th>nOwnVegInOtherBP</th>\n",
       "      <th>nOtherVegInOwnBP</th>\n",
       "      <th>nOtherVegInOtherBP</th>\n",
       "      <th>nAnyVegInOwnBP</th>\n",
       "      <th>nAnyVegInOtherBP</th>\n",
       "      <th>ownHelpingEventsToDate</th>\n",
       "      <th>otherHelpingEventsToDate</th>\n",
       "      <th>ownHelpfulStepsToDate</th>\n",
       "      <th>otherHelpingStepsToDate</th>\n",
       "      <th>farmContents</th>\n",
       "      <th>ownDistanceToClosestOwnVeg</th>\n",
       "      <th>ownDistanceToClosestOtherVeg</th>\n",
       "      <th>otherDistanceToClosestOwnVeg</th>\n",
       "      <th>otherDistanceToClosestOtherVeg</th>\n",
       "    </tr>\n",
       "  </thead>\n",
       "  <tbody>\n",
       "    <tr>\n",
       "      <th>0</th>\n",
       "      <td>4ISIiFA1J99JScred</td>\n",
       "      <td>4ISIiFA1J99JSc</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>high</td>\n",
       "      <td>even</td>\n",
       "      <td>full</td>\n",
       "      <td>True</td>\n",
       "      <td>12.0</td>\n",
       "      <td>Items04</td>\n",
       "      <td>targetPicked</td>\n",
       "      <td>0</td>\n",
       "      <td>red</td>\n",
       "      <td>Tomato04</td>\n",
       "      <td>1678818021993</td>\n",
       "      <td>3356.0</td>\n",
       "      <td>1678818025349</td>\n",
       "      <td>1678818025349</td>\n",
       "      <td>2</td>\n",
       "      <td>15</td>\n",
       "      <td>3</td>\n",
       "      <td>16</td>\n",
       "      <td>Tomato04(9,13) Turnip02(14,15) Turnip00(11,7) ...</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>False</td>\n",
       "      <td>red_none(2,15) Tomato04(9,13) Turnip02(14,15) ...</td>\n",
       "      <td>4</td>\n",
       "      <td>100</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>4</td>\n",
       "      <td>100</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>False</td>\n",
       "      <td>18.0</td>\n",
       "      <td>red</td>\n",
       "      <td>redVeg</td>\n",
       "      <td>ownVeg</td>\n",
       "      <td>0.0</td>\n",
       "      <td>4</td>\n",
       "      <td>9.0</td>\n",
       "      <td>0.0</td>\n",
       "      <td>9.0</td>\n",
       "      <td>0.0</td>\n",
       "      <td>0</td>\n",
       "      <td>False</td>\n",
       "      <td>True</td>\n",
       "      <td>9</td>\n",
       "      <td>0</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>4</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>100</td>\n",
       "      <td>100</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>7</td>\n",
       "      <td>7</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>Tomato04(9,13) Turnip02(14,15) Turnip00(11,7) ...</td>\n",
       "      <td>8.0</td>\n",
       "      <td>8.0</td>\n",
       "      <td>8.0</td>\n",
       "      <td>8.0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>1</th>\n",
       "      <td>4ISIiFA1J99JScpurple</td>\n",
       "      <td>4ISIiFA1J99JSc</td>\n",
       "      <td>3</td>\n",
       "      <td>0</td>\n",
       "      <td>high</td>\n",
       "      <td>even</td>\n",
       "      <td>full</td>\n",
       "      <td>True</td>\n",
       "      <td>12.0</td>\n",
       "      <td>Items04</td>\n",
       "      <td>targetPicked</td>\n",
       "      <td>1</td>\n",
       "      <td>purple</td>\n",
       "      <td>Eggplant01</td>\n",
       "      <td>1678818026786</td>\n",
       "      <td>3786.0</td>\n",
       "      <td>1678818030572</td>\n",
       "      <td>1678818030573</td>\n",
       "      <td>9</td>\n",
       "      <td>14</td>\n",
       "      <td>3</td>\n",
       "      <td>16</td>\n",
       "      <td>Tomato04(22,2) Turnip02(14,15) Turnip00(11,7) ...</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>Tomato04(22,2)</td>\n",
       "      <td>False</td>\n",
       "      <td>purple_none(3,16) Turnip02(14,15) Turnip00(11,...</td>\n",
       "      <td>4</td>\n",
       "      <td>100</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>4</td>\n",
       "      <td>82</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>False</td>\n",
       "      <td>16.0</td>\n",
       "      <td>purple</td>\n",
       "      <td>purpleVeg</td>\n",
       "      <td>ownVeg</td>\n",
       "      <td>0.0</td>\n",
       "      <td>4</td>\n",
       "      <td>8.0</td>\n",
       "      <td>0.0</td>\n",
       "      <td>8.0</td>\n",
       "      <td>0.0</td>\n",
       "      <td>0</td>\n",
       "      <td>False</td>\n",
       "      <td>False</td>\n",
       "      <td>8</td>\n",
       "      <td>0</td>\n",
       "      <td>NaN</td>\n",
       "      <td>False</td>\n",
       "      <td>NaN</td>\n",
       "      <td>0.0</td>\n",
       "      <td>NaN</td>\n",
       "      <td>0.0</td>\n",
       "      <td>4</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>100</td>\n",
       "      <td>82</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>NaN</td>\n",
       "      <td>Tomato04(22,2)</td>\n",
       "      <td>7</td>\n",
       "      <td>6</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>NaN</td>\n",
       "      <td>0.0</td>\n",
       "      <td>NaN</td>\n",
       "      <td>0.0</td>\n",
       "      <td>Turnip02(14,15) Turnip00(11,7) Strawberry01(13...</td>\n",
       "      <td>8.0</td>\n",
       "      <td>8.0</td>\n",
       "      <td>5.0</td>\n",
       "      <td>0.0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>2</th>\n",
       "      <td>4ISIiFA1J99JScred</td>\n",
       "      <td>4ISIiFA1J99JSc</td>\n",
       "      <td>5</td>\n",
       "      <td>0</td>\n",
       "      <td>high</td>\n",
       "      <td>even</td>\n",
       "      <td>full</td>\n",
       "      <td>True</td>\n",
       "      <td>12.0</td>\n",
       "      <td>Items04</td>\n",
       "      <td>targetPicked</td>\n",
       "      <td>2</td>\n",
       "      <td>red</td>\n",
       "      <td>Strawberry02</td>\n",
       "      <td>1678818033697</td>\n",
       "      <td>1826.0</td>\n",
       "      <td>1678818035523</td>\n",
       "      <td>1678818035523</td>\n",
       "      <td>9</td>\n",
       "      <td>13</td>\n",
       "      <td>5</td>\n",
       "      <td>11</td>\n",
       "      <td>Tomato04(22,2) Turnip02(14,15) Turnip00(11,7) ...</td>\n",
       "      <td>NaN</td>\n",
       "      <td>Eggplant01(22,5)</td>\n",
       "      <td>Tomato04(22,2)</td>\n",
       "      <td>False</td>\n",
       "      <td>red_none(9,13) box(16,5) Turnip02(14,15) Turni...</td>\n",
       "      <td>4</td>\n",
       "      <td>84</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>4</td>\n",
       "      <td>82</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>False</td>\n",
       "      <td>2.0</td>\n",
       "      <td>red</td>\n",
       "      <td>redVeg</td>\n",
       "      <td>ownVeg</td>\n",
       "      <td>0.0</td>\n",
       "      <td>4</td>\n",
       "      <td>1.0</td>\n",
       "      <td>0.0</td>\n",
       "      <td>1.0</td>\n",
       "      <td>0.0</td>\n",
       "      <td>0</td>\n",
       "      <td>False</td>\n",
       "      <td>True</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>False</td>\n",
       "      <td>False</td>\n",
       "      <td>0.0</td>\n",
       "      <td>0.0</td>\n",
       "      <td>0.0</td>\n",
       "      <td>0.0</td>\n",
       "      <td>4</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>82</td>\n",
       "      <td>84</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>Tomato04(22,2)</td>\n",
       "      <td>Eggplant01(22,5)</td>\n",
       "      <td>6</td>\n",
       "      <td>6</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>1</td>\n",
       "      <td>1</td>\n",
       "      <td>0.0</td>\n",
       "      <td>0.0</td>\n",
       "      <td>0.0</td>\n",
       "      <td>0.0</td>\n",
       "      <td>Turnip02(14,15) Turnip00(11,7) Strawberry01(13...</td>\n",
       "      <td>1.0</td>\n",
       "      <td>6.0</td>\n",
       "      <td>6.0</td>\n",
       "      <td>2.0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>3</th>\n",
       "      <td>4ISIiFA1J99JScpurple</td>\n",
       "      <td>4ISIiFA1J99JSc</td>\n",
       "      <td>7</td>\n",
       "      <td>0</td>\n",
       "      <td>high</td>\n",
       "      <td>even</td>\n",
       "      <td>full</td>\n",
       "      <td>True</td>\n",
       "      <td>12.0</td>\n",
       "      <td>Items04</td>\n",
       "      <td>targetPicked</td>\n",
       "      <td>3</td>\n",
       "      <td>purple</td>\n",
       "      <td>Turnip01</td>\n",
       "      <td>1678818034933</td>\n",
       "      <td>1768.0</td>\n",
       "      <td>1678818036701</td>\n",
       "      <td>1678818036701</td>\n",
       "      <td>9</td>\n",
       "      <td>13</td>\n",
       "      <td>5</td>\n",
       "      <td>10</td>\n",
       "      <td>Tomato04(22,2) Turnip02(14,15) Turnip00(11,7) ...</td>\n",
       "      <td>NaN</td>\n",
       "      <td>Eggplant01(22,5)</td>\n",
       "      <td>Tomato04(22,2) Strawberry02(23,2)</td>\n",
       "      <td>False</td>\n",
       "      <td>purple_none(5,10) box(16,5) Turnip02(14,15) Tu...</td>\n",
       "      <td>4</td>\n",
       "      <td>84</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>4</td>\n",
       "      <td>80</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>False</td>\n",
       "      <td>2.0</td>\n",
       "      <td>purple</td>\n",
       "      <td>purpleVeg</td>\n",
       "      <td>ownVeg</td>\n",
       "      <td>0.0</td>\n",
       "      <td>4</td>\n",
       "      <td>1.0</td>\n",
       "      <td>0.0</td>\n",
       "      <td>1.0</td>\n",
       "      <td>0.0</td>\n",
       "      <td>0</td>\n",
       "      <td>False</td>\n",
       "      <td>False</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>False</td>\n",
       "      <td>False</td>\n",
       "      <td>0.0</td>\n",
       "      <td>0.0</td>\n",
       "      <td>0.0</td>\n",
       "      <td>0.0</td>\n",
       "      <td>4</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>84</td>\n",
       "      <td>80</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>Eggplant01(22,5)</td>\n",
       "      <td>Tomato04(22,2) Strawberry02(23,2)</td>\n",
       "      <td>6</td>\n",
       "      <td>5</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>2</td>\n",
       "      <td>1</td>\n",
       "      <td>2</td>\n",
       "      <td>0.0</td>\n",
       "      <td>0.0</td>\n",
       "      <td>0.0</td>\n",
       "      <td>0.0</td>\n",
       "      <td>Turnip02(14,15) Turnip00(11,7) Strawberry01(13...</td>\n",
       "      <td>1.0</td>\n",
       "      <td>5.0</td>\n",
       "      <td>6.0</td>\n",
       "      <td>6.0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>4</th>\n",
       "      <td>4ISIiFA1J99JScred</td>\n",
       "      <td>4ISIiFA1J99JSc</td>\n",
       "      <td>9</td>\n",
       "      <td>0</td>\n",
       "      <td>high</td>\n",
       "      <td>even</td>\n",
       "      <td>full</td>\n",
       "      <td>True</td>\n",
       "      <td>12.0</td>\n",
       "      <td>Items04</td>\n",
       "      <td>targetPicked</td>\n",
       "      <td>4</td>\n",
       "      <td>red</td>\n",
       "      <td>Strawberry01</td>\n",
       "      <td>1678818038081</td>\n",
       "      <td>1127.0</td>\n",
       "      <td>1678818039208</td>\n",
       "      <td>1678818039209</td>\n",
       "      <td>9</td>\n",
       "      <td>14</td>\n",
       "      <td>5</td>\n",
       "      <td>10</td>\n",
       "      <td>Tomato04(22,2) Turnip02(14,15) Turnip00(11,7) ...</td>\n",
       "      <td>NaN</td>\n",
       "      <td>Eggplant01(22,5) Turnip01(23,5)</td>\n",
       "      <td>Tomato04(22,2) Strawberry02(23,2)</td>\n",
       "      <td>False</td>\n",
       "      <td>red_none(9,14) box(16,5) Turnip02(14,15) Turni...</td>\n",
       "      <td>4</td>\n",
       "      <td>82</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>4</td>\n",
       "      <td>80</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>False</td>\n",
       "      <td>14.0</td>\n",
       "      <td>red</td>\n",
       "      <td>redVeg</td>\n",
       "      <td>ownVeg</td>\n",
       "      <td>0.0</td>\n",
       "      <td>4</td>\n",
       "      <td>7.0</td>\n",
       "      <td>0.0</td>\n",
       "      <td>7.0</td>\n",
       "      <td>0.0</td>\n",
       "      <td>0</td>\n",
       "      <td>False</td>\n",
       "      <td>True</td>\n",
       "      <td>7</td>\n",
       "      <td>0</td>\n",
       "      <td>False</td>\n",
       "      <td>False</td>\n",
       "      <td>0.0</td>\n",
       "      <td>0.0</td>\n",
       "      <td>0.0</td>\n",
       "      <td>0.0</td>\n",
       "      <td>4</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>80</td>\n",
       "      <td>82</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>Tomato04(22,2) Strawberry02(23,2)</td>\n",
       "      <td>Eggplant01(22,5) Turnip01(23,5)</td>\n",
       "      <td>5</td>\n",
       "      <td>5</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>2</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>2</td>\n",
       "      <td>2</td>\n",
       "      <td>2</td>\n",
       "      <td>0.0</td>\n",
       "      <td>0.0</td>\n",
       "      <td>0.0</td>\n",
       "      <td>0.0</td>\n",
       "      <td>Turnip02(14,15) Turnip00(11,7) Strawberry01(13...</td>\n",
       "      <td>7.0</td>\n",
       "      <td>5.0</td>\n",
       "      <td>5.0</td>\n",
       "      <td>7.0</td>\n",
       "    </tr>\n",
       "  </tbody>\n",
       "</table>\n",
       "</div>"
      ],
      "text/plain": [
       "                 subjid         session  trialNum  gameNum costCond  \\\n",
       "0     4ISIiFA1J99JScred  4ISIiFA1J99JSc         1        0     high   \n",
       "1  4ISIiFA1J99JScpurple  4ISIiFA1J99JSc         3        0     high   \n",
       "2     4ISIiFA1J99JScred  4ISIiFA1J99JSc         5        0     high   \n",
       "3  4ISIiFA1J99JScpurple  4ISIiFA1J99JSc         7        0     high   \n",
       "4     4ISIiFA1J99JScred  4ISIiFA1J99JSc         9        0     high   \n",
       "\n",
       "  resourceCond visibilityCond  redFirst  counterbalance objectLayer  \\\n",
       "0         even           full      True            12.0     Items04   \n",
       "1         even           full      True            12.0     Items04   \n",
       "2         even           full      True            12.0     Items04   \n",
       "3         even           full      True            12.0     Items04   \n",
       "4         even           full      True            12.0     Items04   \n",
       "\n",
       "      eventName  turnCount   agent        target  turnStartTimestamp  \\\n",
       "0  targetPicked          0     red      Tomato04       1678818021993   \n",
       "1  targetPicked          1  purple    Eggplant01       1678818026786   \n",
       "2  targetPicked          2     red  Strawberry02       1678818033697   \n",
       "3  targetPicked          3  purple      Turnip01       1678818034933   \n",
       "4  targetPicked          4     red  Strawberry01       1678818038081   \n",
       "\n",
       "   responseTime  decisionMadeTimestamp  dataSavedTimestamp  redXloc  redYloc  \\\n",
       "0        3356.0          1678818025349       1678818025349        2       15   \n",
       "1        3786.0          1678818030572       1678818030573        9       14   \n",
       "2        1826.0          1678818035523       1678818035523        9       13   \n",
       "3        1768.0          1678818036701       1678818036701        9       13   \n",
       "4        1127.0          1678818039208       1678818039209        9       14   \n",
       "\n",
       "   purpleXloc  purpleYloc                                          farmItems  \\\n",
       "0           3          16  Tomato04(9,13) Turnip02(14,15) Turnip00(11,7) ...   \n",
       "1           3          16  Tomato04(22,2) Turnip02(14,15) Turnip00(11,7) ...   \n",
       "2           5          11  Tomato04(22,2) Turnip02(14,15) Turnip00(11,7) ...   \n",
       "3           5          10  Tomato04(22,2) Turnip02(14,15) Turnip00(11,7) ...   \n",
       "4           5          10  Tomato04(22,2) Turnip02(14,15) Turnip00(11,7) ...   \n",
       "\n",
       "  farmBox                   purpleBackpack                        redBackpack  \\\n",
       "0     NaN                              NaN                                NaN   \n",
       "1     NaN                              NaN                     Tomato04(22,2)   \n",
       "2     NaN                 Eggplant01(22,5)                     Tomato04(22,2)   \n",
       "3     NaN                 Eggplant01(22,5)  Tomato04(22,2) Strawberry02(23,2)   \n",
       "4     NaN  Eggplant01(22,5) Turnip01(23,5)  Tomato04(22,2) Strawberry02(23,2)   \n",
       "\n",
       "   gameover                                         legalMoves  \\\n",
       "0     False  red_none(2,15) Tomato04(9,13) Turnip02(14,15) ...   \n",
       "1     False  purple_none(3,16) Turnip02(14,15) Turnip00(11,...   \n",
       "2     False  red_none(9,13) box(16,5) Turnip02(14,15) Turni...   \n",
       "3     False  purple_none(5,10) box(16,5) Turnip02(14,15) Tu...   \n",
       "4     False  red_none(9,14) box(16,5) Turnip02(14,15) Turni...   \n",
       "\n",
       "   purpleBackpackSize  purpleEnergy  purpleScore  purplePoints  \\\n",
       "0                   4           100            0             0   \n",
       "1                   4           100            0             0   \n",
       "2                   4            84            0             0   \n",
       "3                   4            84            0             0   \n",
       "4                   4            82            0             0   \n",
       "\n",
       "   purplePointsCumulative  redBackpackSize  redEnergy  redScore  redPoints  \\\n",
       "0                       0                4        100         0          0   \n",
       "1                       0                4         82         0          0   \n",
       "2                       0                4         82         0          0   \n",
       "3                       0                4         80         0          0   \n",
       "4                       0                4         80         0          0   \n",
       "\n",
       "   redPointsCumulative  lastTrial  energy_expense targetColor  targetCat  \\\n",
       "0                    0      False            18.0         red     redVeg   \n",
       "1                    0      False            16.0      purple  purpleVeg   \n",
       "2                    0      False             2.0         red     redVeg   \n",
       "3                    0      False             2.0      purple  purpleVeg   \n",
       "4                    0      False            14.0         red     redVeg   \n",
       "\n",
       "  actionCat  other_expense  ownBPsize  step_expense  step_other_expense  \\\n",
       "0    ownVeg            0.0          4           9.0                 0.0   \n",
       "1    ownVeg            0.0          4           8.0                 0.0   \n",
       "2    ownVeg            0.0          4           1.0                 0.0   \n",
       "3    ownVeg            0.0          4           1.0                 0.0   \n",
       "4    ownVeg            0.0          4           7.0                 0.0   \n",
       "\n",
       "   step_own_expense  step_box_expense  helping_event  helped  meFirst  steps  \\\n",
       "0               9.0               0.0              0   False     True      9   \n",
       "1               8.0               0.0              0   False    False      8   \n",
       "2               1.0               0.0              0   False     True      1   \n",
       "3               1.0               0.0              0   False    False      1   \n",
       "4               7.0               0.0              0   False     True      7   \n",
       "\n",
       "   helpful_steps player_helped_lasttrial partner_helped_lasttrial  \\\n",
       "0              0                     NaN                      NaN   \n",
       "1              0                     NaN                    False   \n",
       "2              0                   False                    False   \n",
       "3              0                   False                    False   \n",
       "4              0                   False                    False   \n",
       "\n",
       "   player_helpingevents_lasttrial  partner_helpingevents_lasttrial  \\\n",
       "0                             NaN                              NaN   \n",
       "1                             NaN                              0.0   \n",
       "2                             0.0                              0.0   \n",
       "3                             0.0                              0.0   \n",
       "4                             0.0                              0.0   \n",
       "\n",
       "   player_helpfulsteps_lasttrial  partner_helpfulsteps_lasttrial  otherBPsize  \\\n",
       "0                            NaN                             NaN            4   \n",
       "1                            NaN                             0.0            4   \n",
       "2                            0.0                             0.0            4   \n",
       "3                            0.0                             0.0            4   \n",
       "4                            0.0                             0.0            4   \n",
       "\n",
       "   ownScore  otherScore  ownEnergy  otherEnergy  ownPoints  otherPoints  \\\n",
       "0         0           0        100          100          0            0   \n",
       "1         0           0        100           82          0            0   \n",
       "2         0           0         82           84          0            0   \n",
       "3         0           0         84           80          0            0   \n",
       "4         0           0         80           82          0            0   \n",
       "\n",
       "   ownPointsCumulative  otherPointsCumulative  \\\n",
       "0                    0                      0   \n",
       "1                    0                      0   \n",
       "2                    0                      0   \n",
       "3                    0                      0   \n",
       "4                    0                      0   \n",
       "\n",
       "                         ownBackpack                      otherBackpack  \\\n",
       "0                                NaN                                NaN   \n",
       "1                                NaN                     Tomato04(22,2)   \n",
       "2                     Tomato04(22,2)                   Eggplant01(22,5)   \n",
       "3                   Eggplant01(22,5)  Tomato04(22,2) Strawberry02(23,2)   \n",
       "4  Tomato04(22,2) Strawberry02(23,2)    Eggplant01(22,5) Turnip01(23,5)   \n",
       "\n",
       "   nOwnVegFarm  nOtherVegFarm  nOwnVegBox  nOtherVegBox  nOwnVegInOwnBP  \\\n",
       "0            7              7           0             0               0   \n",
       "1            7              6           0             0               0   \n",
       "2            6              6           0             0               1   \n",
       "3            6              5           0             0               1   \n",
       "4            5              5           0             0               2   \n",
       "\n",
       "   nOwnVegInOtherBP  nOtherVegInOwnBP  nOtherVegInOtherBP  nAnyVegInOwnBP  \\\n",
       "0                 0                 0                   0               0   \n",
       "1                 0                 0                   1               0   \n",
       "2                 0                 0                   1               1   \n",
       "3                 0                 0                   2               1   \n",
       "4                 0                 0                   2               2   \n",
       "\n",
       "   nAnyVegInOtherBP  ownHelpingEventsToDate  otherHelpingEventsToDate  \\\n",
       "0                 0                     NaN                       NaN   \n",
       "1                 1                     NaN                       0.0   \n",
       "2                 1                     0.0                       0.0   \n",
       "3                 2                     0.0                       0.0   \n",
       "4                 2                     0.0                       0.0   \n",
       "\n",
       "   ownHelpfulStepsToDate  otherHelpingStepsToDate  \\\n",
       "0                    NaN                      NaN   \n",
       "1                    NaN                      0.0   \n",
       "2                    0.0                      0.0   \n",
       "3                    0.0                      0.0   \n",
       "4                    0.0                      0.0   \n",
       "\n",
       "                                        farmContents  \\\n",
       "0  Tomato04(9,13) Turnip02(14,15) Turnip00(11,7) ...   \n",
       "1  Turnip02(14,15) Turnip00(11,7) Strawberry01(13...   \n",
       "2  Turnip02(14,15) Turnip00(11,7) Strawberry01(13...   \n",
       "3  Turnip02(14,15) Turnip00(11,7) Strawberry01(13...   \n",
       "4  Turnip02(14,15) Turnip00(11,7) Strawberry01(13...   \n",
       "\n",
       "   ownDistanceToClosestOwnVeg  ownDistanceToClosestOtherVeg  \\\n",
       "0                         8.0                           8.0   \n",
       "1                         8.0                           8.0   \n",
       "2                         1.0                           6.0   \n",
       "3                         1.0                           5.0   \n",
       "4                         7.0                           5.0   \n",
       "\n",
       "   otherDistanceToClosestOwnVeg  otherDistanceToClosestOtherVeg  \n",
       "0                           8.0                             8.0  \n",
       "1                           5.0                             0.0  \n",
       "2                           6.0                             2.0  \n",
       "3                           6.0                             6.0  \n",
       "4                           5.0                             7.0  "
      ]
     },
     "execution_count": 6,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "# Import data\n",
    "# dir_path  = os.getcwd()\n",
    "# change relative path depending on where you're running this script from\n",
    "df = pd.read_csv('data/trialdf.csv')\n",
    "df.head()\n",
    "\n",
    "# to view the transcript of a single game from subject id and game number\n",
    "# sesh = \"4ISIiFA1J99JSc\"\n",
    "# game = 1\n",
    "# amap.print_transcript(df, sesh, game)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "metadata": {},
   "outputs": [],
   "source": []
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "metadata": {},
   "outputs": [],
   "source": []
  }
 ],
 "metadata": {
  "kernelspec": {
   "display_name": ".venv",
   "language": "python",
   "name": "python3"
  },
  "language_info": {
   "codemirror_mode": {
    "name": "ipython",
    "version": 3
   },
   "file_extension": ".py",
   "mimetype": "text/x-python",
   "name": "python",
   "nbconvert_exporter": "python",
   "pygments_lexer": "ipython3",
   "version": "3.12.3"
  }
 },
 "nbformat": 4,
 "nbformat_minor": 2
}