cleaning_data.ipynb

{
 "cells": [
  {
   "cell_type": "code",
   "execution_count": 1,
   "metadata": {},
   "outputs": [],
   "source": [
    "# Import dependencies\n",
    "from sklearn.model_selection import train_test_split\n",
    "from sklearn.preprocessing import StandardScaler,OneHotEncoder\n",
    "from sklearn.ensemble import RandomForestClassifier\n",
    "from sklearn.metrics import accuracy_score\n",
    "import pandas as pd\n",
    "from scipy import stats\n",
    "# import tensorflow as tf"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 2,
   "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>Are you self-employed?</th>\n",
       "      <th>How many employees does your company or organization have?</th>\n",
       "      <th>Is your employer primarily a tech company/organization?</th>\n",
       "      <th>Is your primary role within your company related to tech/IT?</th>\n",
       "      <th>Does your employer provide mental health benefits as part of healthcare coverage?</th>\n",
       "      <th>Do you know the options for mental health care available under your employer-provided coverage?</th>\n",
       "      <th>Has your employer ever formally discussed mental health (for example, as part of a wellness campaign or other official communication)?</th>\n",
       "      <th>Does your employer offer resources to learn more about mental health concerns and options for seeking help?</th>\n",
       "      <th>Is your anonymity protected if you choose to take advantage of mental health or substance abuse treatment resources provided by your employer?</th>\n",
       "      <th>If a mental health issue prompted you to request a medical leave from work, asking for that leave would be:</th>\n",
       "      <th>...</th>\n",
       "      <th>If you have a mental health issue, do you feel that it interferes with your work when being treated effectively?</th>\n",
       "      <th>If you have a mental health issue, do you feel that it interferes with your work when NOT being treated effectively?</th>\n",
       "      <th>What is your age?</th>\n",
       "      <th>What is your gender?</th>\n",
       "      <th>What country do you live in?</th>\n",
       "      <th>What US state or territory do you live in?</th>\n",
       "      <th>What country do you work in?</th>\n",
       "      <th>What US state or territory do you work in?</th>\n",
       "      <th>Which of the following best describes your work position?</th>\n",
       "      <th>Do you work remotely?</th>\n",
       "    </tr>\n",
       "  </thead>\n",
       "  <tbody>\n",
       "    <tr>\n",
       "      <th>0</th>\n",
       "      <td>0</td>\n",
       "      <td>26-100</td>\n",
       "      <td>1.0</td>\n",
       "      <td>NaN</td>\n",
       "      <td>Not eligible for coverage / N/A</td>\n",
       "      <td>NaN</td>\n",
       "      <td>No</td>\n",
       "      <td>No</td>\n",
       "      <td>I don't know</td>\n",
       "      <td>Very easy</td>\n",
       "      <td>...</td>\n",
       "      <td>Not applicable to me</td>\n",
       "      <td>Not applicable to me</td>\n",
       "      <td>39</td>\n",
       "      <td>Male</td>\n",
       "      <td>United Kingdom</td>\n",
       "      <td>NaN</td>\n",
       "      <td>United Kingdom</td>\n",
       "      <td>NaN</td>\n",
       "      <td>Back-end Developer</td>\n",
       "      <td>Sometimes</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>1</th>\n",
       "      <td>0</td>\n",
       "      <td>6-25</td>\n",
       "      <td>1.0</td>\n",
       "      <td>NaN</td>\n",
       "      <td>No</td>\n",
       "      <td>Yes</td>\n",
       "      <td>Yes</td>\n",
       "      <td>Yes</td>\n",
       "      <td>Yes</td>\n",
       "      <td>Somewhat easy</td>\n",
       "      <td>...</td>\n",
       "      <td>Rarely</td>\n",
       "      <td>Sometimes</td>\n",
       "      <td>29</td>\n",
       "      <td>male</td>\n",
       "      <td>United States of America</td>\n",
       "      <td>Illinois</td>\n",
       "      <td>United States of America</td>\n",
       "      <td>Illinois</td>\n",
       "      <td>Back-end Developer|Front-end Developer</td>\n",
       "      <td>Never</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>2</th>\n",
       "      <td>0</td>\n",
       "      <td>6-25</td>\n",
       "      <td>1.0</td>\n",
       "      <td>NaN</td>\n",
       "      <td>No</td>\n",
       "      <td>NaN</td>\n",
       "      <td>No</td>\n",
       "      <td>No</td>\n",
       "      <td>I don't know</td>\n",
       "      <td>Neither easy nor difficult</td>\n",
       "      <td>...</td>\n",
       "      <td>Not applicable to me</td>\n",
       "      <td>Not applicable to me</td>\n",
       "      <td>38</td>\n",
       "      <td>Male</td>\n",
       "      <td>United Kingdom</td>\n",
       "      <td>NaN</td>\n",
       "      <td>United Kingdom</td>\n",
       "      <td>NaN</td>\n",
       "      <td>Back-end Developer</td>\n",
       "      <td>Always</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>3</th>\n",
       "      <td>1</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>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>...</td>\n",
       "      <td>Sometimes</td>\n",
       "      <td>Sometimes</td>\n",
       "      <td>43</td>\n",
       "      <td>male</td>\n",
       "      <td>United Kingdom</td>\n",
       "      <td>NaN</td>\n",
       "      <td>United Kingdom</td>\n",
       "      <td>NaN</td>\n",
       "      <td>Supervisor/Team Lead</td>\n",
       "      <td>Sometimes</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>4</th>\n",
       "      <td>0</td>\n",
       "      <td>6-25</td>\n",
       "      <td>0.0</td>\n",
       "      <td>1.0</td>\n",
       "      <td>Yes</td>\n",
       "      <td>Yes</td>\n",
       "      <td>No</td>\n",
       "      <td>No</td>\n",
       "      <td>No</td>\n",
       "      <td>Neither easy nor difficult</td>\n",
       "      <td>...</td>\n",
       "      <td>Sometimes</td>\n",
       "      <td>Sometimes</td>\n",
       "      <td>43</td>\n",
       "      <td>Female</td>\n",
       "      <td>United States of America</td>\n",
       "      <td>Illinois</td>\n",
       "      <td>United States of America</td>\n",
       "      <td>Illinois</td>\n",
       "      <td>Executive Leadership|Supervisor/Team Lead|Dev ...</td>\n",
       "      <td>Sometimes</td>\n",
       "    </tr>\n",
       "  </tbody>\n",
       "</table>\n",
       "<p>5 rows × 63 columns</p>\n",
       "</div>"
      ],
      "text/plain": [
       "   Are you self-employed?  \\\n",
       "0                       0   \n",
       "1                       0   \n",
       "2                       0   \n",
       "3                       1   \n",
       "4                       0   \n",
       "\n",
       "  How many employees does your company or organization have?  \\\n",
       "0                                             26-100           \n",
       "1                                               6-25           \n",
       "2                                               6-25           \n",
       "3                                                NaN           \n",
       "4                                               6-25           \n",
       "\n",
       "   Is your employer primarily a tech company/organization?  \\\n",
       "0                                                1.0         \n",
       "1                                                1.0         \n",
       "2                                                1.0         \n",
       "3                                                NaN         \n",
       "4                                                0.0         \n",
       "\n",
       "   Is your primary role within your company related to tech/IT?  \\\n",
       "0                                                NaN              \n",
       "1                                                NaN              \n",
       "2                                                NaN              \n",
       "3                                                NaN              \n",
       "4                                                1.0              \n",
       "\n",
       "  Does your employer provide mental health benefits as part of healthcare coverage?  \\\n",
       "0                    Not eligible for coverage / N/A                                  \n",
       "1                                                 No                                  \n",
       "2                                                 No                                  \n",
       "3                                                NaN                                  \n",
       "4                                                Yes                                  \n",
       "\n",
       "  Do you know the options for mental health care available under your employer-provided coverage?  \\\n",
       "0                                                NaN                                                \n",
       "1                                                Yes                                                \n",
       "2                                                NaN                                                \n",
       "3                                                NaN                                                \n",
       "4                                                Yes                                                \n",
       "\n",
       "  Has your employer ever formally discussed mental health (for example, as part of a wellness campaign or other official communication)?  \\\n",
       "0                                                 No                                                                                       \n",
       "1                                                Yes                                                                                       \n",
       "2                                                 No                                                                                       \n",
       "3                                                NaN                                                                                       \n",
       "4                                                 No                                                                                       \n",
       "\n",
       "  Does your employer offer resources to learn more about mental health concerns and options for seeking help?  \\\n",
       "0                                                 No                                                            \n",
       "1                                                Yes                                                            \n",
       "2                                                 No                                                            \n",
       "3                                                NaN                                                            \n",
       "4                                                 No                                                            \n",
       "\n",
       "  Is your anonymity protected if you choose to take advantage of mental health or substance abuse treatment resources provided by your employer?  \\\n",
       "0                                       I don't know                                                                                               \n",
       "1                                                Yes                                                                                               \n",
       "2                                       I don't know                                                                                               \n",
       "3                                                NaN                                                                                               \n",
       "4                                                 No                                                                                               \n",
       "\n",
       "  If a mental health issue prompted you to request a medical leave from work, asking for that leave would be:  \\\n",
       "0                                          Very easy                                                            \n",
       "1                                      Somewhat easy                                                            \n",
       "2                         Neither easy nor difficult                                                            \n",
       "3                                                NaN                                                            \n",
       "4                         Neither easy nor difficult                                                            \n",
       "\n",
       "   ...  \\\n",
       "0  ...   \n",
       "1  ...   \n",
       "2  ...   \n",
       "3  ...   \n",
       "4  ...   \n",
       "\n",
       "  If you have a mental health issue, do you feel that it interferes with your work when being treated effectively?  \\\n",
       "0                               Not applicable to me                                                                 \n",
       "1                                             Rarely                                                                 \n",
       "2                               Not applicable to me                                                                 \n",
       "3                                          Sometimes                                                                 \n",
       "4                                          Sometimes                                                                 \n",
       "\n",
       "  If you have a mental health issue, do you feel that it interferes with your work when NOT being treated effectively?  \\\n",
       "0                               Not applicable to me                                                                     \n",
       "1                                          Sometimes                                                                     \n",
       "2                               Not applicable to me                                                                     \n",
       "3                                          Sometimes                                                                     \n",
       "4                                          Sometimes                                                                     \n",
       "\n",
       "  What is your age? What is your gender? What country do you live in?  \\\n",
       "0                39                 Male               United Kingdom   \n",
       "1                29                 male     United States of America   \n",
       "2                38                Male                United Kingdom   \n",
       "3                43                 male               United Kingdom   \n",
       "4                43               Female     United States of America   \n",
       "\n",
       "  What US state or territory do you live in?  What country do you work in?  \\\n",
       "0                                        NaN                United Kingdom   \n",
       "1                                   Illinois      United States of America   \n",
       "2                                        NaN                United Kingdom   \n",
       "3                                        NaN                United Kingdom   \n",
       "4                                   Illinois      United States of America   \n",
       "\n",
       "  What US state or territory do you work in?  \\\n",
       "0                                        NaN   \n",
       "1                                   Illinois   \n",
       "2                                        NaN   \n",
       "3                                        NaN   \n",
       "4                                   Illinois   \n",
       "\n",
       "  Which of the following best describes your work position?  \\\n",
       "0                                 Back-end Developer          \n",
       "1             Back-end Developer|Front-end Developer          \n",
       "2                                 Back-end Developer          \n",
       "3                               Supervisor/Team Lead          \n",
       "4  Executive Leadership|Supervisor/Team Lead|Dev ...          \n",
       "\n",
       "  Do you work remotely?  \n",
       "0             Sometimes  \n",
       "1                 Never  \n",
       "2                Always  \n",
       "3             Sometimes  \n",
       "4             Sometimes  \n",
       "\n",
       "[5 rows x 63 columns]"
      ]
     },
     "execution_count": 2,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "# Import our mental health dataset and create a dataframe\n",
    "mental_df = pd.read_csv('./Resources/mental-heath-in-tech-2016_20161114.csv')\n",
    "mental_df.head()"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 3,
   "metadata": {},
   "outputs": [
    {
     "data": {
      "text/plain": [
       "(1433, 63)"
      ]
     },
     "execution_count": 3,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "mental_df.shape"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 4,
   "metadata": {
    "scrolled": false
   },
   "outputs": [
    {
     "data": {
      "text/plain": [
       "['Is your primary role within your company related to tech/IT?',\n",
       " 'Do you have medical coverage (private insurance or state-provided) which includes treatment of \\xa0mental health issues?',\n",
       " 'Do you know local or online resources to seek help for a mental health disorder?',\n",
       " 'If you have been diagnosed or treated for a mental health disorder, do you ever reveal this to clients or business contacts?',\n",
       " 'If you have revealed a mental health issue to a client or business contact, do you believe this has impacted you negatively?',\n",
       " 'If you have been diagnosed or treated for a mental health disorder, do you ever reveal this to coworkers or employees?',\n",
       " 'If you have revealed a mental health issue to a coworker or employee, do you believe this has impacted you negatively?',\n",
       " 'Do you believe your productivity is ever affected by a mental health issue?',\n",
       " 'If yes, what percentage of your work time (time performing primary or secondary job functions) is affected by a mental health issue?',\n",
       " 'Have your observations of how another individual who discussed a mental health disorder made you less likely to reveal a mental health issue yourself in your current workplace?',\n",
       " 'If yes, what condition(s) have you been diagnosed with?',\n",
       " 'If maybe, what condition(s) do you believe you have?',\n",
       " 'If so, what condition(s) were you diagnosed with?',\n",
       " 'What US state or territory do you live in?',\n",
       " 'What US state or territory do you work in?']"
      ]
     },
     "execution_count": 4,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "# Check for low response rate columns < 70% response rate\n",
    "low_response_columns = []\n",
    "for column in mental_df.columns:\n",
    "    if ((mental_df[column].count() / 1433) < .7):\n",
    "        low_response_columns.append(column)\n",
    "\n",
    "low_response_columns"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 5,
   "metadata": {
    "scrolled": true
   },
   "outputs": [
    {
     "data": {
      "text/plain": [
       "(1433, 48)"
      ]
     },
     "execution_count": 5,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "# drop low response columns\n",
    "mental_df.drop(columns=low_response_columns, inplace=True)\n",
    "mental_df.shape"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 6,
   "metadata": {
    "scrolled": true
   },
   "outputs": [
    {
     "data": {
      "text/plain": [
       "['Why or why not?',\n",
       " 'Why or why not?.1',\n",
       " 'Which of the following best describes your work position?']"
      ]
     },
     "execution_count": 6,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "# Check for many distinct answers indicating long for response or irrelavant data\n",
    "many_distinct_columns = []\n",
    "for column in mental_df.columns:\n",
    "    if (mental_df[column].nunique() > 10):\n",
    "        many_distinct_columns.append(column)\n",
    "    \n",
    "\n",
    "many_distinct_columns.remove('What is your age?')\n",
    "many_distinct_columns.remove('What country do you live in?')\n",
    "many_distinct_columns.remove('What country do you work in?')\n",
    "many_distinct_columns.remove('What is your gender?')\n",
    "\n",
    "\n",
    "\n",
    "many_distinct_columns "
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 7,
   "metadata": {},
   "outputs": [
    {
     "data": {
      "text/plain": [
       "(1433, 45)"
      ]
     },
     "execution_count": 7,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "# drop many distinct answers columns\n",
    "mental_df.drop(columns=many_distinct_columns, inplace=True)\n",
    "mental_df.shape"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 8,
   "metadata": {},
   "outputs": [],
   "source": [
    "# Fix gender data to be consistent\n",
    "mental_df['What is your gender?'] = mental_df['What is your gender?'].replace('male', 'Male')\n",
    "mental_df['What is your gender?'] = mental_df['What is your gender?'].replace('m', 'Male')\n",
    "mental_df['What is your gender?'] = mental_df['What is your gender?'].replace('M', 'Male')\n",
    "mental_df['What is your gender?'] = mental_df['What is your gender?'].replace('f', 'Male')\n",
    "mental_df['What is your gender?'] = mental_df['What is your gender?'].replace('F', 'Female')\n",
    "mental_df['What is your gender?'] = mental_df['What is your gender?'].replace('female', 'Female')"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 9,
   "metadata": {
    "scrolled": true
   },
   "outputs": [
    {
     "data": {
      "text/plain": [
       "Male                                       1047\n",
       "Female                                      286\n",
       "Male                                         11\n",
       "Female                                        9\n",
       "non-binary                                    4\n",
       "                                           ... \n",
       "female-bodied; no feelings about gender       1\n",
       "Other/Transfeminine                           1\n",
       "mail                                          1\n",
       "Male (trans, FtM)                             1\n",
       "cis man                                       1\n",
       "Name: What is your gender?, Length: 64, dtype: int64"
      ]
     },
     "execution_count": 9,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "# Print out gender counts\n",
    "gender_counts = mental_df['What is your gender?'].value_counts()\n",
    "gender_counts"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 10,
   "metadata": {},
   "outputs": [],
   "source": [
    "# Determine which values to replace\n",
    "replace_genders = list(gender_counts[gender_counts < 20].index)\n",
    "\n",
    "# Replace in DataFrame\n",
    "for gender in replace_genders:\n",
    "    mental_df['What is your gender?'] = mental_df['What is your gender?'].replace(gender,\"Other\")\n"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 11,
   "metadata": {},
   "outputs": [
    {
     "data": {
      "text/plain": [
       "Male      1047\n",
       "Female     286\n",
       "Other       97\n",
       "Name: What is your gender?, dtype: int64"
      ]
     },
     "execution_count": 11,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "# Print out gender counts\n",
    "gender_counts = mental_df['What is your gender?'].value_counts()\n",
    "gender_counts"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 12,
   "metadata": {},
   "outputs": [
    {
     "data": {
      "text/plain": [
       "United States of America    840\n",
       "United Kingdom              180\n",
       "Canada                       78\n",
       "Germany                      58\n",
       "Netherlands                  48\n",
       "Australia                    35\n",
       "Sweden                       19\n",
       "France                       16\n",
       "Ireland                      15\n",
       "Switzerland                  10\n",
       "Brazil                       10\n",
       "India                         9\n",
       "Russia                        9\n",
       "New Zealand                   9\n",
       "Finland                       7\n",
       "Denmark                       7\n",
       "Bulgaria                      7\n",
       "Italy                         5\n",
       "Belgium                       5\n",
       "Poland                        4\n",
       "Romania                       4\n",
       "Spain                         4\n",
       "South Africa                  4\n",
       "Austria                       4\n",
       "Czech Republic                3\n",
       "Pakistan                      3\n",
       "Chile                         3\n",
       "Norway                        3\n",
       "Bosnia and Herzegovina        2\n",
       "Japan                         2\n",
       "Israel                        2\n",
       "Colombia                      2\n",
       "Other                         2\n",
       "Lithuania                     2\n",
       "Estonia                       2\n",
       "Afghanistan                   2\n",
       "Mexico                        2\n",
       "Venezuela                     1\n",
       "Hungary                       1\n",
       "Vietnam                       1\n",
       "China                         1\n",
       "Serbia                        1\n",
       "Iran                          1\n",
       "Brunei                        1\n",
       "Taiwan                        1\n",
       "Bangladesh                    1\n",
       "Ecuador                       1\n",
       "Costa Rica                    1\n",
       "Slovakia                      1\n",
       "Greece                        1\n",
       "Argentina                     1\n",
       "Algeria                       1\n",
       "Guatemala                     1\n",
       "Name: What country do you live in?, dtype: int64"
      ]
     },
     "execution_count": 12,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "# Print out country counts\n",
    "country_counts = mental_df['What country do you live in?'].value_counts()\n",
    "country_counts"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 13,
   "metadata": {},
   "outputs": [],
   "source": [
    "# Determine which values to replace\n",
    "replace_countries = list(country_counts[country_counts < 20].index)\n",
    "\n",
    "# Replace in DataFrame\n",
    "for country in replace_countries:\n",
    "    mental_df['What country do you live in?'] = mental_df['What country do you live in?'].replace(country,\"Other\")\n"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 14,
   "metadata": {},
   "outputs": [
    {
     "data": {
      "text/plain": [
       "United States of America    840\n",
       "Other                       194\n",
       "United Kingdom              180\n",
       "Canada                       78\n",
       "Germany                      58\n",
       "Netherlands                  48\n",
       "Australia                    35\n",
       "Name: What country do you live in?, dtype: int64"
      ]
     },
     "execution_count": 14,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "# Print out country counts\n",
    "country_counts = mental_df['What country do you live in?'].value_counts()\n",
    "country_counts"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 15,
   "metadata": {},
   "outputs": [
    {
     "data": {
      "text/plain": [
       "United States of America    851\n",
       "United Kingdom              183\n",
       "Canada                       74\n",
       "Germany                      58\n",
       "Netherlands                  47\n",
       "Australia                    34\n",
       "Sweden                       20\n",
       "Ireland                      15\n",
       "France                       14\n",
       "Brazil                       10\n",
       "Switzerland                  10\n",
       "India                         9\n",
       "Russia                        9\n",
       "New Zealand                   9\n",
       "Finland                       7\n",
       "Bulgaria                      7\n",
       "Denmark                       7\n",
       "Belgium                       5\n",
       "Poland                        4\n",
       "South Africa                  4\n",
       "Austria                       4\n",
       "Spain                         3\n",
       "Czech Republic                3\n",
       "Italy                         3\n",
       "Chile                         3\n",
       "Norway                        3\n",
       "Romania                       3\n",
       "Bosnia and Herzegovina        2\n",
       "Pakistan                      2\n",
       "Colombia                      2\n",
       "Other                         2\n",
       "Mexico                        2\n",
       "Estonia                       2\n",
       "Afghanistan                   2\n",
       "Israel                        2\n",
       "Ecuador                       1\n",
       "Venezuela                     1\n",
       "Hungary                       1\n",
       "Vietnam                       1\n",
       "China                         1\n",
       "Serbia                        1\n",
       "Turkey                        1\n",
       "Iran                          1\n",
       "Bangladesh                    1\n",
       "Brunei                        1\n",
       "United Arab Emirates          1\n",
       "Costa Rica                    1\n",
       "Slovakia                      1\n",
       "Greece                        1\n",
       "Argentina                     1\n",
       "Lithuania                     1\n",
       "Guatemala                     1\n",
       "Japan                         1\n",
       "Name: What country do you work in?, dtype: int64"
      ]
     },
     "execution_count": 15,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "# Print out country counts\n",
    "country_counts = mental_df['What country do you work in?'].value_counts()\n",
    "country_counts"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 16,
   "metadata": {},
   "outputs": [],
   "source": [
    "# Determine which values to replace\n",
    "replace_countries = list(country_counts[country_counts < 20].index)\n",
    "\n",
    "# Replace in DataFrame\n",
    "for country in replace_countries:\n",
    "    mental_df['What country do you work in?'] = mental_df['What country do you work in?'].replace(country,\"Other\")\n"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 17,
   "metadata": {},
   "outputs": [
    {
     "data": {
      "text/plain": [
       "United States of America    851\n",
       "United Kingdom              183\n",
       "Other                       166\n",
       "Canada                       74\n",
       "Germany                      58\n",
       "Netherlands                  47\n",
       "Australia                    34\n",
       "Sweden                       20\n",
       "Name: What country do you work in?, dtype: int64"
      ]
     },
     "execution_count": 17,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "# Print out country counts\n",
    "country_counts = mental_df['What country do you work in?'].value_counts()\n",
    "country_counts"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 18,
   "metadata": {},
   "outputs": [],
   "source": [
    "# function for comparing two columns with a chi-square-test\n",
    "\n",
    "def chi_square(column):\n",
    "    crosstab = pd.crosstab(mental_df['Have you been diagnosed with a mental health condition by a medical professional?'], mental_df[column])\n",
    "    return stats.chi2_contingency(crosstab)[1]"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 19,
   "metadata": {},
   "outputs": [
    {
     "data": {
      "text/plain": [
       "['Are you self-employed?',\n",
       " 'How many employees does your company or organization have?',\n",
       " 'Is your employer primarily a tech company/organization?',\n",
       " 'Has your employer ever formally discussed mental health (for example, as part of a wellness campaign or other official communication)?',\n",
       " 'Is your anonymity protected if you choose to take advantage of mental health or substance abuse treatment resources provided by your employer?',\n",
       " 'Would you feel comfortable discussing a mental health disorder with your coworkers?',\n",
       " 'Would you feel comfortable discussing a mental health disorder with your direct supervisor(s)?',\n",
       " 'Did your previous employers provide resources to learn more about mental health issues and how to seek help?',\n",
       " 'Do you think that discussing a physical health issue with previous employers would have negative consequences?',\n",
       " 'Would you have been willing to discuss a mental health issue with your previous co-workers?',\n",
       " 'Would you be willing to bring up a physical health issue with a potential employer in an interview?',\n",
       " 'What is your age?',\n",
       " 'Do you work remotely?']"
      ]
     },
     "execution_count": 19,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "# run chi-square test and drop non significant columns\n",
    "\n",
    "insignificant_p_columns = []\n",
    "for column in mental_df.columns:\n",
    "    p_value = chi_square(column)\n",
    "    if p_value >= .05:\n",
    "        insignificant_p_columns.append(column)\n",
    "        \n",
    "insignificant_p_columns"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 20,
   "metadata": {},
   "outputs": [
    {
     "data": {
      "text/plain": [
       "(1433, 32)"
      ]
     },
     "execution_count": 20,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "# drop insignificant p value columns\n",
    "mental_df.drop(columns=insignificant_p_columns, inplace=True)\n",
    "mental_df.shape"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 21,
   "metadata": {
    "scrolled": false
   },
   "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>Does your employer provide mental health benefits as part of healthcare coverage?</th>\n",
       "      <th>Do you know the options for mental health care available under your employer-provided coverage?</th>\n",
       "      <th>Does your employer offer resources to learn more about mental health concerns and options for seeking help?</th>\n",
       "      <th>If a mental health issue prompted you to request a medical leave from work, asking for that leave would be:</th>\n",
       "      <th>Do you think that discussing a mental health disorder with your employer would have negative consequences?</th>\n",
       "      <th>Do you think that discussing a physical health issue with your employer would have negative consequences?</th>\n",
       "      <th>Do you feel that your employer takes mental health as seriously as physical health?</th>\n",
       "      <th>Have you heard of or observed negative consequences for co-workers who have been open about mental health issues in your workplace?</th>\n",
       "      <th>Do you have previous employers?</th>\n",
       "      <th>Have your previous employers provided mental health benefits?</th>\n",
       "      <th>...</th>\n",
       "      <th>Do you have a family history of mental illness?</th>\n",
       "      <th>Have you had a mental health disorder in the past?</th>\n",
       "      <th>Do you currently have a mental health disorder?</th>\n",
       "      <th>Have you been diagnosed with a mental health condition by a medical professional?</th>\n",
       "      <th>Have you ever sought treatment for a mental health issue from a mental health professional?</th>\n",
       "      <th>If you have a mental health issue, do you feel that it interferes with your work when being treated effectively?</th>\n",
       "      <th>If you have a mental health issue, do you feel that it interferes with your work when NOT being treated effectively?</th>\n",
       "      <th>What is your gender?</th>\n",
       "      <th>What country do you live in?</th>\n",
       "      <th>What country do you work in?</th>\n",
       "    </tr>\n",
       "  </thead>\n",
       "  <tbody>\n",
       "    <tr>\n",
       "      <th>0</th>\n",
       "      <td>Not eligible for coverage / N/A</td>\n",
       "      <td>NaN</td>\n",
       "      <td>No</td>\n",
       "      <td>Very easy</td>\n",
       "      <td>No</td>\n",
       "      <td>No</td>\n",
       "      <td>I don't know</td>\n",
       "      <td>No</td>\n",
       "      <td>1</td>\n",
       "      <td>No, none did</td>\n",
       "      <td>...</td>\n",
       "      <td>No</td>\n",
       "      <td>Yes</td>\n",
       "      <td>No</td>\n",
       "      <td>Yes</td>\n",
       "      <td>0</td>\n",
       "      <td>Not applicable to me</td>\n",
       "      <td>Not applicable to me</td>\n",
       "      <td>Male</td>\n",
       "      <td>United Kingdom</td>\n",
       "      <td>United Kingdom</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>1</th>\n",
       "      <td>No</td>\n",
       "      <td>Yes</td>\n",
       "      <td>Yes</td>\n",
       "      <td>Somewhat easy</td>\n",
       "      <td>No</td>\n",
       "      <td>No</td>\n",
       "      <td>Yes</td>\n",
       "      <td>No</td>\n",
       "      <td>1</td>\n",
       "      <td>Yes, they all did</td>\n",
       "      <td>...</td>\n",
       "      <td>Yes</td>\n",
       "      <td>Yes</td>\n",
       "      <td>Yes</td>\n",
       "      <td>Yes</td>\n",
       "      <td>1</td>\n",
       "      <td>Rarely</td>\n",
       "      <td>Sometimes</td>\n",
       "      <td>Male</td>\n",
       "      <td>United States of America</td>\n",
       "      <td>United States of America</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>2</th>\n",
       "      <td>No</td>\n",
       "      <td>NaN</td>\n",
       "      <td>No</td>\n",
       "      <td>Neither easy nor difficult</td>\n",
       "      <td>Maybe</td>\n",
       "      <td>No</td>\n",
       "      <td>I don't know</td>\n",
       "      <td>No</td>\n",
       "      <td>1</td>\n",
       "      <td>No, none did</td>\n",
       "      <td>...</td>\n",
       "      <td>No</td>\n",
       "      <td>Maybe</td>\n",
       "      <td>No</td>\n",
       "      <td>No</td>\n",
       "      <td>1</td>\n",
       "      <td>Not applicable to me</td>\n",
       "      <td>Not applicable to me</td>\n",
       "      <td>Other</td>\n",
       "      <td>United Kingdom</td>\n",
       "      <td>United Kingdom</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>3</th>\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>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>1</td>\n",
       "      <td>Some did</td>\n",
       "      <td>...</td>\n",
       "      <td>No</td>\n",
       "      <td>Yes</td>\n",
       "      <td>Yes</td>\n",
       "      <td>Yes</td>\n",
       "      <td>1</td>\n",
       "      <td>Sometimes</td>\n",
       "      <td>Sometimes</td>\n",
       "      <td>Male</td>\n",
       "      <td>United Kingdom</td>\n",
       "      <td>United Kingdom</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>4</th>\n",
       "      <td>Yes</td>\n",
       "      <td>Yes</td>\n",
       "      <td>No</td>\n",
       "      <td>Neither easy nor difficult</td>\n",
       "      <td>Yes</td>\n",
       "      <td>Maybe</td>\n",
       "      <td>No</td>\n",
       "      <td>No</td>\n",
       "      <td>1</td>\n",
       "      <td>I don't know</td>\n",
       "      <td>...</td>\n",
       "      <td>Yes</td>\n",
       "      <td>Yes</td>\n",
       "      <td>Yes</td>\n",
       "      <td>Yes</td>\n",
       "      <td>1</td>\n",
       "      <td>Sometimes</td>\n",
       "      <td>Sometimes</td>\n",
       "      <td>Female</td>\n",
       "      <td>United States of America</td>\n",
       "      <td>United States of America</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",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\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>1428</th>\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>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>1</td>\n",
       "      <td>Yes, they all did</td>\n",
       "      <td>...</td>\n",
       "      <td>Yes</td>\n",
       "      <td>No</td>\n",
       "      <td>No</td>\n",
       "      <td>No</td>\n",
       "      <td>1</td>\n",
       "      <td>Not applicable to me</td>\n",
       "      <td>Not applicable to me</td>\n",
       "      <td>Other</td>\n",
       "      <td>United States of America</td>\n",
       "      <td>United States of America</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>1429</th>\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>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>0</td>\n",
       "      <td>NaN</td>\n",
       "      <td>...</td>\n",
       "      <td>Yes</td>\n",
       "      <td>No</td>\n",
       "      <td>No</td>\n",
       "      <td>Yes</td>\n",
       "      <td>0</td>\n",
       "      <td>Sometimes</td>\n",
       "      <td>Often</td>\n",
       "      <td>Other</td>\n",
       "      <td>United States of America</td>\n",
       "      <td>Other</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>1430</th>\n",
       "      <td>Yes</td>\n",
       "      <td>Yes</td>\n",
       "      <td>Yes</td>\n",
       "      <td>Somewhat difficult</td>\n",
       "      <td>Maybe</td>\n",
       "      <td>Maybe</td>\n",
       "      <td>I don't know</td>\n",
       "      <td>Yes</td>\n",
       "      <td>1</td>\n",
       "      <td>Some did</td>\n",
       "      <td>...</td>\n",
       "      <td>Yes</td>\n",
       "      <td>Yes</td>\n",
       "      <td>Maybe</td>\n",
       "      <td>Yes</td>\n",
       "      <td>1</td>\n",
       "      <td>Rarely</td>\n",
       "      <td>Sometimes</td>\n",
       "      <td>Male</td>\n",
       "      <td>United States of America</td>\n",
       "      <td>United States of America</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>1431</th>\n",
       "      <td>I don't know</td>\n",
       "      <td>I am not sure</td>\n",
       "      <td>Yes</td>\n",
       "      <td>Somewhat difficult</td>\n",
       "      <td>Maybe</td>\n",
       "      <td>No</td>\n",
       "      <td>No</td>\n",
       "      <td>No</td>\n",
       "      <td>1</td>\n",
       "      <td>No, none did</td>\n",
       "      <td>...</td>\n",
       "      <td>Yes</td>\n",
       "      <td>Maybe</td>\n",
       "      <td>Yes</td>\n",
       "      <td>Yes</td>\n",
       "      <td>0</td>\n",
       "      <td>Sometimes</td>\n",
       "      <td>Often</td>\n",
       "      <td>Female</td>\n",
       "      <td>United States of America</td>\n",
       "      <td>United States of America</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>1432</th>\n",
       "      <td>Yes</td>\n",
       "      <td>No</td>\n",
       "      <td>No</td>\n",
       "      <td>Very difficult</td>\n",
       "      <td>Maybe</td>\n",
       "      <td>No</td>\n",
       "      <td>No</td>\n",
       "      <td>No</td>\n",
       "      <td>0</td>\n",
       "      <td>NaN</td>\n",
       "      <td>...</td>\n",
       "      <td>I don't know</td>\n",
       "      <td>Yes</td>\n",
       "      <td>Yes</td>\n",
       "      <td>No</td>\n",
       "      <td>0</td>\n",
       "      <td>Not applicable to me</td>\n",
       "      <td>Often</td>\n",
       "      <td>Other</td>\n",
       "      <td>Canada</td>\n",
       "      <td>Canada</td>\n",
       "    </tr>\n",
       "  </tbody>\n",
       "</table>\n",
       "<p>1433 rows × 32 columns</p>\n",
       "</div>"
      ],
      "text/plain": [
       "     Does your employer provide mental health benefits as part of healthcare coverage?  \\\n",
       "0                       Not eligible for coverage / N/A                                  \n",
       "1                                                    No                                  \n",
       "2                                                    No                                  \n",
       "3                                                   NaN                                  \n",
       "4                                                   Yes                                  \n",
       "...                                                 ...                                  \n",
       "1428                                                NaN                                  \n",
       "1429                                                NaN                                  \n",
       "1430                                                Yes                                  \n",
       "1431                                       I don't know                                  \n",
       "1432                                                Yes                                  \n",
       "\n",
       "     Do you know the options for mental health care available under your employer-provided coverage?  \\\n",
       "0                                                   NaN                                                \n",
       "1                                                   Yes                                                \n",
       "2                                                   NaN                                                \n",
       "3                                                   NaN                                                \n",
       "4                                                   Yes                                                \n",
       "...                                                 ...                                                \n",
       "1428                                                NaN                                                \n",
       "1429                                                NaN                                                \n",
       "1430                                                Yes                                                \n",
       "1431                                      I am not sure                                                \n",
       "1432                                                 No                                                \n",
       "\n",
       "     Does your employer offer resources to learn more about mental health concerns and options for seeking help?  \\\n",
       "0                                                    No                                                            \n",
       "1                                                   Yes                                                            \n",
       "2                                                    No                                                            \n",
       "3                                                   NaN                                                            \n",
       "4                                                    No                                                            \n",
       "...                                                 ...                                                            \n",
       "1428                                                NaN                                                            \n",
       "1429                                                NaN                                                            \n",
       "1430                                                Yes                                                            \n",
       "1431                                                Yes                                                            \n",
       "1432                                                 No                                                            \n",
       "\n",
       "     If a mental health issue prompted you to request a medical leave from work, asking for that leave would be:  \\\n",
       "0                                             Very easy                                                            \n",
       "1                                         Somewhat easy                                                            \n",
       "2                            Neither easy nor difficult                                                            \n",
       "3                                                   NaN                                                            \n",
       "4                            Neither easy nor difficult                                                            \n",
       "...                                                 ...                                                            \n",
       "1428                                                NaN                                                            \n",
       "1429                                                NaN                                                            \n",
       "1430                                 Somewhat difficult                                                            \n",
       "1431                                 Somewhat difficult                                                            \n",
       "1432                                     Very difficult                                                            \n",
       "\n",
       "     Do you think that discussing a mental health disorder with your employer would have negative consequences?  \\\n",
       "0                                                    No                                                           \n",
       "1                                                    No                                                           \n",
       "2                                                 Maybe                                                           \n",
       "3                                                   NaN                                                           \n",
       "4                                                   Yes                                                           \n",
       "...                                                 ...                                                           \n",
       "1428                                                NaN                                                           \n",
       "1429                                                NaN                                                           \n",
       "1430                                              Maybe                                                           \n",
       "1431                                              Maybe                                                           \n",
       "1432                                              Maybe                                                           \n",
       "\n",
       "     Do you think that discussing a physical health issue with your employer would have negative consequences?  \\\n",
       "0                                                    No                                                          \n",
       "1                                                    No                                                          \n",
       "2                                                    No                                                          \n",
       "3                                                   NaN                                                          \n",
       "4                                                 Maybe                                                          \n",
       "...                                                 ...                                                          \n",
       "1428                                                NaN                                                          \n",
       "1429                                                NaN                                                          \n",
       "1430                                              Maybe                                                          \n",
       "1431                                                 No                                                          \n",
       "1432                                                 No                                                          \n",
       "\n",
       "     Do you feel that your employer takes mental health as seriously as physical health?  \\\n",
       "0                                          I don't know                                    \n",
       "1                                                   Yes                                    \n",
       "2                                          I don't know                                    \n",
       "3                                                   NaN                                    \n",
       "4                                                    No                                    \n",
       "...                                                 ...                                    \n",
       "1428                                                NaN                                    \n",
       "1429                                                NaN                                    \n",
       "1430                                       I don't know                                    \n",
       "1431                                                 No                                    \n",
       "1432                                                 No                                    \n",
       "\n",
       "     Have you heard of or observed negative consequences for co-workers who have been open about mental health issues in your workplace?  \\\n",
       "0                                                    No                                                                                    \n",
       "1                                                    No                                                                                    \n",
       "2                                                    No                                                                                    \n",
       "3                                                   NaN                                                                                    \n",
       "4                                                    No                                                                                    \n",
       "...                                                 ...                                                                                    \n",
       "1428                                                NaN                                                                                    \n",
       "1429                                                NaN                                                                                    \n",
       "1430                                                Yes                                                                                    \n",
       "1431                                                 No                                                                                    \n",
       "1432                                                 No                                                                                    \n",
       "\n",
       "      Do you have previous employers?  \\\n",
       "0                                   1   \n",
       "1                                   1   \n",
       "2                                   1   \n",
       "3                                   1   \n",
       "4                                   1   \n",
       "...                               ...   \n",
       "1428                                1   \n",
       "1429                                0   \n",
       "1430                                1   \n",
       "1431                                1   \n",
       "1432                                0   \n",
       "\n",
       "     Have your previous employers provided mental health benefits?  ...  \\\n",
       "0                                          No, none did             ...   \n",
       "1                                     Yes, they all did             ...   \n",
       "2                                          No, none did             ...   \n",
       "3                                              Some did             ...   \n",
       "4                                          I don't know             ...   \n",
       "...                                                 ...             ...   \n",
       "1428                                  Yes, they all did             ...   \n",
       "1429                                                NaN             ...   \n",
       "1430                                           Some did             ...   \n",
       "1431                                       No, none did             ...   \n",
       "1432                                                NaN             ...   \n",
       "\n",
       "     Do you have a family history of mental illness?  \\\n",
       "0                                                 No   \n",
       "1                                                Yes   \n",
       "2                                                 No   \n",
       "3                                                 No   \n",
       "4                                                Yes   \n",
       "...                                              ...   \n",
       "1428                                             Yes   \n",
       "1429                                             Yes   \n",
       "1430                                             Yes   \n",
       "1431                                             Yes   \n",
       "1432                                    I don't know   \n",
       "\n",
       "     Have you had a mental health disorder in the past?  \\\n",
       "0                                                   Yes   \n",
       "1                                                   Yes   \n",
       "2                                                 Maybe   \n",
       "3                                                   Yes   \n",
       "4                                                   Yes   \n",
       "...                                                 ...   \n",
       "1428                                                 No   \n",
       "1429                                                 No   \n",
       "1430                                                Yes   \n",
       "1431                                              Maybe   \n",
       "1432                                                Yes   \n",
       "\n",
       "     Do you currently have a mental health disorder?  \\\n",
       "0                                                 No   \n",
       "1                                                Yes   \n",
       "2                                                 No   \n",
       "3                                                Yes   \n",
       "4                                                Yes   \n",
       "...                                              ...   \n",
       "1428                                              No   \n",
       "1429                                              No   \n",
       "1430                                           Maybe   \n",
       "1431                                             Yes   \n",
       "1432                                             Yes   \n",
       "\n",
       "     Have you been diagnosed with a mental health condition by a medical professional?  \\\n",
       "0                                                   Yes                                  \n",
       "1                                                   Yes                                  \n",
       "2                                                    No                                  \n",
       "3                                                   Yes                                  \n",
       "4                                                   Yes                                  \n",
       "...                                                 ...                                  \n",
       "1428                                                 No                                  \n",
       "1429                                                Yes                                  \n",
       "1430                                                Yes                                  \n",
       "1431                                                Yes                                  \n",
       "1432                                                 No                                  \n",
       "\n",
       "     Have you ever sought treatment for a mental health issue from a mental health professional?  \\\n",
       "0                                                     0                                            \n",
       "1                                                     1                                            \n",
       "2                                                     1                                            \n",
       "3                                                     1                                            \n",
       "4                                                     1                                            \n",
       "...                                                 ...                                            \n",
       "1428                                                  1                                            \n",
       "1429                                                  0                                            \n",
       "1430                                                  1                                            \n",
       "1431                                                  0                                            \n",
       "1432                                                  0                                            \n",
       "\n",
       "     If you have a mental health issue, do you feel that it interferes with your work when being treated effectively?  \\\n",
       "0                                  Not applicable to me                                                                 \n",
       "1                                                Rarely                                                                 \n",
       "2                                  Not applicable to me                                                                 \n",
       "3                                             Sometimes                                                                 \n",
       "4                                             Sometimes                                                                 \n",
       "...                                                 ...                                                                 \n",
       "1428                               Not applicable to me                                                                 \n",
       "1429                                          Sometimes                                                                 \n",
       "1430                                             Rarely                                                                 \n",
       "1431                                          Sometimes                                                                 \n",
       "1432                               Not applicable to me                                                                 \n",
       "\n",
       "     If you have a mental health issue, do you feel that it interferes with your work when NOT being treated effectively?  \\\n",
       "0                                  Not applicable to me                                                                     \n",
       "1                                             Sometimes                                                                     \n",
       "2                                  Not applicable to me                                                                     \n",
       "3                                             Sometimes                                                                     \n",
       "4                                             Sometimes                                                                     \n",
       "...                                                 ...                                                                     \n",
       "1428                               Not applicable to me                                                                     \n",
       "1429                                              Often                                                                     \n",
       "1430                                          Sometimes                                                                     \n",
       "1431                                              Often                                                                     \n",
       "1432                                              Often                                                                     \n",
       "\n",
       "     What is your gender? What country do you live in?  \\\n",
       "0                    Male               United Kingdom   \n",
       "1                    Male     United States of America   \n",
       "2                   Other               United Kingdom   \n",
       "3                    Male               United Kingdom   \n",
       "4                  Female     United States of America   \n",
       "...                   ...                          ...   \n",
       "1428                Other     United States of America   \n",
       "1429                Other     United States of America   \n",
       "1430                 Male     United States of America   \n",
       "1431               Female     United States of America   \n",
       "1432                Other                       Canada   \n",
       "\n",
       "     What country do you work in?  \n",
       "0                  United Kingdom  \n",
       "1        United States of America  \n",
       "2                  United Kingdom  \n",
       "3                  United Kingdom  \n",
       "4        United States of America  \n",
       "...                           ...  \n",
       "1428     United States of America  \n",
       "1429                        Other  \n",
       "1430     United States of America  \n",
       "1431     United States of America  \n",
       "1432                       Canada  \n",
       "\n",
       "[1433 rows x 32 columns]"
      ]
     },
     "execution_count": 21,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "mental_df"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 22,
   "metadata": {},
   "outputs": [],
   "source": [
    "mental_df.dropna(inplace=True)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 23,
   "metadata": {},
   "outputs": [
    {
     "data": {
      "text/plain": [
       "(861, 32)"
      ]
     },
     "execution_count": 23,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "mental_df.shape"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 24,
   "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>Does your employer provide mental health benefits as part of healthcare coverage?</th>\n",
       "      <th>Do you know the options for mental health care available under your employer-provided coverage?</th>\n",
       "      <th>Does your employer offer resources to learn more about mental health concerns and options for seeking help?</th>\n",
       "      <th>If a mental health issue prompted you to request a medical leave from work, asking for that leave would be:</th>\n",
       "      <th>Do you think that discussing a mental health disorder with your employer would have negative consequences?</th>\n",
       "      <th>Do you think that discussing a physical health issue with your employer would have negative consequences?</th>\n",
       "      <th>Do you feel that your employer takes mental health as seriously as physical health?</th>\n",
       "      <th>Have you heard of or observed negative consequences for co-workers who have been open about mental health issues in your workplace?</th>\n",
       "      <th>Do you have previous employers?</th>\n",
       "      <th>Have your previous employers provided mental health benefits?</th>\n",
       "      <th>...</th>\n",
       "      <th>Do you have a family history of mental illness?</th>\n",
       "      <th>Have you had a mental health disorder in the past?</th>\n",
       "      <th>Do you currently have a mental health disorder?</th>\n",
       "      <th>Have you been diagnosed with a mental health condition by a medical professional?</th>\n",
       "      <th>Have you ever sought treatment for a mental health issue from a mental health professional?</th>\n",
       "      <th>If you have a mental health issue, do you feel that it interferes with your work when being treated effectively?</th>\n",
       "      <th>If you have a mental health issue, do you feel that it interferes with your work when NOT being treated effectively?</th>\n",
       "      <th>What is your gender?</th>\n",
       "      <th>What country do you live in?</th>\n",
       "      <th>What country do you work in?</th>\n",
       "    </tr>\n",
       "  </thead>\n",
       "  <tbody>\n",
       "    <tr>\n",
       "      <th>1</th>\n",
       "      <td>No</td>\n",
       "      <td>Yes</td>\n",
       "      <td>Yes</td>\n",
       "      <td>Somewhat easy</td>\n",
       "      <td>No</td>\n",
       "      <td>No</td>\n",
       "      <td>Yes</td>\n",
       "      <td>No</td>\n",
       "      <td>1</td>\n",
       "      <td>Yes, they all did</td>\n",
       "      <td>...</td>\n",
       "      <td>Yes</td>\n",
       "      <td>Yes</td>\n",
       "      <td>Yes</td>\n",
       "      <td>Yes</td>\n",
       "      <td>1</td>\n",
       "      <td>Rarely</td>\n",
       "      <td>Sometimes</td>\n",
       "      <td>Male</td>\n",
       "      <td>United States of America</td>\n",
       "      <td>United States of America</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>4</th>\n",
       "      <td>Yes</td>\n",
       "      <td>Yes</td>\n",
       "      <td>No</td>\n",
       "      <td>Neither easy nor difficult</td>\n",
       "      <td>Yes</td>\n",
       "      <td>Maybe</td>\n",
       "      <td>No</td>\n",
       "      <td>No</td>\n",
       "      <td>1</td>\n",
       "      <td>I don't know</td>\n",
       "      <td>...</td>\n",
       "      <td>Yes</td>\n",
       "      <td>Yes</td>\n",
       "      <td>Yes</td>\n",
       "      <td>Yes</td>\n",
       "      <td>1</td>\n",
       "      <td>Sometimes</td>\n",
       "      <td>Sometimes</td>\n",
       "      <td>Female</td>\n",
       "      <td>United States of America</td>\n",
       "      <td>United States of America</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>5</th>\n",
       "      <td>Yes</td>\n",
       "      <td>I am not sure</td>\n",
       "      <td>Yes</td>\n",
       "      <td>Somewhat easy</td>\n",
       "      <td>Yes</td>\n",
       "      <td>Yes</td>\n",
       "      <td>No</td>\n",
       "      <td>Yes</td>\n",
       "      <td>1</td>\n",
       "      <td>No, none did</td>\n",
       "      <td>...</td>\n",
       "      <td>No</td>\n",
       "      <td>No</td>\n",
       "      <td>Yes</td>\n",
       "      <td>No</td>\n",
       "      <td>1</td>\n",
       "      <td>Not applicable to me</td>\n",
       "      <td>Often</td>\n",
       "      <td>Male</td>\n",
       "      <td>United Kingdom</td>\n",
       "      <td>United Kingdom</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>6</th>\n",
       "      <td>I don't know</td>\n",
       "      <td>No</td>\n",
       "      <td>No</td>\n",
       "      <td>Somewhat easy</td>\n",
       "      <td>No</td>\n",
       "      <td>No</td>\n",
       "      <td>Yes</td>\n",
       "      <td>No</td>\n",
       "      <td>1</td>\n",
       "      <td>Some did</td>\n",
       "      <td>...</td>\n",
       "      <td>No</td>\n",
       "      <td>No</td>\n",
       "      <td>No</td>\n",
       "      <td>No</td>\n",
       "      <td>0</td>\n",
       "      <td>Not applicable to me</td>\n",
       "      <td>Not applicable to me</td>\n",
       "      <td>Male</td>\n",
       "      <td>United States of America</td>\n",
       "      <td>United States of America</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>7</th>\n",
       "      <td>Yes</td>\n",
       "      <td>Yes</td>\n",
       "      <td>Yes</td>\n",
       "      <td>Very easy</td>\n",
       "      <td>No</td>\n",
       "      <td>No</td>\n",
       "      <td>I don't know</td>\n",
       "      <td>No</td>\n",
       "      <td>1</td>\n",
       "      <td>Some did</td>\n",
       "      <td>...</td>\n",
       "      <td>Yes</td>\n",
       "      <td>Yes</td>\n",
       "      <td>Yes</td>\n",
       "      <td>Yes</td>\n",
       "      <td>1</td>\n",
       "      <td>Sometimes</td>\n",
       "      <td>Often</td>\n",
       "      <td>Female</td>\n",
       "      <td>United States of America</td>\n",
       "      <td>United States of America</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",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\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>1425</th>\n",
       "      <td>Yes</td>\n",
       "      <td>I am not sure</td>\n",
       "      <td>Yes</td>\n",
       "      <td>Somewhat easy</td>\n",
       "      <td>No</td>\n",
       "      <td>No</td>\n",
       "      <td>Yes</td>\n",
       "      <td>No</td>\n",
       "      <td>1</td>\n",
       "      <td>Some did</td>\n",
       "      <td>...</td>\n",
       "      <td>I don't know</td>\n",
       "      <td>Maybe</td>\n",
       "      <td>Maybe</td>\n",
       "      <td>No</td>\n",
       "      <td>0</td>\n",
       "      <td>Rarely</td>\n",
       "      <td>Rarely</td>\n",
       "      <td>Male</td>\n",
       "      <td>Canada</td>\n",
       "      <td>Canada</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>1426</th>\n",
       "      <td>I don't know</td>\n",
       "      <td>I am not sure</td>\n",
       "      <td>I don't know</td>\n",
       "      <td>Somewhat easy</td>\n",
       "      <td>Maybe</td>\n",
       "      <td>No</td>\n",
       "      <td>I don't know</td>\n",
       "      <td>No</td>\n",
       "      <td>1</td>\n",
       "      <td>I don't know</td>\n",
       "      <td>...</td>\n",
       "      <td>I don't know</td>\n",
       "      <td>Yes</td>\n",
       "      <td>Yes</td>\n",
       "      <td>Yes</td>\n",
       "      <td>1</td>\n",
       "      <td>Rarely</td>\n",
       "      <td>Often</td>\n",
       "      <td>Female</td>\n",
       "      <td>Canada</td>\n",
       "      <td>Canada</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>1427</th>\n",
       "      <td>Yes</td>\n",
       "      <td>No</td>\n",
       "      <td>No</td>\n",
       "      <td>Somewhat easy</td>\n",
       "      <td>No</td>\n",
       "      <td>No</td>\n",
       "      <td>Yes</td>\n",
       "      <td>No</td>\n",
       "      <td>1</td>\n",
       "      <td>Some did</td>\n",
       "      <td>...</td>\n",
       "      <td>Yes</td>\n",
       "      <td>Yes</td>\n",
       "      <td>Yes</td>\n",
       "      <td>Yes</td>\n",
       "      <td>1</td>\n",
       "      <td>Rarely</td>\n",
       "      <td>Often</td>\n",
       "      <td>Female</td>\n",
       "      <td>United States of America</td>\n",
       "      <td>United States of America</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>1430</th>\n",
       "      <td>Yes</td>\n",
       "      <td>Yes</td>\n",
       "      <td>Yes</td>\n",
       "      <td>Somewhat difficult</td>\n",
       "      <td>Maybe</td>\n",
       "      <td>Maybe</td>\n",
       "      <td>I don't know</td>\n",
       "      <td>Yes</td>\n",
       "      <td>1</td>\n",
       "      <td>Some did</td>\n",
       "      <td>...</td>\n",
       "      <td>Yes</td>\n",
       "      <td>Yes</td>\n",
       "      <td>Maybe</td>\n",
       "      <td>Yes</td>\n",
       "      <td>1</td>\n",
       "      <td>Rarely</td>\n",
       "      <td>Sometimes</td>\n",
       "      <td>Male</td>\n",
       "      <td>United States of America</td>\n",
       "      <td>United States of America</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>1431</th>\n",
       "      <td>I don't know</td>\n",
       "      <td>I am not sure</td>\n",
       "      <td>Yes</td>\n",
       "      <td>Somewhat difficult</td>\n",
       "      <td>Maybe</td>\n",
       "      <td>No</td>\n",
       "      <td>No</td>\n",
       "      <td>No</td>\n",
       "      <td>1</td>\n",
       "      <td>No, none did</td>\n",
       "      <td>...</td>\n",
       "      <td>Yes</td>\n",
       "      <td>Maybe</td>\n",
       "      <td>Yes</td>\n",
       "      <td>Yes</td>\n",
       "      <td>0</td>\n",
       "      <td>Sometimes</td>\n",
       "      <td>Often</td>\n",
       "      <td>Female</td>\n",
       "      <td>United States of America</td>\n",
       "      <td>United States of America</td>\n",
       "    </tr>\n",
       "  </tbody>\n",
       "</table>\n",
       "<p>861 rows × 32 columns</p>\n",
       "</div>"
      ],
      "text/plain": [
       "     Does your employer provide mental health benefits as part of healthcare coverage?  \\\n",
       "1                                                    No                                  \n",
       "4                                                   Yes                                  \n",
       "5                                                   Yes                                  \n",
       "6                                          I don't know                                  \n",
       "7                                                   Yes                                  \n",
       "...                                                 ...                                  \n",
       "1425                                                Yes                                  \n",
       "1426                                       I don't know                                  \n",
       "1427                                                Yes                                  \n",
       "1430                                                Yes                                  \n",
       "1431                                       I don't know                                  \n",
       "\n",
       "     Do you know the options for mental health care available under your employer-provided coverage?  \\\n",
       "1                                                   Yes                                                \n",
       "4                                                   Yes                                                \n",
       "5                                         I am not sure                                                \n",
       "6                                                    No                                                \n",
       "7                                                   Yes                                                \n",
       "...                                                 ...                                                \n",
       "1425                                      I am not sure                                                \n",
       "1426                                      I am not sure                                                \n",
       "1427                                                 No                                                \n",
       "1430                                                Yes                                                \n",
       "1431                                      I am not sure                                                \n",
       "\n",
       "     Does your employer offer resources to learn more about mental health concerns and options for seeking help?  \\\n",
       "1                                                   Yes                                                            \n",
       "4                                                    No                                                            \n",
       "5                                                   Yes                                                            \n",
       "6                                                    No                                                            \n",
       "7                                                   Yes                                                            \n",
       "...                                                 ...                                                            \n",
       "1425                                                Yes                                                            \n",
       "1426                                       I don't know                                                            \n",
       "1427                                                 No                                                            \n",
       "1430                                                Yes                                                            \n",
       "1431                                                Yes                                                            \n",
       "\n",
       "     If a mental health issue prompted you to request a medical leave from work, asking for that leave would be:  \\\n",
       "1                                         Somewhat easy                                                            \n",
       "4                            Neither easy nor difficult                                                            \n",
       "5                                         Somewhat easy                                                            \n",
       "6                                         Somewhat easy                                                            \n",
       "7                                             Very easy                                                            \n",
       "...                                                 ...                                                            \n",
       "1425                                      Somewhat easy                                                            \n",
       "1426                                      Somewhat easy                                                            \n",
       "1427                                      Somewhat easy                                                            \n",
       "1430                                 Somewhat difficult                                                            \n",
       "1431                                 Somewhat difficult                                                            \n",
       "\n",
       "     Do you think that discussing a mental health disorder with your employer would have negative consequences?  \\\n",
       "1                                                    No                                                           \n",
       "4                                                   Yes                                                           \n",
       "5                                                   Yes                                                           \n",
       "6                                                    No                                                           \n",
       "7                                                    No                                                           \n",
       "...                                                 ...                                                           \n",
       "1425                                                 No                                                           \n",
       "1426                                              Maybe                                                           \n",
       "1427                                                 No                                                           \n",
       "1430                                              Maybe                                                           \n",
       "1431                                              Maybe                                                           \n",
       "\n",
       "     Do you think that discussing a physical health issue with your employer would have negative consequences?  \\\n",
       "1                                                    No                                                          \n",
       "4                                                 Maybe                                                          \n",
       "5                                                   Yes                                                          \n",
       "6                                                    No                                                          \n",
       "7                                                    No                                                          \n",
       "...                                                 ...                                                          \n",
       "1425                                                 No                                                          \n",
       "1426                                                 No                                                          \n",
       "1427                                                 No                                                          \n",
       "1430                                              Maybe                                                          \n",
       "1431                                                 No                                                          \n",
       "\n",
       "     Do you feel that your employer takes mental health as seriously as physical health?  \\\n",
       "1                                                   Yes                                    \n",
       "4                                                    No                                    \n",
       "5                                                    No                                    \n",
       "6                                                   Yes                                    \n",
       "7                                          I don't know                                    \n",
       "...                                                 ...                                    \n",
       "1425                                                Yes                                    \n",
       "1426                                       I don't know                                    \n",
       "1427                                                Yes                                    \n",
       "1430                                       I don't know                                    \n",
       "1431                                                 No                                    \n",
       "\n",
       "     Have you heard of or observed negative consequences for co-workers who have been open about mental health issues in your workplace?  \\\n",
       "1                                                    No                                                                                    \n",
       "4                                                    No                                                                                    \n",
       "5                                                   Yes                                                                                    \n",
       "6                                                    No                                                                                    \n",
       "7                                                    No                                                                                    \n",
       "...                                                 ...                                                                                    \n",
       "1425                                                 No                                                                                    \n",
       "1426                                                 No                                                                                    \n",
       "1427                                                 No                                                                                    \n",
       "1430                                                Yes                                                                                    \n",
       "1431                                                 No                                                                                    \n",
       "\n",
       "      Do you have previous employers?  \\\n",
       "1                                   1   \n",
       "4                                   1   \n",
       "5                                   1   \n",
       "6                                   1   \n",
       "7                                   1   \n",
       "...                               ...   \n",
       "1425                                1   \n",
       "1426                                1   \n",
       "1427                                1   \n",
       "1430                                1   \n",
       "1431                                1   \n",
       "\n",
       "     Have your previous employers provided mental health benefits?  ...  \\\n",
       "1                                     Yes, they all did             ...   \n",
       "4                                          I don't know             ...   \n",
       "5                                          No, none did             ...   \n",
       "6                                              Some did             ...   \n",
       "7                                              Some did             ...   \n",
       "...                                                 ...             ...   \n",
       "1425                                           Some did             ...   \n",
       "1426                                       I don't know             ...   \n",
       "1427                                           Some did             ...   \n",
       "1430                                           Some did             ...   \n",
       "1431                                       No, none did             ...   \n",
       "\n",
       "     Do you have a family history of mental illness?  \\\n",
       "1                                                Yes   \n",
       "4                                                Yes   \n",
       "5                                                 No   \n",
       "6                                                 No   \n",
       "7                                                Yes   \n",
       "...                                              ...   \n",
       "1425                                    I don't know   \n",
       "1426                                    I don't know   \n",
       "1427                                             Yes   \n",
       "1430                                             Yes   \n",
       "1431                                             Yes   \n",
       "\n",
       "     Have you had a mental health disorder in the past?  \\\n",
       "1                                                   Yes   \n",
       "4                                                   Yes   \n",
       "5                                                    No   \n",
       "6                                                    No   \n",
       "7                                                   Yes   \n",
       "...                                                 ...   \n",
       "1425                                              Maybe   \n",
       "1426                                                Yes   \n",
       "1427                                                Yes   \n",
       "1430                                                Yes   \n",
       "1431                                              Maybe   \n",
       "\n",
       "     Do you currently have a mental health disorder?  \\\n",
       "1                                                Yes   \n",
       "4                                                Yes   \n",
       "5                                                Yes   \n",
       "6                                                 No   \n",
       "7                                                Yes   \n",
       "...                                              ...   \n",
       "1425                                           Maybe   \n",
       "1426                                             Yes   \n",
       "1427                                             Yes   \n",
       "1430                                           Maybe   \n",
       "1431                                             Yes   \n",
       "\n",
       "     Have you been diagnosed with a mental health condition by a medical professional?  \\\n",
       "1                                                   Yes                                  \n",
       "4                                                   Yes                                  \n",
       "5                                                    No                                  \n",
       "6                                                    No                                  \n",
       "7                                                   Yes                                  \n",
       "...                                                 ...                                  \n",
       "1425                                                 No                                  \n",
       "1426                                                Yes                                  \n",
       "1427                                                Yes                                  \n",
       "1430                                                Yes                                  \n",
       "1431                                                Yes                                  \n",
       "\n",
       "     Have you ever sought treatment for a mental health issue from a mental health professional?  \\\n",
       "1                                                     1                                            \n",
       "4                                                     1                                            \n",
       "5                                                     1                                            \n",
       "6                                                     0                                            \n",
       "7                                                     1                                            \n",
       "...                                                 ...                                            \n",
       "1425                                                  0                                            \n",
       "1426                                                  1                                            \n",
       "1427                                                  1                                            \n",
       "1430                                                  1                                            \n",
       "1431                                                  0                                            \n",
       "\n",
       "     If you have a mental health issue, do you feel that it interferes with your work when being treated effectively?  \\\n",
       "1                                                Rarely                                                                 \n",
       "4                                             Sometimes                                                                 \n",
       "5                                  Not applicable to me                                                                 \n",
       "6                                  Not applicable to me                                                                 \n",
       "7                                             Sometimes                                                                 \n",
       "...                                                 ...                                                                 \n",
       "1425                                             Rarely                                                                 \n",
       "1426                                             Rarely                                                                 \n",
       "1427                                             Rarely                                                                 \n",
       "1430                                             Rarely                                                                 \n",
       "1431                                          Sometimes                                                                 \n",
       "\n",
       "     If you have a mental health issue, do you feel that it interferes with your work when NOT being treated effectively?  \\\n",
       "1                                             Sometimes                                                                     \n",
       "4                                             Sometimes                                                                     \n",
       "5                                                 Often                                                                     \n",
       "6                                  Not applicable to me                                                                     \n",
       "7                                                 Often                                                                     \n",
       "...                                                 ...                                                                     \n",
       "1425                                             Rarely                                                                     \n",
       "1426                                              Often                                                                     \n",
       "1427                                              Often                                                                     \n",
       "1430                                          Sometimes                                                                     \n",
       "1431                                              Often                                                                     \n",
       "\n",
       "     What is your gender? What country do you live in?  \\\n",
       "1                    Male     United States of America   \n",
       "4                  Female     United States of America   \n",
       "5                    Male               United Kingdom   \n",
       "6                    Male     United States of America   \n",
       "7                  Female     United States of America   \n",
       "...                   ...                          ...   \n",
       "1425                 Male                       Canada   \n",
       "1426               Female                       Canada   \n",
       "1427               Female     United States of America   \n",
       "1430                 Male     United States of America   \n",
       "1431               Female     United States of America   \n",
       "\n",
       "     What country do you work in?  \n",
       "1        United States of America  \n",
       "4        United States of America  \n",
       "5                  United Kingdom  \n",
       "6        United States of America  \n",
       "7        United States of America  \n",
       "...                           ...  \n",
       "1425                       Canada  \n",
       "1426                       Canada  \n",
       "1427     United States of America  \n",
       "1430     United States of America  \n",
       "1431     United States of America  \n",
       "\n",
       "[861 rows x 32 columns]"
      ]
     },
     "execution_count": 24,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "mental_df.replace(\",\", \"\")"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 25,
   "metadata": {},
   "outputs": [],
   "source": [
    "mental_df.to_csv('./Resources/cleaned_data.csv', index=False)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 30,
   "metadata": {},
   "outputs": [
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "create table cleaned_data (\n",
      "= DOES_YOUR_EMPLOYER_PROVIDE_MENTAL_HEALTH_BENEFITS_AS_PART_OF_HEALTHCARE_COVERAGE? % varchar,\n",
      "= DO_YOU_KNOW_THE_OPTIONS_FOR_MENTAL_HEALTH_CARE_AVAILABLE_UNDER_YOUR_EMPLOYER-PROVIDED_COVERAGE? % varchar,\n",
      "= DOES_YOUR_EMPLOYER_OFFER_RESOURCES_TO_LEARN_MORE_ABOUT_MENTAL_HEALTH_CONCERNS_AND_OPTIONS_FOR_SEEKING_HELP? % varchar,\n",
      "= IF_A_MENTAL_HEALTH_ISSUE_PROMPTED_YOU_TO_REQUEST_A_MEDICAL_LEAVE_FROM_WORK,_ASKING_FOR_THAT_LEAVE_WOULD_BE: % varchar,\n",
      "= DO_YOU_THINK_THAT_DISCUSSING_A_MENTAL_HEALTH_DISORDER_WITH_YOUR_EMPLOYER_WOULD_HAVE_NEGATIVE_CONSEQUENCES? % varchar,\n",
      "= DO_YOU_THINK_THAT_DISCUSSING_A_PHYSICAL_HEALTH_ISSUE_WITH_YOUR_EMPLOYER_WOULD_HAVE_NEGATIVE_CONSEQUENCES? % varchar,\n",
      "= DO_YOU_FEEL_THAT_YOUR_EMPLOYER_TAKES_MENTAL_HEALTH_AS_SERIOUSLY_AS_PHYSICAL_HEALTH? % varchar,\n",
      "= HAVE_YOU_HEARD_OF_OR_OBSERVED_NEGATIVE_CONSEQUENCES_FOR_CO-WORKERS_WHO_HAVE_BEEN_OPEN_ABOUT_MENTAL_HEALTH_ISSUES_IN_YOUR_WORKPLACE? % varchar,\n",
      "= DO_YOU_HAVE_PREVIOUS_EMPLOYERS? % varchar,\n",
      "= HAVE_YOUR_PREVIOUS_EMPLOYERS_PROVIDED_MENTAL_HEALTH_BENEFITS? % varchar,\n",
      "= WERE_YOU_AWARE_OF_THE_OPTIONS_FOR_MENTAL_HEALTH_CARE_PROVIDED_BY_YOUR_PREVIOUS_EMPLOYERS? % varchar,\n",
      "= DID_YOUR_PREVIOUS_EMPLOYERS_EVER_FORMALLY_DISCUSS_MENTAL_HEALTH_(AS_PART_OF_A_WELLNESS_CAMPAIGN_OR_OTHER_OFFICIAL_COMMUNICATION)? % varchar,\n",
      "= WAS_YOUR_ANONYMITY_PROTECTED_IF_YOU_CHOSE_TO_TAKE_ADVANTAGE_OF_MENTAL_HEALTH_OR_SUBSTANCE_ABUSE_TREATMENT_RESOURCES_WITH_PREVIOUS_EMPLOYERS? % varchar,\n",
      "= DO_YOU_THINK_THAT_DISCUSSING_A_MENTAL_HEALTH_DISORDER_WITH_PREVIOUS_EMPLOYERS_WOULD_HAVE_NEGATIVE_CONSEQUENCES? % varchar,\n",
      "= WOULD_YOU_HAVE_BEEN_WILLING_TO_DISCUSS_A_MENTAL_HEALTH_ISSUE_WITH_YOUR_DIRECT_SUPERVISOR(S)? % varchar,\n",
      "= DID_YOU_FEEL_THAT_YOUR_PREVIOUS_EMPLOYERS_TOOK_MENTAL_HEALTH_AS_SERIOUSLY_AS_PHYSICAL_HEALTH? % varchar,\n",
      "= DID_YOU_HEAR_OF_OR_OBSERVE_NEGATIVE_CONSEQUENCES_FOR_CO-WORKERS_WITH_MENTAL_HEALTH_ISSUES_IN_YOUR_PREVIOUS_WORKPLACES? % varchar,\n",
      "= WOULD_YOU_BRING_UP_A_MENTAL_HEALTH_ISSUE_WITH_A_POTENTIAL_EMPLOYER_IN_AN_INTERVIEW? % varchar,\n",
      "= DO_YOU_FEEL_THAT_BEING_IDENTIFIED_AS_A_PERSON_WITH_A_MENTAL_HEALTH_ISSUE_WOULD_HURT_YOUR_CAREER? % varchar,\n",
      "= DO_YOU_THINK_THAT_TEAM_MEMBERS/CO-WORKERS_WOULD_VIEW_YOU_MORE_NEGATIVELY_IF_THEY_KNEW_YOU_SUFFERED_FROM_A_MENTAL_HEALTH_ISSUE? % varchar,\n",
      "= HOW_WILLING_WOULD_YOU_BE_TO_SHARE_WITH_FRIENDS_AND_FAMILY_THAT_YOU_HAVE_A_MENTAL_ILLNESS? % varchar,\n",
      "= HAVE_YOU_OBSERVED_OR_EXPERIENCED_AN_UNSUPPORTIVE_OR_BADLY_HANDLED_RESPONSE_TO_A_MENTAL_HEALTH_ISSUE_IN_YOUR_CURRENT_OR_PREVIOUS_WORKPLACE? % varchar,\n",
      "= DO_YOU_HAVE_A_FAMILY_HISTORY_OF_MENTAL_ILLNESS? % varchar,\n",
      "= HAVE_YOU_HAD_A_MENTAL_HEALTH_DISORDER_IN_THE_PAST? % varchar,\n",
      "= DO_YOU_CURRENTLY_HAVE_A_MENTAL_HEALTH_DISORDER? % varchar,\n",
      "= HAVE_YOU_BEEN_DIAGNOSED_WITH_A_MENTAL_HEALTH_CONDITION_BY_A_MEDICAL_PROFESSIONAL? % varchar,\n",
      "= HAVE_YOU_EVER_SOUGHT_TREATMENT_FOR_A_MENTAL_HEALTH_ISSUE_FROM_A_MENTAL_HEALTH_PROFESSIONAL? % varchar,\n",
      "= IF_YOU_HAVE_A_MENTAL_HEALTH_ISSUE,_DO_YOU_FEEL_THAT_IT_INTERFERES_WITH_YOUR_WORK_WHEN_BEING_TREATED_EFFECTIVELY? % varchar,\n",
      "= IF_YOU_HAVE_A_MENTAL_HEALTH_ISSUE,_DO_YOU_FEEL_THAT_IT_INTERFERES_WITH_YOUR_WORK_WHEN_NOT_BEING_TREATED_EFFECTIVELY? % varchar,\n",
      "= WHAT_IS_YOUR_GENDER? % varchar,\n",
      "= WHAT_COUNTRY_DO_YOU_LIVE_IN? % varchar,\n",
      "= WHAT_COUNTRY_DO_YOU_WORK_IN? % varchar,\n"
     ]
    }
   ],
   "source": [
    "print(\"create table cleaned_data (\")\n",
    "for column in mental_df.columns: \n",
    "    print(\"=\", column.replace(\" \", \"_\").upper(), \"% varchar,\")"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "metadata": {},
   "outputs": [],
   "source": []
  }
 ],
 "metadata": {
  "kernelspec": {
   "display_name": "env",
   "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.8.16"
  }
 },
 "nbformat": 4,
 "nbformat_minor": 2
}