diff --git a/codigo/decision-tree.ipynb b/codigo/decision-tree.ipynb index 8f8028e..3d8b375 100644 --- a/codigo/decision-tree.ipynb +++ b/codigo/decision-tree.ipynb @@ -2,16 +2,16 @@ "cells": [ { "cell_type": "code", - "execution_count": 4, + "execution_count": 146, "metadata": {}, "outputs": [], "source": [ - "import pandas as pd" + "import pandas as pd # type: ignore" ] }, { "cell_type": "code", - "execution_count": 5, + "execution_count": 148, "metadata": {}, "outputs": [ { @@ -36,58 +36,148 @@ " \n", " \n", " id\n", + " occasion\n", + " elapsed.days\n", " intervention\n", - " sex\n", - " age\n", - " educ\n", - " income\n", + " ahi01\n", + " ahi02\n", + " ahi03\n", + " ahi04\n", + " ahi05\n", + " ahi06\n", + " ...\n", + " cesd13\n", + " cesd14\n", + " cesd15\n", + " cesd16\n", + " cesd17\n", + " cesd18\n", + " cesd19\n", + " cesd20\n", + " ahiTotal\n", + " cesdTotal\n", " \n", " \n", " \n", " \n", " 0\n", " 1\n", + " 0\n", + " 0.000000\n", " 4\n", " 2\n", - " 35\n", - " 5\n", " 3\n", + " 2\n", + " 3\n", + " 3\n", + " 2\n", + " ...\n", + " 2\n", + " 3\n", + " 1\n", + " 2\n", + " 1\n", + " 1\n", + " 2\n", + " 2\n", + " 63\n", + " 14\n", " \n", " \n", " 1\n", + " 1\n", + " 1\n", + " 11.772731\n", + " 4\n", + " 3\n", + " 3\n", + " 4\n", + " 3\n", + " 3\n", + " 4\n", + " ...\n", + " 1\n", " 2\n", " 1\n", + " 3\n", + " 1\n", " 1\n", - " 59\n", " 1\n", " 1\n", + " 73\n", + " 6\n", " \n", " \n", " 2\n", + " 2\n", + " 0\n", + " 0.000000\n", + " 1\n", " 3\n", " 4\n", - " 1\n", - " 51\n", + " 3\n", " 4\n", + " 2\n", " 3\n", + " ...\n", + " 1\n", + " 1\n", + " 1\n", + " 4\n", + " 1\n", + " 1\n", + " 1\n", + " 1\n", + " 73\n", + " 7\n", " \n", " \n", " 3\n", + " 2\n", + " 1\n", + " 8.017523\n", + " 1\n", + " 3\n", + " 4\n", " 4\n", + " 4\n", + " 3\n", " 3\n", + " ...\n", " 1\n", - " 50\n", - " 5\n", - " 2\n", + " 1\n", + " 1\n", + " 3\n", + " 1\n", + " 1\n", + " 1\n", + " 1\n", + " 89\n", + " 10\n", " \n", " \n", " 4\n", - " 5\n", " 2\n", " 2\n", - " 58\n", - " 5\n", - " 2\n", + " 14.304086\n", + " 1\n", + " 3\n", + " 4\n", + " 4\n", + " 4\n", + " 3\n", + " 3\n", + " ...\n", + " 3\n", + " 1\n", + " 1\n", + " 1\n", + " 1\n", + " 1\n", + " 1\n", + " 1\n", + " 89\n", + " 13\n", " \n", " \n", " ...\n", @@ -97,82 +187,5639 @@ " ...\n", " ...\n", " ...\n", + " ...\n", + " ...\n", + " ...\n", + " ...\n", + " ...\n", + " ...\n", + " ...\n", + " ...\n", + " ...\n", + " ...\n", + " ...\n", + " ...\n", + " ...\n", + " ...\n", + " ...\n", " \n", " \n", - " 290\n", - " 291\n", + " 987\n", + " 292\n", + " 0\n", + " 0.000000\n", + " 4\n", + " 2\n", + " 2\n", + " 2\n", + " 2\n", + " 2\n", + " 4\n", + " ...\n", + " 2\n", " 2\n", " 1\n", - " 39\n", - " 5\n", " 3\n", + " 2\n", + " 3\n", + " 1\n", + " 4\n", + " 50\n", + " 30\n", " \n", " \n", - " 291\n", + " 988\n", " 292\n", - " 4\n", " 1\n", - " 47\n", + " 9.940718\n", + " 4\n", + " 4\n", + " 3\n", + " 2\n", + " 3\n", + " 3\n", " 4\n", + " ...\n", " 2\n", + " 1\n", + " 1\n", + " 4\n", + " 1\n", + " 1\n", + " 1\n", + " 3\n", + " 70\n", + " 6\n", " \n", " \n", - " 292\n", + " 989\n", " 293\n", + " 0\n", + " 0.000000\n", " 2\n", " 2\n", - " 52\n", - " 4\n", + " 2\n", + " 2\n", + " 2\n", + " 2\n", + " 1\n", + " ...\n", + " 1\n", + " 3\n", + " 1\n", + " 2\n", + " 1\n", + " 2\n", + " 1\n", " 3\n", + " 47\n", + " 18\n", " \n", " \n", - " 293\n", + " 990\n", " 294\n", + " 0\n", + " 0.000000\n", " 3\n", - " 1\n", - " 59\n", - " 5\n", + " 3\n", + " 3\n", + " 2\n", + " 3\n", + " 3\n", + " 3\n", + " ...\n", " 2\n", + " 2\n", + " 1\n", + " 3\n", + " 1\n", + " 1\n", + " 1\n", + " 1\n", + " 70\n", + " 7\n", " \n", " \n", - " 294\n", + " 991\n", " 295\n", + " 0\n", + " 0.000000\n", + " 1\n", + " 3\n", + " 3\n", + " 2\n", + " 3\n", + " 2\n", + " 3\n", + " ...\n", + " 1\n", " 1\n", " 1\n", - " 28\n", " 4\n", + " 1\n", " 2\n", + " 1\n", + " 1\n", + " 65\n", + " 6\n", " \n", " \n", "\n", - "

295 rows × 6 columns

\n", + "

992 rows × 50 columns

\n", "" ], "text/plain": [ - " id intervention sex age educ income\n", - "0 1 4 2 35 5 3\n", - "1 2 1 1 59 1 1\n", - "2 3 4 1 51 4 3\n", - "3 4 3 1 50 5 2\n", - "4 5 2 2 58 5 2\n", - ".. ... ... ... ... ... ...\n", - "290 291 2 1 39 5 3\n", - "291 292 4 1 47 4 2\n", - "292 293 2 2 52 4 3\n", - "293 294 3 1 59 5 2\n", - "294 295 1 1 28 4 2\n", + " id occasion elapsed.days intervention ahi01 ahi02 ahi03 ahi04 \\\n", + "0 1 0 0.000000 4 2 3 2 3 \n", + "1 1 1 11.772731 4 3 3 4 3 \n", + "2 2 0 0.000000 1 3 4 3 4 \n", + "3 2 1 8.017523 1 3 4 4 4 \n", + "4 2 2 14.304086 1 3 4 4 4 \n", + ".. ... ... ... ... ... ... ... ... \n", + "987 292 0 0.000000 4 2 2 2 2 \n", + "988 292 1 9.940718 4 4 3 2 3 \n", + "989 293 0 0.000000 2 2 2 2 2 \n", + "990 294 0 0.000000 3 3 3 2 3 \n", + "991 295 0 0.000000 1 3 3 2 3 \n", "\n", - "[295 rows x 6 columns]" + " ahi05 ahi06 ... cesd13 cesd14 cesd15 cesd16 cesd17 cesd18 \\\n", + "0 3 2 ... 2 3 1 2 1 1 \n", + "1 3 4 ... 1 2 1 3 1 1 \n", + "2 2 3 ... 1 1 1 4 1 1 \n", + "3 3 3 ... 1 1 1 3 1 1 \n", + "4 3 3 ... 3 1 1 1 1 1 \n", + ".. ... ... ... ... ... ... ... ... ... \n", + "987 2 4 ... 2 2 1 3 2 3 \n", + "988 3 4 ... 2 1 1 4 1 1 \n", + "989 2 1 ... 1 3 1 2 1 2 \n", + "990 3 3 ... 2 2 1 3 1 1 \n", + "991 2 3 ... 1 1 1 4 1 2 \n", + "\n", + " cesd19 cesd20 ahiTotal cesdTotal \n", + "0 2 2 63 14 \n", + "1 1 1 73 6 \n", + "2 1 1 73 7 \n", + "3 1 1 89 10 \n", + "4 1 1 89 13 \n", + ".. ... ... ... ... \n", + "987 1 4 50 30 \n", + "988 1 3 70 6 \n", + "989 1 3 47 18 \n", + "990 1 1 70 7 \n", + "991 1 1 65 6 \n", + "\n", + "[992 rows x 50 columns]" + ] + }, + "execution_count": 148, + "metadata": {}, + "output_type": "execute_result" + } + ], + "source": [ + "measurements_all_ocassions = pd.read_csv('dados/ahi-cesd.csv')\n", + "measurements_all_ocassions" + ] + }, + { + "cell_type": "code", + "execution_count": 149, + "metadata": {}, + "outputs": [ + { + "data": { + "text/html": [ + "
\n", + "\n", + "\n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + "
idoccasionelapsed.daysinterventionahi01ahi02ahi03ahi04ahi05ahi06...cesd13cesd14cesd15cesd16cesd17cesd18cesd19cesd20ahiTotalcesdTotal
0100.04232332...231211226314
2200.01343423...11141111737
8300.04332423...12141111773
10400.03223223...441233116031
14500.02123222...231212134127
..................................................................
98629100.02233434...11241122809
98729200.04222224...221323145030
98929300.02222221...131212134718
99029400.03332333...22131111707
99129500.01332323...11141211656
\n", + "

295 rows × 50 columns

\n", + "
" + ], + "text/plain": [ + " id occasion elapsed.days intervention ahi01 ahi02 ahi03 ahi04 \\\n", + "0 1 0 0.0 4 2 3 2 3 \n", + "2 2 0 0.0 1 3 4 3 4 \n", + "8 3 0 0.0 4 3 3 2 4 \n", + "10 4 0 0.0 3 2 2 3 2 \n", + "14 5 0 0.0 2 1 2 3 2 \n", + ".. ... ... ... ... ... ... ... ... \n", + "986 291 0 0.0 2 2 3 3 4 \n", + "987 292 0 0.0 4 2 2 2 2 \n", + "989 293 0 0.0 2 2 2 2 2 \n", + "990 294 0 0.0 3 3 3 2 3 \n", + "991 295 0 0.0 1 3 3 2 3 \n", + "\n", + " ahi05 ahi06 ... cesd13 cesd14 cesd15 cesd16 cesd17 cesd18 \\\n", + "0 3 2 ... 2 3 1 2 1 1 \n", + "2 2 3 ... 1 1 1 4 1 1 \n", + "8 2 3 ... 1 2 1 4 1 1 \n", + "10 2 3 ... 4 4 1 2 3 3 \n", + "14 2 2 ... 2 3 1 2 1 2 \n", + ".. ... ... ... ... ... ... ... ... ... \n", + "986 3 4 ... 1 1 2 4 1 1 \n", + "987 2 4 ... 2 2 1 3 2 3 \n", + "989 2 1 ... 1 3 1 2 1 2 \n", + "990 3 3 ... 2 2 1 3 1 1 \n", + "991 2 3 ... 1 1 1 4 1 2 \n", + "\n", + " cesd19 cesd20 ahiTotal cesdTotal \n", + "0 2 2 63 14 \n", + "2 1 1 73 7 \n", + "8 1 1 77 3 \n", + "10 1 1 60 31 \n", + "14 1 3 41 27 \n", + ".. ... ... ... ... \n", + "986 2 2 80 9 \n", + "987 1 4 50 30 \n", + "989 1 3 47 18 \n", + "990 1 1 70 7 \n", + "991 1 1 65 6 \n", + "\n", + "[295 rows x 50 columns]" + ] + }, + "execution_count": 149, + "metadata": {}, + "output_type": "execute_result" + } + ], + "source": [ + "measurements_pre_intervention = measurements_all_ocassions.query('occasion == 0')\n", + "measurements_pre_intervention" + ] + }, + { + "cell_type": "code", + "execution_count": 150, + "metadata": {}, + "outputs": [ + { + "data": { + "text/html": [ + "
\n", + "\n", + "\n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + "
occasionelapsed.daysinterventionahi01ahi02ahi03ahi04ahi05ahi06ahi07...cesd13cesd14cesd15cesd16cesd17cesd18cesd19cesd20ahiTotalcesdTotal
100.042323323...231211226314
200.013434234...11141111737
300.043324234...12141111773
400.032232233...441233116031
500.021232221...231212134127
..................................................................
29100.022334344...11241122809
29200.042222241...221323145030
29300.022222212...131212134718
29400.033323333...22131111707
29500.013323233...11141211656
\n", + "

295 rows × 49 columns

\n", + "
" + ], + "text/plain": [ + " occasion elapsed.days intervention ahi01 ahi02 ahi03 ahi04 ahi05 \\\n", + "1 0 0.0 4 2 3 2 3 3 \n", + "2 0 0.0 1 3 4 3 4 2 \n", + "3 0 0.0 4 3 3 2 4 2 \n", + "4 0 0.0 3 2 2 3 2 2 \n", + "5 0 0.0 2 1 2 3 2 2 \n", + ".. ... ... ... ... ... ... ... ... \n", + "291 0 0.0 2 2 3 3 4 3 \n", + "292 0 0.0 4 2 2 2 2 2 \n", + "293 0 0.0 2 2 2 2 2 2 \n", + "294 0 0.0 3 3 3 2 3 3 \n", + "295 0 0.0 1 3 3 2 3 2 \n", + "\n", + " ahi06 ahi07 ... cesd13 cesd14 cesd15 cesd16 cesd17 cesd18 \\\n", + "1 2 3 ... 2 3 1 2 1 1 \n", + "2 3 4 ... 1 1 1 4 1 1 \n", + "3 3 4 ... 1 2 1 4 1 1 \n", + "4 3 3 ... 4 4 1 2 3 3 \n", + "5 2 1 ... 2 3 1 2 1 2 \n", + ".. ... ... ... ... ... ... ... ... ... \n", + "291 4 4 ... 1 1 2 4 1 1 \n", + "292 4 1 ... 2 2 1 3 2 3 \n", + "293 1 2 ... 1 3 1 2 1 2 \n", + "294 3 3 ... 2 2 1 3 1 1 \n", + "295 3 3 ... 1 1 1 4 1 2 \n", + "\n", + " cesd19 cesd20 ahiTotal cesdTotal \n", + "1 2 2 63 14 \n", + "2 1 1 73 7 \n", + "3 1 1 77 3 \n", + "4 1 1 60 31 \n", + "5 1 3 41 27 \n", + ".. ... ... ... ... \n", + "291 2 2 80 9 \n", + "292 1 4 50 30 \n", + "293 1 3 47 18 \n", + "294 1 1 70 7 \n", + "295 1 1 65 6 \n", + "\n", + "[295 rows x 49 columns]" + ] + }, + "execution_count": 150, + "metadata": {}, + "output_type": "execute_result" + } + ], + "source": [ + "measurements_pre_intervention = measurements_pre_intervention.set_index('id')\n", + "\n", + "measurements_pre_intervention = \\\n", + " measurements_pre_intervention.rename_axis(index=None)\n", + "\n", + "measurements_pre_intervention" + ] + }, + { + "cell_type": "code", + "execution_count": 151, + "metadata": {}, + "outputs": [ + { + "data": { + "text/html": [ + "
\n", + "\n", + "\n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + "
idoccasionelapsed.daysinterventionahi01ahi02ahi03ahi04ahi05ahi06...cesd13cesd14cesd15cesd16cesd17cesd18cesd19cesd20ahiTotalcesdTotal
11111.7727314334334...12131111736
3218.0175231344433...111311118910
11417.4194563323423...341244136731
15517.7488082123222...332132234132
24818.8007642221221...242114344545
..................................................................
95528117.5191324333433...11141111761
96128218.0084951232244...111443246317
96728317.0948152432443...311311228512
976287110.0080792333434...11142212769
98829219.9407184432334...21141113706
\n", + "

147 rows × 50 columns

\n", + "
" + ], + "text/plain": [ + " id occasion elapsed.days intervention ahi01 ahi02 ahi03 ahi04 \\\n", + "1 1 1 11.772731 4 3 3 4 3 \n", + "3 2 1 8.017523 1 3 4 4 4 \n", + "11 4 1 7.419456 3 3 2 3 4 \n", + "15 5 1 7.748808 2 1 2 3 2 \n", + "24 8 1 8.800764 2 2 2 1 2 \n", + ".. ... ... ... ... ... ... ... ... \n", + "955 281 1 7.519132 4 3 3 3 4 \n", + "961 282 1 8.008495 1 2 3 2 2 \n", + "967 283 1 7.094815 2 4 3 2 4 \n", + "976 287 1 10.008079 2 3 3 3 4 \n", + "988 292 1 9.940718 4 4 3 2 3 \n", + "\n", + " ahi05 ahi06 ... cesd13 cesd14 cesd15 cesd16 cesd17 cesd18 \\\n", + "1 3 4 ... 1 2 1 3 1 1 \n", + "3 3 3 ... 1 1 1 3 1 1 \n", + "11 2 3 ... 3 4 1 2 4 4 \n", + "15 2 2 ... 3 3 2 1 3 2 \n", + "24 2 1 ... 2 4 2 1 1 4 \n", + ".. ... ... ... ... ... ... ... ... ... \n", + "955 3 3 ... 1 1 1 4 1 1 \n", + "961 4 4 ... 1 1 1 4 4 3 \n", + "967 4 3 ... 3 1 1 3 1 1 \n", + "976 3 4 ... 1 1 1 4 2 2 \n", + "988 3 4 ... 2 1 1 4 1 1 \n", + "\n", + " cesd19 cesd20 ahiTotal cesdTotal \n", + "1 1 1 73 6 \n", + "3 1 1 89 10 \n", + "11 1 3 67 31 \n", + "15 2 3 41 32 \n", + "24 3 4 45 45 \n", + ".. ... ... ... ... \n", + "955 1 1 76 1 \n", + "961 2 4 63 17 \n", + "967 2 2 85 12 \n", + "976 1 2 76 9 \n", + "988 1 3 70 6 \n", + "\n", + "[147 rows x 50 columns]" + ] + }, + "execution_count": 151, + "metadata": {}, + "output_type": "execute_result" + } + ], + "source": [ + "measurements_post_intervention = measurements_all_ocassions.query('occasion == 1')\n", + "measurements_post_intervention" + ] + }, + { + "cell_type": "code", + "execution_count": 152, + "metadata": {}, + "outputs": [ + { + "data": { + "text/html": [ + "
\n", + "\n", + "\n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + "
occasionelapsed.daysinterventionahi01ahi02ahi03ahi04ahi05ahi06ahi07...cesd13cesd14cesd15cesd16cesd17cesd18cesd19cesd20ahiTotalcesdTotal
1111.77273143343344...12131111736
218.01752313444334...111311118910
417.41945633234233...341244136731
517.74880821232221...332132234132
818.80076422212213...242114344545
..................................................................
28117.51913243334334...11141111761
28218.00849512322443...111443246317
28317.09481524324434...311311228512
287110.00807923334344...11142212769
29219.94071844323342...21141113706
\n", + "

147 rows × 49 columns

\n", + "
" + ], + "text/plain": [ + " occasion elapsed.days intervention ahi01 ahi02 ahi03 ahi04 ahi05 \\\n", + "1 1 11.772731 4 3 3 4 3 3 \n", + "2 1 8.017523 1 3 4 4 4 3 \n", + "4 1 7.419456 3 3 2 3 4 2 \n", + "5 1 7.748808 2 1 2 3 2 2 \n", + "8 1 8.800764 2 2 2 1 2 2 \n", + ".. ... ... ... ... ... ... ... ... \n", + "281 1 7.519132 4 3 3 3 4 3 \n", + "282 1 8.008495 1 2 3 2 2 4 \n", + "283 1 7.094815 2 4 3 2 4 4 \n", + "287 1 10.008079 2 3 3 3 4 3 \n", + "292 1 9.940718 4 4 3 2 3 3 \n", + "\n", + " ahi06 ahi07 ... cesd13 cesd14 cesd15 cesd16 cesd17 cesd18 \\\n", + "1 4 4 ... 1 2 1 3 1 1 \n", + "2 3 4 ... 1 1 1 3 1 1 \n", + "4 3 3 ... 3 4 1 2 4 4 \n", + "5 2 1 ... 3 3 2 1 3 2 \n", + "8 1 3 ... 2 4 2 1 1 4 \n", + ".. ... ... ... ... ... ... ... ... ... \n", + "281 3 4 ... 1 1 1 4 1 1 \n", + "282 4 3 ... 1 1 1 4 4 3 \n", + "283 3 4 ... 3 1 1 3 1 1 \n", + "287 4 4 ... 1 1 1 4 2 2 \n", + "292 4 2 ... 2 1 1 4 1 1 \n", + "\n", + " cesd19 cesd20 ahiTotal cesdTotal \n", + "1 1 1 73 6 \n", + "2 1 1 89 10 \n", + "4 1 3 67 31 \n", + "5 2 3 41 32 \n", + "8 3 4 45 45 \n", + ".. ... ... ... ... \n", + "281 1 1 76 1 \n", + "282 2 4 63 17 \n", + "283 2 2 85 12 \n", + "287 1 2 76 9 \n", + "292 1 3 70 6 \n", + "\n", + "[147 rows x 49 columns]" + ] + }, + "execution_count": 152, + "metadata": {}, + "output_type": "execute_result" + } + ], + "source": [ + "measurements_post_intervention = measurements_post_intervention.set_index('id')\n", + "\n", + "measurements_post_intervention = \\\n", + " measurements_post_intervention.rename_axis(index=None)\n", + "\n", + "measurements_post_intervention" + ] + }, + { + "cell_type": "code", + "execution_count": 153, + "metadata": {}, + "outputs": [ + { + "data": { + "text/html": [ + "
\n", + "\n", + "\n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + "
ahiTotalcesdTotal
1736
28910
46731
54132
84545
.........
281761
2826317
2838512
287769
292706
\n", + "

147 rows × 2 columns

\n", + "
" + ], + "text/plain": [ + " ahiTotal cesdTotal\n", + "1 73 6\n", + "2 89 10\n", + "4 67 31\n", + "5 41 32\n", + "8 45 45\n", + ".. ... ...\n", + "281 76 1\n", + "282 63 17\n", + "283 85 12\n", + "287 76 9\n", + "292 70 6\n", + "\n", + "[147 rows x 2 columns]" + ] + }, + "execution_count": 153, + "metadata": {}, + "output_type": "execute_result" + } + ], + "source": [ + "measurements_post_intervention = measurements_post_intervention[['ahiTotal', 'cesdTotal']]\n", + "measurements_post_intervention" + ] + }, + { + "cell_type": "code", + "execution_count": 154, + "metadata": {}, + "outputs": [ + { + "data": { + "text/html": [ + "
\n", + "\n", + "\n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + "
occasionelapsed.daysinterventionahi01ahi02ahi03ahi04ahi05ahi06ahi07...cesd15cesd16cesd17cesd18cesd19cesd20ahiTotal.precesdTotal.preahiTotal.postcesdTotal.post
100.042323323...1211226314736
200.013434234...1411117378910
400.032232233...12331160316731
500.021232221...12121341274132
800.023222214...12242359304545
..................................................................
28100.043324234...141211694761
28200.012322453...14113366116317
28300.023334444...1411129188512
28700.022313411...1333236430769
29200.042222241...1323145030706
\n", + "

147 rows × 51 columns

\n", + "
" + ], + "text/plain": [ + " occasion elapsed.days intervention ahi01 ahi02 ahi03 ahi04 ahi05 \\\n", + "1 0 0.0 4 2 3 2 3 3 \n", + "2 0 0.0 1 3 4 3 4 2 \n", + "4 0 0.0 3 2 2 3 2 2 \n", + "5 0 0.0 2 1 2 3 2 2 \n", + "8 0 0.0 2 3 2 2 2 2 \n", + ".. ... ... ... ... ... ... ... ... \n", + "281 0 0.0 4 3 3 2 4 2 \n", + "282 0 0.0 1 2 3 2 2 4 \n", + "283 0 0.0 2 3 3 3 4 4 \n", + "287 0 0.0 2 2 3 1 3 4 \n", + "292 0 0.0 4 2 2 2 2 2 \n", + "\n", + " ahi06 ahi07 ... cesd15 cesd16 cesd17 cesd18 cesd19 cesd20 \\\n", + "1 2 3 ... 1 2 1 1 2 2 \n", + "2 3 4 ... 1 4 1 1 1 1 \n", + "4 3 3 ... 1 2 3 3 1 1 \n", + "5 2 1 ... 1 2 1 2 1 3 \n", + "8 1 4 ... 1 2 2 4 2 3 \n", + ".. ... ... ... ... ... ... ... ... ... \n", + "281 3 4 ... 1 4 1 2 1 1 \n", + "282 5 3 ... 1 4 1 1 3 3 \n", + "283 4 4 ... 1 4 1 1 1 2 \n", + "287 1 1 ... 1 3 3 3 2 3 \n", + "292 4 1 ... 1 3 2 3 1 4 \n", + "\n", + " ahiTotal.pre cesdTotal.pre ahiTotal.post cesdTotal.post \n", + "1 63 14 73 6 \n", + "2 73 7 89 10 \n", + "4 60 31 67 31 \n", + "5 41 27 41 32 \n", + "8 59 30 45 45 \n", + ".. ... ... ... ... \n", + "281 69 4 76 1 \n", + "282 66 11 63 17 \n", + "283 91 8 85 12 \n", + "287 64 30 76 9 \n", + "292 50 30 70 6 \n", + "\n", + "[147 rows x 51 columns]" + ] + }, + "execution_count": 154, + "metadata": {}, + "output_type": "execute_result" + } + ], + "source": [ + "measurements_pre_post_intervention = measurements_pre_intervention.join(measurements_post_intervention, how='inner', lsuffix='.pre', rsuffix='.post')\n", + "measurements_pre_post_intervention" + ] + }, + { + "cell_type": "code", + "execution_count": 155, + "metadata": {}, + "outputs": [ + { + "data": { + "text/html": [ + "
\n", + "\n", + "\n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + "
interventionahi01ahi02ahi03ahi04ahi05ahi06ahi07ahi08ahi09...cesd15cesd16cesd17cesd18cesd19cesd20ahiTotal.precesdTotal.preahiTotal.postcesdTotal.post
14232332333...1211226314736
21343423433...1411117378910
43223223315...12331160316731
52123222112...12121341274132
82322221444...12242359304545
..................................................................
2814332423443...141211694761
2821232245324...14113366116317
2832333444434...1411129188512
2872231341134...1333236430769
2924222224124...1323145030706
\n", + "

147 rows × 49 columns

\n", + "
" + ], + "text/plain": [ + " intervention ahi01 ahi02 ahi03 ahi04 ahi05 ahi06 ahi07 ahi08 \\\n", + "1 4 2 3 2 3 3 2 3 3 \n", + "2 1 3 4 3 4 2 3 4 3 \n", + "4 3 2 2 3 2 2 3 3 1 \n", + "5 2 1 2 3 2 2 2 1 1 \n", + "8 2 3 2 2 2 2 1 4 4 \n", + ".. ... ... ... ... ... ... ... ... ... \n", + "281 4 3 3 2 4 2 3 4 4 \n", + "282 1 2 3 2 2 4 5 3 2 \n", + "283 2 3 3 3 4 4 4 4 3 \n", + "287 2 2 3 1 3 4 1 1 3 \n", + "292 4 2 2 2 2 2 4 1 2 \n", + "\n", + " ahi09 ... cesd15 cesd16 cesd17 cesd18 cesd19 cesd20 ahiTotal.pre \\\n", + "1 3 ... 1 2 1 1 2 2 63 \n", + "2 3 ... 1 4 1 1 1 1 73 \n", + "4 5 ... 1 2 3 3 1 1 60 \n", + "5 2 ... 1 2 1 2 1 3 41 \n", + "8 4 ... 1 2 2 4 2 3 59 \n", + ".. ... ... ... ... ... ... ... ... ... \n", + "281 3 ... 1 4 1 2 1 1 69 \n", + "282 4 ... 1 4 1 1 3 3 66 \n", + "283 4 ... 1 4 1 1 1 2 91 \n", + "287 4 ... 1 3 3 3 2 3 64 \n", + "292 4 ... 1 3 2 3 1 4 50 \n", + "\n", + " cesdTotal.pre ahiTotal.post cesdTotal.post \n", + "1 14 73 6 \n", + "2 7 89 10 \n", + "4 31 67 31 \n", + "5 27 41 32 \n", + "8 30 45 45 \n", + ".. ... ... ... \n", + "281 4 76 1 \n", + "282 11 63 17 \n", + "283 8 85 12 \n", + "287 30 76 9 \n", + "292 30 70 6 \n", + "\n", + "[147 rows x 49 columns]" + ] + }, + "execution_count": 155, + "metadata": {}, + "output_type": "execute_result" + } + ], + "source": [ + "measurements_pre_post_intervention = \\\n", + " measurements_pre_post_intervention.drop(columns=[\n", + " 'occasion',\n", + " 'elapsed.days'\n", + " ])\n", + "\n", + "measurements_pre_post_intervention" + ] + }, + { + "cell_type": "code", + "execution_count": 156, + "metadata": {}, + "outputs": [ + { + "data": { + "text/html": [ + "
\n", + "\n", + "\n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + "
interventionahi01ahi02ahi03ahi04ahi05ahi06ahi07ahi08ahi09...cesd16cesd17cesd18cesd19cesd20ahiTotal.precesdTotal.preahiTotal.postcesdTotal.postahiChange
14232332333...21122631473610
21343423433...41111737891016
43223223315...23311603167317
52123222112...21213412741320
82322221444...2242359304545-14
..................................................................
2814332423443...412116947617
2821232245324...4113366116317-3
2832333444434...411129188512-6
2872231341134...33323643076912
2924222224124...32314503070620
\n", + "

147 rows × 50 columns

\n", + "
" + ], + "text/plain": [ + " intervention ahi01 ahi02 ahi03 ahi04 ahi05 ahi06 ahi07 ahi08 \\\n", + "1 4 2 3 2 3 3 2 3 3 \n", + "2 1 3 4 3 4 2 3 4 3 \n", + "4 3 2 2 3 2 2 3 3 1 \n", + "5 2 1 2 3 2 2 2 1 1 \n", + "8 2 3 2 2 2 2 1 4 4 \n", + ".. ... ... ... ... ... ... ... ... ... \n", + "281 4 3 3 2 4 2 3 4 4 \n", + "282 1 2 3 2 2 4 5 3 2 \n", + "283 2 3 3 3 4 4 4 4 3 \n", + "287 2 2 3 1 3 4 1 1 3 \n", + "292 4 2 2 2 2 2 4 1 2 \n", + "\n", + " ahi09 ... cesd16 cesd17 cesd18 cesd19 cesd20 ahiTotal.pre \\\n", + "1 3 ... 2 1 1 2 2 63 \n", + "2 3 ... 4 1 1 1 1 73 \n", + "4 5 ... 2 3 3 1 1 60 \n", + "5 2 ... 2 1 2 1 3 41 \n", + "8 4 ... 2 2 4 2 3 59 \n", + ".. ... ... ... ... ... ... ... ... \n", + "281 3 ... 4 1 2 1 1 69 \n", + "282 4 ... 4 1 1 3 3 66 \n", + "283 4 ... 4 1 1 1 2 91 \n", + "287 4 ... 3 3 3 2 3 64 \n", + "292 4 ... 3 2 3 1 4 50 \n", + "\n", + " cesdTotal.pre ahiTotal.post cesdTotal.post ahiChange \n", + "1 14 73 6 10 \n", + "2 7 89 10 16 \n", + "4 31 67 31 7 \n", + "5 27 41 32 0 \n", + "8 30 45 45 -14 \n", + ".. ... ... ... ... \n", + "281 4 76 1 7 \n", + "282 11 63 17 -3 \n", + "283 8 85 12 -6 \n", + "287 30 76 9 12 \n", + "292 30 70 6 20 \n", + "\n", + "[147 rows x 50 columns]" + ] + }, + "execution_count": 156, + "metadata": {}, + "output_type": "execute_result" + } + ], + "source": [ + "measurements_pre_post_intervention['ahiChange'] = \\\n", + " measurements_pre_post_intervention['ahiTotal.post'] - measurements_pre_post_intervention['ahiTotal.pre']\n", + "\n", + "measurements_pre_post_intervention" + ] + }, + { + "cell_type": "code", + "execution_count": 157, + "metadata": {}, + "outputs": [ + { + "data": { + "text/html": [ + "
\n", + "\n", + "\n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + "
interventionahi01ahi02ahi03ahi04ahi05ahi06ahi07ahi08ahi09...cesd17cesd18cesd19cesd20ahiTotal.precesdTotal.preahiTotal.postcesdTotal.postahiChangeahiResponse
14232332333...1122631473610True
21343423433...1111737891016True
43223223315...3311603167317False
52123222112...1213412741320False
82322221444...242359304545-14False
..................................................................
2814332423443...12116947617False
2821232245324...113366116317-3False
2832333444434...11129188512-6False
2872231341134...3323643076912True
2924222224124...2314503070620True
\n", + "

147 rows × 51 columns

\n", + "
" + ], + "text/plain": [ + " intervention ahi01 ahi02 ahi03 ahi04 ahi05 ahi06 ahi07 ahi08 \\\n", + "1 4 2 3 2 3 3 2 3 3 \n", + "2 1 3 4 3 4 2 3 4 3 \n", + "4 3 2 2 3 2 2 3 3 1 \n", + "5 2 1 2 3 2 2 2 1 1 \n", + "8 2 3 2 2 2 2 1 4 4 \n", + ".. ... ... ... ... ... ... ... ... ... \n", + "281 4 3 3 2 4 2 3 4 4 \n", + "282 1 2 3 2 2 4 5 3 2 \n", + "283 2 3 3 3 4 4 4 4 3 \n", + "287 2 2 3 1 3 4 1 1 3 \n", + "292 4 2 2 2 2 2 4 1 2 \n", + "\n", + " ahi09 ... cesd17 cesd18 cesd19 cesd20 ahiTotal.pre cesdTotal.pre \\\n", + "1 3 ... 1 1 2 2 63 14 \n", + "2 3 ... 1 1 1 1 73 7 \n", + "4 5 ... 3 3 1 1 60 31 \n", + "5 2 ... 1 2 1 3 41 27 \n", + "8 4 ... 2 4 2 3 59 30 \n", + ".. ... ... ... ... ... ... ... ... \n", + "281 3 ... 1 2 1 1 69 4 \n", + "282 4 ... 1 1 3 3 66 11 \n", + "283 4 ... 1 1 1 2 91 8 \n", + "287 4 ... 3 3 2 3 64 30 \n", + "292 4 ... 2 3 1 4 50 30 \n", + "\n", + " ahiTotal.post cesdTotal.post ahiChange ahiResponse \n", + "1 73 6 10 True \n", + "2 89 10 16 True \n", + "4 67 31 7 False \n", + "5 41 32 0 False \n", + "8 45 45 -14 False \n", + ".. ... ... ... ... \n", + "281 76 1 7 False \n", + "282 63 17 -3 False \n", + "283 85 12 -6 False \n", + "287 76 9 12 True \n", + "292 70 6 20 True \n", + "\n", + "[147 rows x 51 columns]" + ] + }, + "execution_count": 157, + "metadata": {}, + "output_type": "execute_result" + } + ], + "source": [ + "measurements_pre_post_intervention['ahiResponse'] = \\\n", + " measurements_pre_post_intervention['ahiChange'] >= 10\n", + "\n", + "measurements_pre_post_intervention" + ] + }, + { + "cell_type": "code", + "execution_count": 158, + "metadata": {}, + "outputs": [ + { + "data": { + "text/html": [ + "
\n", + "\n", + "\n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + "
interventionahi01ahi02ahi03ahi04ahi05ahi06ahi07ahi08ahi09...cesd18cesd19cesd20ahiTotal.precesdTotal.preahiTotal.postcesdTotal.postahiChangeahiResponsecesdChange
14232332333...122631473610True-8
21343423433...111737891016True3
43223223315...311603167317False0
52123222112...213412741320False5
82322221444...42359304545-14False15
..................................................................
2814332423443...2116947617False-3
2821232245324...13366116317-3False6
2832333444434...1129188512-6False4
2872231341134...323643076912True-21
2924222224124...314503070620True-24
\n", + "

147 rows × 52 columns

\n", + "
" + ], + "text/plain": [ + " intervention ahi01 ahi02 ahi03 ahi04 ahi05 ahi06 ahi07 ahi08 \\\n", + "1 4 2 3 2 3 3 2 3 3 \n", + "2 1 3 4 3 4 2 3 4 3 \n", + "4 3 2 2 3 2 2 3 3 1 \n", + "5 2 1 2 3 2 2 2 1 1 \n", + "8 2 3 2 2 2 2 1 4 4 \n", + ".. ... ... ... ... ... ... ... ... ... \n", + "281 4 3 3 2 4 2 3 4 4 \n", + "282 1 2 3 2 2 4 5 3 2 \n", + "283 2 3 3 3 4 4 4 4 3 \n", + "287 2 2 3 1 3 4 1 1 3 \n", + "292 4 2 2 2 2 2 4 1 2 \n", + "\n", + " ahi09 ... cesd18 cesd19 cesd20 ahiTotal.pre cesdTotal.pre \\\n", + "1 3 ... 1 2 2 63 14 \n", + "2 3 ... 1 1 1 73 7 \n", + "4 5 ... 3 1 1 60 31 \n", + "5 2 ... 2 1 3 41 27 \n", + "8 4 ... 4 2 3 59 30 \n", + ".. ... ... ... ... ... ... ... \n", + "281 3 ... 2 1 1 69 4 \n", + "282 4 ... 1 3 3 66 11 \n", + "283 4 ... 1 1 2 91 8 \n", + "287 4 ... 3 2 3 64 30 \n", + "292 4 ... 3 1 4 50 30 \n", + "\n", + " ahiTotal.post cesdTotal.post ahiChange ahiResponse cesdChange \n", + "1 73 6 10 True -8 \n", + "2 89 10 16 True 3 \n", + "4 67 31 7 False 0 \n", + "5 41 32 0 False 5 \n", + "8 45 45 -14 False 15 \n", + ".. ... ... ... ... ... \n", + "281 76 1 7 False -3 \n", + "282 63 17 -3 False 6 \n", + "283 85 12 -6 False 4 \n", + "287 76 9 12 True -21 \n", + "292 70 6 20 True -24 \n", + "\n", + "[147 rows x 52 columns]" + ] + }, + "execution_count": 158, + "metadata": {}, + "output_type": "execute_result" + } + ], + "source": [ + "measurements_pre_post_intervention['cesdChange'] = \\\n", + " measurements_pre_post_intervention['cesdTotal.post'] - measurements_pre_post_intervention['cesdTotal.pre']\n", + "\n", + "measurements_pre_post_intervention" + ] + }, + { + "cell_type": "code", + "execution_count": 159, + "metadata": {}, + "outputs": [ + { + "data": { + "text/html": [ + "
\n", + "\n", + "\n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + "
interventionahi01ahi02ahi03ahi04ahi05ahi06ahi07ahi08ahi09...cesd19cesd20ahiTotal.precesdTotal.preahiTotal.postcesdTotal.postahiChangeahiResponsecesdChangecesdResponse
14232332333...22631473610True-8False
21343423433...11737891016True3False
43223223315...11603167317False0False
52123222112...13412741320False5False
82322221444...2359304545-14False15False
..................................................................
2814332423443...116947617False-3False
2821232245324...3366116317-3False6False
2832333444434...129188512-6False4False
2872231341134...23643076912True-21True
2924222224124...14503070620True-24True
\n", + "

147 rows × 53 columns

\n", + "
" + ], + "text/plain": [ + " intervention ahi01 ahi02 ahi03 ahi04 ahi05 ahi06 ahi07 ahi08 \\\n", + "1 4 2 3 2 3 3 2 3 3 \n", + "2 1 3 4 3 4 2 3 4 3 \n", + "4 3 2 2 3 2 2 3 3 1 \n", + "5 2 1 2 3 2 2 2 1 1 \n", + "8 2 3 2 2 2 2 1 4 4 \n", + ".. ... ... ... ... ... ... ... ... ... \n", + "281 4 3 3 2 4 2 3 4 4 \n", + "282 1 2 3 2 2 4 5 3 2 \n", + "283 2 3 3 3 4 4 4 4 3 \n", + "287 2 2 3 1 3 4 1 1 3 \n", + "292 4 2 2 2 2 2 4 1 2 \n", + "\n", + " ahi09 ... cesd19 cesd20 ahiTotal.pre cesdTotal.pre ahiTotal.post \\\n", + "1 3 ... 2 2 63 14 73 \n", + "2 3 ... 1 1 73 7 89 \n", + "4 5 ... 1 1 60 31 67 \n", + "5 2 ... 1 3 41 27 41 \n", + "8 4 ... 2 3 59 30 45 \n", + ".. ... ... ... ... ... ... ... \n", + "281 3 ... 1 1 69 4 76 \n", + "282 4 ... 3 3 66 11 63 \n", + "283 4 ... 1 2 91 8 85 \n", + "287 4 ... 2 3 64 30 76 \n", + "292 4 ... 1 4 50 30 70 \n", + "\n", + " cesdTotal.post ahiChange ahiResponse cesdChange cesdResponse \n", + "1 6 10 True -8 False \n", + "2 10 16 True 3 False \n", + "4 31 7 False 0 False \n", + "5 32 0 False 5 False \n", + "8 45 -14 False 15 False \n", + ".. ... ... ... ... ... \n", + "281 1 7 False -3 False \n", + "282 17 -3 False 6 False \n", + "283 12 -6 False 4 False \n", + "287 9 12 True -21 True \n", + "292 6 20 True -24 True \n", + "\n", + "[147 rows x 53 columns]" + ] + }, + "execution_count": 159, + "metadata": {}, + "output_type": "execute_result" + } + ], + "source": [ + "measurements_pre_post_intervention['cesdResponse'] = \\\n", + " measurements_pre_post_intervention['cesdChange'] <= -10\n", + "\n", + "measurements_pre_post_intervention" + ] + }, + { + "cell_type": "code", + "execution_count": 160, + "metadata": {}, + "outputs": [ + { + "data": { + "text/html": [ + "
\n", + "\n", + "\n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + "
idinterventionsexageeducincome
01423553
12115911
23415143
34315052
45225852
.....................
290291213953
291292414742
292293225243
293294315952
294295112842
\n", + "

295 rows × 6 columns

\n", + "
" + ], + "text/plain": [ + " id intervention sex age educ income\n", + "0 1 4 2 35 5 3\n", + "1 2 1 1 59 1 1\n", + "2 3 4 1 51 4 3\n", + "3 4 3 1 50 5 2\n", + "4 5 2 2 58 5 2\n", + ".. ... ... ... ... ... ...\n", + "290 291 2 1 39 5 3\n", + "291 292 4 1 47 4 2\n", + "292 293 2 2 52 4 3\n", + "293 294 3 1 59 5 2\n", + "294 295 1 1 28 4 2\n", + "\n", + "[295 rows x 6 columns]" + ] + }, + "execution_count": 160, + "metadata": {}, + "output_type": "execute_result" + } + ], + "source": [ + "demographics = pd.read_csv('dados/participant-info.csv')\n", + "demographics" + ] + }, + { + "cell_type": "code", + "execution_count": 161, + "metadata": {}, + "outputs": [ + { + "data": { + "text/html": [ + "
\n", + "\n", + "\n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + "
interventionsexageeducincome
1423553
2115911
3415143
4315052
5225852
..................
291213953
292414742
293225243
294315952
295112842
\n", + "

295 rows × 5 columns

\n", + "
" + ], + "text/plain": [ + " intervention sex age educ income\n", + "1 4 2 35 5 3\n", + "2 1 1 59 1 1\n", + "3 4 1 51 4 3\n", + "4 3 1 50 5 2\n", + "5 2 2 58 5 2\n", + ".. ... ... ... ... ...\n", + "291 2 1 39 5 3\n", + "292 4 1 47 4 2\n", + "293 2 2 52 4 3\n", + "294 3 1 59 5 2\n", + "295 1 1 28 4 2\n", + "\n", + "[295 rows x 5 columns]" + ] + }, + "execution_count": 161, + "metadata": {}, + "output_type": "execute_result" + } + ], + "source": [ + "demographics = demographics.set_index('id')\n", + "\n", + "demographics = \\\n", + " demographics.rename_axis(index=None)\n", + "\n", + "demographics" + ] + }, + { + "cell_type": "code", + "execution_count": 162, + "metadata": {}, + "outputs": [ + { + "data": { + "text/html": [ + "
\n", + "\n", + "\n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + "
sexageeducincome
123553
215911
315143
415052
525852
...............
29113953
29214742
29325243
29415952
29512842
\n", + "

295 rows × 4 columns

\n", + "
" + ], + "text/plain": [ + " sex age educ income\n", + "1 2 35 5 3\n", + "2 1 59 1 1\n", + "3 1 51 4 3\n", + "4 1 50 5 2\n", + "5 2 58 5 2\n", + ".. ... ... ... ...\n", + "291 1 39 5 3\n", + "292 1 47 4 2\n", + "293 2 52 4 3\n", + "294 1 59 5 2\n", + "295 1 28 4 2\n", + "\n", + "[295 rows x 4 columns]" + ] + }, + "execution_count": 162, + "metadata": {}, + "output_type": "execute_result" + } + ], + "source": [ + "demographics = demographics.drop(columns=['intervention'])\n", + "demographics" + ] + }, + { + "cell_type": "code", + "execution_count": 163, + "metadata": {}, + "outputs": [ + { + "data": { + "text/html": [ + "
\n", + "\n", + "\n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + "
sexageeducincomeinterventionahi01ahi02ahi03ahi04ahi05...cesd19cesd20ahiTotal.precesdTotal.preahiTotal.postcesdTotal.postahiChangeahiResponsecesdChangecesdResponse
123553423233...22631473610True-8False
215911134342...11737891016True3False
415052322322...11603167317False0False
525852212322...13412741320False5False
815742232222...2359304545-14False15False
..................................................................
28126742433242...116947617False-3False
28214322123224...3366116317-3False6False
28313942233344...129188512-6False4False
28714342223134...23643076912True-21True
29214742422222...14503070620True-24True
\n", + "

147 rows × 57 columns

\n", + "
" + ], + "text/plain": [ + " sex age educ income intervention ahi01 ahi02 ahi03 ahi04 ahi05 \\\n", + "1 2 35 5 3 4 2 3 2 3 3 \n", + "2 1 59 1 1 1 3 4 3 4 2 \n", + "4 1 50 5 2 3 2 2 3 2 2 \n", + "5 2 58 5 2 2 1 2 3 2 2 \n", + "8 1 57 4 2 2 3 2 2 2 2 \n", + ".. ... ... ... ... ... ... ... ... ... ... \n", + "281 2 67 4 2 4 3 3 2 4 2 \n", + "282 1 43 2 2 1 2 3 2 2 4 \n", + "283 1 39 4 2 2 3 3 3 4 4 \n", + "287 1 43 4 2 2 2 3 1 3 4 \n", + "292 1 47 4 2 4 2 2 2 2 2 \n", + "\n", + " ... cesd19 cesd20 ahiTotal.pre cesdTotal.pre ahiTotal.post \\\n", + "1 ... 2 2 63 14 73 \n", + "2 ... 1 1 73 7 89 \n", + "4 ... 1 1 60 31 67 \n", + "5 ... 1 3 41 27 41 \n", + "8 ... 2 3 59 30 45 \n", + ".. ... ... ... ... ... ... \n", + "281 ... 1 1 69 4 76 \n", + "282 ... 3 3 66 11 63 \n", + "283 ... 1 2 91 8 85 \n", + "287 ... 2 3 64 30 76 \n", + "292 ... 1 4 50 30 70 \n", + "\n", + " cesdTotal.post ahiChange ahiResponse cesdChange cesdResponse \n", + "1 6 10 True -8 False \n", + "2 10 16 True 3 False \n", + "4 31 7 False 0 False \n", + "5 32 0 False 5 False \n", + "8 45 -14 False 15 False \n", + ".. ... ... ... ... ... \n", + "281 1 7 False -3 False \n", + "282 17 -3 False 6 False \n", + "283 12 -6 False 4 False \n", + "287 9 12 True -21 True \n", + "292 6 20 True -24 True \n", + "\n", + "[147 rows x 57 columns]" + ] + }, + "execution_count": 163, + "metadata": {}, + "output_type": "execute_result" + } + ], + "source": [ + "data = demographics.join(measurements_pre_post_intervention, how='inner')\n", + "data" + ] + }, + { + "cell_type": "code", + "execution_count": 165, + "metadata": {}, + "outputs": [ + { + "data": { + "text/html": [ + "
\n", + "\n", + "\n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + "
sexageeducincomeinterventionahi01ahi02ahi03ahi04ahi05...cesd19cesd20ahiTotal.precesdTotal.preahiTotal.postcesdTotal.postahiChangeahiResponsecesdChangecesdResponse
215911134342...11737891016True3False
415052322322...11603167317False0False
525852212322...13412741320False5False
815742232222...2359304545-14False15False
1115651222342...1162195919-3False0False
..................................................................
27315552323342...127618004False-1False
27613752122223...2159246889False-16True
28214322123224...3366116317-3False6False
28313942233344...129188512-6False4False
28714342223134...23643076912True-21True
\n", + "

102 rows × 57 columns

\n", + "
" + ], + "text/plain": [ + " sex age educ income intervention ahi01 ahi02 ahi03 ahi04 ahi05 \\\n", + "2 1 59 1 1 1 3 4 3 4 2 \n", + "4 1 50 5 2 3 2 2 3 2 2 \n", + "5 2 58 5 2 2 1 2 3 2 2 \n", + "8 1 57 4 2 2 3 2 2 2 2 \n", + "11 1 56 5 1 2 2 2 3 4 2 \n", + ".. ... ... ... ... ... ... ... ... ... ... \n", + "273 1 55 5 2 3 2 3 3 4 2 \n", + "276 1 37 5 2 1 2 2 2 2 3 \n", + "282 1 43 2 2 1 2 3 2 2 4 \n", + "283 1 39 4 2 2 3 3 3 4 4 \n", + "287 1 43 4 2 2 2 3 1 3 4 \n", + "\n", + " ... cesd19 cesd20 ahiTotal.pre cesdTotal.pre ahiTotal.post \\\n", + "2 ... 1 1 73 7 89 \n", + "4 ... 1 1 60 31 67 \n", + "5 ... 1 3 41 27 41 \n", + "8 ... 2 3 59 30 45 \n", + "11 ... 1 1 62 19 59 \n", + ".. ... ... ... ... ... ... \n", + "273 ... 1 2 76 1 80 \n", + "276 ... 2 1 59 24 68 \n", + "282 ... 3 3 66 11 63 \n", + "283 ... 1 2 91 8 85 \n", + "287 ... 2 3 64 30 76 \n", + "\n", + " cesdTotal.post ahiChange ahiResponse cesdChange cesdResponse \n", + "2 10 16 True 3 False \n", + "4 31 7 False 0 False \n", + "5 32 0 False 5 False \n", + "8 45 -14 False 15 False \n", + "11 19 -3 False 0 False \n", + ".. ... ... ... ... ... \n", + "273 0 4 False -1 False \n", + "276 8 9 False -16 True \n", + "282 17 -3 False 6 False \n", + "283 12 -6 False 4 False \n", + "287 9 12 True -21 True \n", + "\n", + "[102 rows x 57 columns]" + ] + }, + "execution_count": 165, + "metadata": {}, + "output_type": "execute_result" + } + ], + "source": [ + "data = data.query('intervention != 4')\n", + "data" + ] + }, + { + "cell_type": "code", + "execution_count": 166, + "metadata": {}, + "outputs": [ + { + "data": { + "text/html": [ + "
\n", + "\n", + "\n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + "
sexageeducincomeahi01ahi02ahi03ahi04ahi05ahi06...cesd19cesd20ahiTotal.precesdTotal.preahiTotal.postcesdTotal.postahiChangeahiResponsecesdChangecesdResponse
215911343423...11737891016True3False
415052223223...11603167317False0False
525852123222...13412741320False5False
815742322221...2359304545-14False15False
1115651223423...1162195919-3False0False
..................................................................
27315552233424...127618004False-1False
27613752222233...2159246889False-16True
28214322232245...3366116317-3False6False
28313942333444...129188512-6False4False
28714342231341...23643076912True-21True
\n", + "

102 rows × 56 columns

\n", + "
" + ], + "text/plain": [ + " sex age educ income ahi01 ahi02 ahi03 ahi04 ahi05 ahi06 ... \\\n", + "2 1 59 1 1 3 4 3 4 2 3 ... \n", + "4 1 50 5 2 2 2 3 2 2 3 ... \n", + "5 2 58 5 2 1 2 3 2 2 2 ... \n", + "8 1 57 4 2 3 2 2 2 2 1 ... \n", + "11 1 56 5 1 2 2 3 4 2 3 ... \n", + ".. ... ... ... ... ... ... ... ... ... ... ... \n", + "273 1 55 5 2 2 3 3 4 2 4 ... \n", + "276 1 37 5 2 2 2 2 2 3 3 ... \n", + "282 1 43 2 2 2 3 2 2 4 5 ... \n", + "283 1 39 4 2 3 3 3 4 4 4 ... \n", + "287 1 43 4 2 2 3 1 3 4 1 ... \n", + "\n", + " cesd19 cesd20 ahiTotal.pre cesdTotal.pre ahiTotal.post \\\n", + "2 1 1 73 7 89 \n", + "4 1 1 60 31 67 \n", + "5 1 3 41 27 41 \n", + "8 2 3 59 30 45 \n", + "11 1 1 62 19 59 \n", + ".. ... ... ... ... ... \n", + "273 1 2 76 1 80 \n", + "276 2 1 59 24 68 \n", + "282 3 3 66 11 63 \n", + "283 1 2 91 8 85 \n", + "287 2 3 64 30 76 \n", + "\n", + " cesdTotal.post ahiChange ahiResponse cesdChange cesdResponse \n", + "2 10 16 True 3 False \n", + "4 31 7 False 0 False \n", + "5 32 0 False 5 False \n", + "8 45 -14 False 15 False \n", + "11 19 -3 False 0 False \n", + ".. ... ... ... ... ... \n", + "273 0 4 False -1 False \n", + "276 8 9 False -16 True \n", + "282 17 -3 False 6 False \n", + "283 12 -6 False 4 False \n", + "287 9 12 True -21 True \n", + "\n", + "[102 rows x 56 columns]" ] }, - "execution_count": 5, + "execution_count": 166, "metadata": {}, "output_type": "execute_result" } ], "source": [ - "participant_info = pd.read_csv('dados/participant-info.csv')\n", - "participant_info" + "data = data.drop(columns=['intervention'])\n", + "data" ] } ],