Project branch

.ipynb_checkpoints/TEST PCA-checkpoint.ipynb

@@ -0,0 +1,6 @@
+{
+ "cells": [],
+ "metadata": {},
+ "nbformat": 4,
+ "nbformat_minor": 4
+}

.ipynb_checkpoints/create data frames for viz-checkpoint.ipynb

@@ -0,0 +1,291 @@
+{
+ "cells": [
+  {
+   "cell_type": "code",
+   "execution_count": 2,
+   "metadata": {
+    "ExecuteTime": {
+     "end_time": "2020-06-04T16:55:28.761296Z",
+     "start_time": "2020-06-04T16:55:28.546730Z"
+    }
+   },
+   "outputs": [],
+   "source": [
+    "import pandas as pd"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": 6,
+   "metadata": {
+    "ExecuteTime": {
+     "end_time": "2020-06-04T16:57:35.037958Z",
+     "start_time": "2020-06-04T16:57:34.980862Z"
+    }
+   },
+   "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>id</th>\n",
+       "      <th>Gender</th>\n",
+       "      <th>subtype</th>\n",
+       "      <th>adhdtype</th>\n",
+       "      <th>brain_oscillation</th>\n",
+       "      <th>electrode</th>\n",
+       "      <th>fft_abs_power</th>\n",
+       "    </tr>\n",
+       "  </thead>\n",
+       "  <tbody>\n",
+       "    <tr>\n",
+       "      <th>0</th>\n",
+       "      <td>1</td>\n",
+       "      <td>1</td>\n",
+       "      <td>mixed</td>\n",
+       "      <td>2</td>\n",
+       "      <td>Delta</td>\n",
+       "      <td>FP1</td>\n",
+       "      <td>15.376173</td>\n",
+       "    </tr>\n",
+       "    <tr>\n",
+       "      <th>133</th>\n",
+       "      <td>1</td>\n",
+       "      <td>1</td>\n",
+       "      <td>mixed</td>\n",
+       "      <td>2</td>\n",
+       "      <td>Beta2</td>\n",
+       "      <td>Pz</td>\n",
+       "      <td>1.870920</td>\n",
+       "    </tr>\n",
+       "    <tr>\n",
+       "      <th>134</th>\n",
+       "      <td>1</td>\n",
+       "      <td>1</td>\n",
+       "      <td>mixed</td>\n",
+       "      <td>2</td>\n",
+       "      <td>Beta2</td>\n",
+       "      <td>P3</td>\n",
+       "      <td>1.690150</td>\n",
+       "    </tr>\n",
+       "    <tr>\n",
+       "      <th>135</th>\n",
+       "      <td>1</td>\n",
+       "      <td>1</td>\n",
+       "      <td>mixed</td>\n",
+       "      <td>2</td>\n",
+       "      <td>Beta2</td>\n",
+       "      <td>T5</td>\n",
+       "      <td>1.138450</td>\n",
+       "    </tr>\n",
+       "    <tr>\n",
+       "      <th>136</th>\n",
+       "      <td>1</td>\n",
+       "      <td>1</td>\n",
+       "      <td>mixed</td>\n",
+       "      <td>2</td>\n",
+       "      <td>Beta2</td>\n",
+       "      <td>T4</td>\n",
+       "      <td>2.171898</td>\n",
+       "    </tr>\n",
+       "    <tr>\n",
+       "      <th>...</th>\n",
+       "      <td>...</td>\n",
+       "      <td>...</td>\n",
+       "      <td>...</td>\n",
+       "      <td>...</td>\n",
+       "      <td>...</td>\n",
+       "      <td>...</td>\n",
+       "      <td>...</td>\n",
+       "    </tr>\n",
+       "    <tr>\n",
+       "      <th>15958</th>\n",
+       "      <td>702</td>\n",
+       "      <td>1</td>\n",
+       "      <td>inat</td>\n",
+       "      <td>1</td>\n",
+       "      <td>Theta</td>\n",
+       "      <td>Fz</td>\n",
+       "      <td>15.355240</td>\n",
+       "    </tr>\n",
+       "    <tr>\n",
+       "      <th>15959</th>\n",
+       "      <td>702</td>\n",
+       "      <td>1</td>\n",
+       "      <td>inat</td>\n",
+       "      <td>1</td>\n",
+       "      <td>Alpha</td>\n",
+       "      <td>P3</td>\n",
+       "      <td>26.927847</td>\n",
+       "    </tr>\n",
+       "    <tr>\n",
+       "      <th>15960</th>\n",
+       "      <td>702</td>\n",
+       "      <td>1</td>\n",
+       "      <td>inat</td>\n",
+       "      <td>1</td>\n",
+       "      <td>Theta</td>\n",
+       "      <td>F4</td>\n",
+       "      <td>12.414647</td>\n",
+       "    </tr>\n",
+       "    <tr>\n",
+       "      <th>15935</th>\n",
+       "      <td>702</td>\n",
+       "      <td>1</td>\n",
+       "      <td>inat</td>\n",
+       "      <td>1</td>\n",
+       "      <td>Theta</td>\n",
+       "      <td>F3</td>\n",
+       "      <td>12.058871</td>\n",
+       "    </tr>\n",
+       "    <tr>\n",
+       "      <th>16092</th>\n",
+       "      <td>702</td>\n",
+       "      <td>1</td>\n",
+       "      <td>inat</td>\n",
+       "      <td>1</td>\n",
+       "      <td>Alpha2</td>\n",
+       "      <td>F3</td>\n",
+       "      <td>7.689915</td>\n",
+       "    </tr>\n",
+       "  </tbody>\n",
+       "</table>\n",
+       "<p>20064 rows × 7 columns</p>\n",
+       "</div>"
+      ],
+      "text/plain": [
+       "        id  Gender subtype  adhdtype brain_oscillation electrode  \\\n",
+       "0        1       1   mixed         2             Delta       FP1   \n",
+       "133      1       1   mixed         2             Beta2        Pz   \n",
+       "134      1       1   mixed         2             Beta2        P3   \n",
+       "135      1       1   mixed         2             Beta2        T5   \n",
+       "136      1       1   mixed         2             Beta2        T4   \n",
+       "...    ...     ...     ...       ...               ...       ...   \n",
+       "15958  702       1    inat         1             Theta        Fz   \n",
+       "15959  702       1    inat         1             Alpha        P3   \n",
+       "15960  702       1    inat         1             Theta        F4   \n",
+       "15935  702       1    inat         1             Theta        F3   \n",
+       "16092  702       1    inat         1            Alpha2        F3   \n",
+       "\n",
+       "       fft_abs_power  \n",
+       "0          15.376173  \n",
+       "133         1.870920  \n",
+       "134         1.690150  \n",
+       "135         1.138450  \n",
+       "136         2.171898  \n",
+       "...              ...  \n",
+       "15958      15.355240  \n",
+       "15959      26.927847  \n",
+       "15960      12.414647  \n",
+       "15935      12.058871  \n",
+       "16092       7.689915  \n",
+       "\n",
+       "[20064 rows x 7 columns]"
+      ]
+     },
+     "execution_count": 6,
+     "metadata": {},
+     "output_type": "execute_result"
+    }
+   ],
+   "source": [
+    "df_analysis = pd.read_csv('Data/df_analysis', index_col=[0])\n",
+    "df_analysis = df_analysis[['id', 'Gender', 'subtype', 'adhdtype', 'brain_oscillation', 'electrode', 'fft_abs_power']]\n",
+    "df_analysis"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": 9,
+   "metadata": {
+    "ExecuteTime": {
+     "end_time": "2020-06-04T17:02:53.663898Z",
+     "start_time": "2020-06-04T17:02:53.631124Z"
+    }
+   },
+   "outputs": [],
+   "source": [
+    "df_Delta  = df_analysis.loc[df_analysis['brain_oscillation'] == 'Delta']\n",
+    "df_Theta  = df_analysis.loc[df_analysis['brain_oscillation'] == 'Theta']\n",
+    "df_Alpha  = df_analysis.loc[df_analysis['brain_oscillation'] == 'Alpha']\n",
+    "df_Alpha1  = df_analysis.loc[df_analysis['brain_oscillation'] == 'Alpha1']\n",
+    "df_Alpha2  = df_analysis.loc[df_analysis['brain_oscillation'] == 'Alpha2']\n",
+    "df_Beta  = df_analysis.loc[df_analysis['brain_oscillation'] == 'Beta']\n",
+    "df_Beta1  = df_analysis.loc[df_analysis['brain_oscillation'] == 'Beta1']\n",
+    "df_Beta2  = df_analysis.loc[df_analysis['brain_oscillation'] == 'Beta2']"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": 10,
+   "metadata": {
+    "ExecuteTime": {
+     "end_time": "2020-06-04T17:03:30.319250Z",
+     "start_time": "2020-06-04T17:03:30.247119Z"
+    }
+   },
+   "outputs": [],
+   "source": [
+    "df_Delta.to_csv('Data/df_Delta')\n",
+    "df_Theta.to_csv('Data/df_Theta')\n",
+    "df_Alpha.to_csv('Data/df_Alpha')\n",
+    "df_Alpha1.to_csv('Data/df_Alpha1')\n",
+    "df_Alpha2.to_csv('Data/df_Alpha2')\n",
+    "df_Beta.to_csv('Data/df_Beta')\n",
+    "df_Beta1.to_csv('Data/df_Beta1')\n",
+    "df_Beta2.to_csv('Data/df_Beta2')"
+   ]
+  }
+ ],
+ "metadata": {
+  "kernelspec": {
+   "display_name": "Python 3",
+   "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.7.7"
+  },
+  "toc": {
+   "base_numbering": 1,
+   "nav_menu": {},
+   "number_sections": true,
+   "sideBar": true,
+   "skip_h1_title": false,
+   "title_cell": "Table of Contents",
+   "title_sidebar": "Contents",
+   "toc_cell": false,
+   "toc_position": {},
+   "toc_section_display": true,
+   "toc_window_display": false
+  }
+ },
+ "nbformat": 4,
+ "nbformat_minor": 4
+}