vaccine_clean_4.ipynb

{
 "cells": [
  {
   "cell_type": "code",
   "execution_count": 1,
   "id": "6c017d8e",
   "metadata": {},
   "outputs": [],
   "source": [
    "import pandas as pd\n",
    "import numpy as np\n",
    "import matplotlib.pyplot as plt\n",
    "import arabic_reshaper\n",
    "from arabic_reshaper import reshape\n",
    "from bidi.algorithm import get_display\n",
    "import jalali_pandas\n",
    "from persiantools.jdatetime import JalaliDate, JalaliDateTime\n",
    "from sklearn.preprocessing import LabelEncoder\n",
    "from collections import Counter\n",
    "import warnings\n",
    "warnings.filterwarnings(\"ignore\")"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 2,
   "id": "dcb000e2",
   "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>code</th>\n",
       "      <th>تاریخ پذیرش</th>\n",
       "      <th>سن</th>\n",
       "      <th>نام بخش بستری</th>\n",
       "      <th>اینتوباسیون</th>\n",
       "      <th>میزان Po2</th>\n",
       "      <th>وضعیت اکسیژن تراپی</th>\n",
       "      <th>سینا-نام واکسن مرحله 1</th>\n",
       "      <th>سینا-نام واکسن مرحله 2</th>\n",
       "      <th>سینا-نام واکسن مرحله 3</th>\n",
       "      <th>...</th>\n",
       "      <th>سینا-تاریخ تزریق دوز دوم</th>\n",
       "      <th>سینا-تاریخ تزریق دوز سوم</th>\n",
       "      <th>سینا-تاریخ تزریق دوز چهارم</th>\n",
       "      <th>دوز دريافت واكسن</th>\n",
       "      <th>sex</th>\n",
       "      <th>Early_symptoms</th>\n",
       "      <th>Digestion_symptoms</th>\n",
       "      <th>temp</th>\n",
       "      <th>Underlying_disease</th>\n",
       "      <th>labels</th>\n",
       "    </tr>\n",
       "  </thead>\n",
       "  <tbody>\n",
       "    <tr>\n",
       "      <th>0</th>\n",
       "      <td>982722885</td>\n",
       "      <td>1401/04/28</td>\n",
       "      <td>62</td>\n",
       "      <td>2</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>...</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>4</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>1</th>\n",
       "      <td>1060818078</td>\n",
       "      <td>1400/11/03</td>\n",
       "      <td>87</td>\n",
       "      <td>2</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>...</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>3</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>2</th>\n",
       "      <td>890540012</td>\n",
       "      <td>1400/05/04</td>\n",
       "      <td>21</td>\n",
       "      <td>2</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>...</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>2</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>4</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>3</th>\n",
       "      <td>749844469</td>\n",
       "      <td>1400/06/05</td>\n",
       "      <td>33</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>...</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>4</th>\n",
       "      <td>937231959</td>\n",
       "      <td>1400/06/16</td>\n",
       "      <td>66</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>...</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>2</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>3</td>\n",
       "      <td>2</td>\n",
       "    </tr>\n",
       "  </tbody>\n",
       "</table>\n",
       "<p>5 rows × 22 columns</p>\n",
       "</div>"
      ],
      "text/plain": [
       "         code تاریخ پذیرش  سن  نام بخش بستری  اینتوباسیون  میزان Po2  \\\n",
       "0   982722885  1401/04/28  62              2            0          1   \n",
       "1  1060818078  1400/11/03  87              2            0          0   \n",
       "2   890540012  1400/05/04  21              2            0          1   \n",
       "3   749844469  1400/06/05  33              0            0          0   \n",
       "4   937231959  1400/06/16  66              1            0          0   \n",
       "\n",
       "   وضعیت اکسیژن تراپی سینا-نام واکسن مرحله 1 سینا-نام واکسن مرحله 2  \\\n",
       "0                   0                      0                      0   \n",
       "1                   1                      0                      0   \n",
       "2                   0                      0                      0   \n",
       "3                   1                      0                      0   \n",
       "4                   0                      0                      0   \n",
       "\n",
       "  سینا-نام واکسن مرحله 3  ... سینا-تاریخ تزریق دوز دوم  \\\n",
       "0                      0  ...                        0   \n",
       "1                      0  ...                        0   \n",
       "2                      0  ...                        0   \n",
       "3                      0  ...                        0   \n",
       "4                      0  ...                        0   \n",
       "\n",
       "  سینا-تاریخ تزریق دوز سوم سینا-تاریخ تزریق دوز چهارم دوز دريافت واكسن sex  \\\n",
       "0                        0                          0                0   1   \n",
       "1                        0                          0                0   0   \n",
       "2                        0                          0                0   1   \n",
       "3                        0                          0                0   0   \n",
       "4                        0                          0                0   0   \n",
       "\n",
       "  Early_symptoms  Digestion_symptoms  temp  Underlying_disease  labels  \n",
       "0              1                   0     0                   0       4  \n",
       "1              3                   0     0                   0       1  \n",
       "2              0                   2     0                   0       4  \n",
       "3              1                   0     0                   0       1  \n",
       "4              2                   0     1                   3       2  \n",
       "\n",
       "[5 rows x 22 columns]"
      ]
     },
     "execution_count": 2,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "df=pd.read_csv(r'clustered_4.csv')\n",
    "df.head()"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 3,
   "id": "b4872987",
   "metadata": {},
   "outputs": [],
   "source": [
    "rows_to_fill = df['سینا-نام واکسن مرحله 3'] == 0\n",
    "df.loc[rows_to_fill, 'سینا-نام واکسن مرحله 3'] = df.loc[rows_to_fill, 'سینا-نام واکسن مرحله 4']"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 4,
   "id": "d8b72208",
   "metadata": {},
   "outputs": [],
   "source": [
    "#Dose Number 1\n",
    "df.loc[df['سینا-نام واکسن مرحله 1']=='Covid 19 - Bharat Biotech -کرونا بهارات','vaccine_01']='Bharat'\n",
    "df.loc[df['سینا-نام واکسن مرحله 1']=='Covid19- Astrazeneca -کروناآسترازنکا','vaccine_01']='Astrazeneca'\n",
    "df.loc[df['سینا-نام واکسن مرحله 1']=='Covid19-sinopharm -کرونا سینوفارم','vaccine_01']='Sinopharm'\n",
    "df.loc[df['سینا-نام واکسن مرحله 1']=='Covid19-Sputnic -کرونااسپوتنیک','vaccine_01']='Sputnic'\n",
    "df.loc[df['سینا-نام واکسن مرحله 1']=='COVIran Barakat -کووایران برکت','vaccine_01']='Barakat'\n",
    "df.loc[df['سینا-نام واکسن مرحله 1']=='PASTOCOVAC Plus -پاستوکووک پلاس','vaccine_01']='PASTOCOVAC'\n",
    "df.loc[df['سینا-نام واکسن مرحله 1']=='Razi Cov Pars-رازی','vaccine_01']='Razi'\n",
    "df.loc[df['سینا-نام واکسن مرحله 1']=='SpikoGen-اسپایکوژن','vaccine_01']='SpikoGen'\n",
    "df.loc[df['سینا-نام واکسن مرحله 1']=='سوبرانا 02- SOBERANA 02','vaccine_01']='Soberana'\n",
    "\n",
    "#Dose Number 2\n",
    "df.loc[df['سینا-نام واکسن مرحله 2']=='Covid 19 - Bharat Biotech -کرونا بهارات','vaccine_02']='Bharat'\n",
    "df.loc[df['سینا-نام واکسن مرحله 2']=='Covid19- Astrazeneca -کروناآسترازنکا','vaccine_02']='Astrazeneca'\n",
    "df.loc[df['سینا-نام واکسن مرحله 2']=='Covid19-sinopharm -کرونا سینوفارم','vaccine_02']='Sinopharm'\n",
    "df.loc[df['سینا-نام واکسن مرحله 2']=='Covid19-Sputnic -کرونااسپوتنیک','vaccine_02']='Sputnic'\n",
    "df.loc[df['سینا-نام واکسن مرحله 2']=='COVIran Barakat -کووایران برکت','vaccine_02']='Barakat'\n",
    "df.loc[df['سینا-نام واکسن مرحله 2']=='PASTOCOVAC Plus -پاستوکووک پلاس','vaccine_02']='PASTOCOVAC'\n",
    "df.loc[df['سینا-نام واکسن مرحله 2']=='Razi Cov Pars-رازی','vaccine_02']='Razi'\n",
    "df.loc[df['سینا-نام واکسن مرحله 2']=='SpikoGen-اسپایکوژن','vaccine_02']='SpikoGen'\n",
    "df.loc[df['سینا-نام واکسن مرحله 2']=='سوبرانا 02- SOBERANA 02','vaccine_02']='Soberana'\n",
    "\n",
    "#Dose Number 3\n",
    "df.loc[df['سینا-نام واکسن مرحله 3']=='Covid 19 - Bharat Biotech -کرونا بهارات','vaccine_03']='Bharat'\n",
    "df.loc[df['سینا-نام واکسن مرحله 3']=='Covid19- Astrazeneca -کروناآسترازنکا','vaccine_03']='Astrazeneca'\n",
    "df.loc[df['سینا-نام واکسن مرحله 3']=='Covid19-sinopharm -کرونا سینوفارم','vaccine_03']='Sinopharm'\n",
    "df.loc[df['سینا-نام واکسن مرحله 3']=='Covid19-Sputnic -کرونااسپوتنیک','vaccine_03']='Sputnic'\n",
    "df.loc[df['سینا-نام واکسن مرحله 3']=='COVIran Barakat -کووایران برکت','vaccine_03']='Barakat'\n",
    "df.loc[df['سینا-نام واکسن مرحله 3']=='PASTOCOVAC Plus -پاستوکووک پلاس','vaccine_03']='PASTOCOVAC'\n",
    "df.loc[df['سینا-نام واکسن مرحله 3']=='Razi Cov Pars-رازی','vaccine_03']='Razi'\n",
    "df.loc[df['سینا-نام واکسن مرحله 3']=='SpikoGen-اسپایکوژن','vaccine_03']='SpikoGen'\n",
    "df.loc[df['سینا-نام واکسن مرحله 3']=='سوبرانا 02- SOBERANA 02','vaccine_03']='Soberana'\n",
    "df.loc[df['سینا-نام واکسن مرحله 3']=='FAKHRAVAC-فخرا','vaccine_03']='Fakhra'\n",
    "df.loc[df['سینا-نام واکسن مرحله 3']=='نورا- Noora','vaccine_03']='Noora'\n",
    "\n",
    "#Dose Number 2\n",
    "# df.loc[df['سینا-نام واکسن مرحله 4']=='Covid19- Astrazeneca -کروناآسترازنکا','vaccine_04']='Astrazeneca'\n",
    "# df.loc[df['سینا-نام واکسن مرحله 4']=='Covid19-sinopharm -کرونا سینوفارم','vaccine_04']='Sinopharm'\n",
    "# df.loc[df['سینا-نام واکسن مرحله 4']=='COVIran Barakat -کووایران برکت','vaccine_04']='Barakat'\n",
    "# df.loc[df['سینا-نام واکسن مرحله 4']=='Razi Cov Pars-رازی','vaccine_04']='Razi'\n",
    "# df.loc[df['سینا-نام واکسن مرحله 4']=='SpikoGen-اسپایکوژن','vaccine_04']='SpikoGen'\n",
    "# df.loc[df['سینا-نام واکسن مرحله 4']=='سوبرانا 02- SOBERANA 02','vaccine_04']='Soberana'\n",
    "# df.loc[df['سینا-نام واکسن مرحله 4']=='FAKHRAVAC-فخرا','vaccine_04']='Fakhra'\n",
    "# df.loc[df['سینا-نام واکسن مرحله 4']=='نورا- Noora','vaccine_04']='Noora'"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 5,
   "id": "ad004221",
   "metadata": {},
   "outputs": [],
   "source": [
    "df.loc[(df['vaccine_01']=='Sputnic')|(df['vaccine_01']=='Astrazeneca'),'vac_1']=1\n",
    "df.loc[df['vaccine_01'].isna()==True,'vac_1']=0\n",
    "df['vac_1'].fillna(2,inplace=True)\n",
    "\n",
    "df.loc[(df['vaccine_02']=='Sputnic')|(df['vaccine_02']=='Astrazeneca'),'vac_2']=1\n",
    "df.loc[df['vaccine_02'].isna()==True,'vac_2']=0\n",
    "df['vac_2'].fillna(2,inplace=True)\n",
    "\n",
    "df.loc[(df['vaccine_03']=='Sputnic')|(df['vaccine_03']=='Astrazeneca'),'vac_3']=1\n",
    "df.loc[df['vaccine_03'].isna()==True,'vac_3']=0\n",
    "df['vac_3'].fillna(2,inplace=True)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 6,
   "id": "2e7a8e16",
   "metadata": {},
   "outputs": [],
   "source": [
    "df.loc[(df['vac_1']==0) & (df['vac_2']==0) & (df['vac_3']==0), 'vac_sequence'] = 0\n",
    "df.loc[(df['vac_1']==1) & (df['vac_2']==0) & (df['vac_3']==0), 'vac_sequence'] = 1\n",
    "df.loc[(df['vac_1']==2) & (df['vac_2']==0) & (df['vac_3']==0), 'vac_sequence'] = 2\n",
    "df.loc[(df['vac_1']==1) & (df['vac_2']==1) & (df['vac_3']==0), 'vac_sequence'] = 3\n",
    "df.loc[(df['vac_1']==1) & (df['vac_2']==2) & (df['vac_3']==0), 'vac_sequence'] = 4\n",
    "df.loc[(df['vac_1']==2) & (df['vac_2']==2) & (df['vac_3']==0), 'vac_sequence'] = 5\n",
    "df.loc[(df['vac_1']==2) & (df['vac_2']==1) & (df['vac_3']==0), 'vac_sequence'] = 6\n",
    "df.loc[(df['vac_1']==1) & (df['vac_2']==1) & (df['vac_3']==1), 'vac_sequence'] = 7\n",
    "df.loc[(df['vac_1']==1) & (df['vac_2']==1) & (df['vac_3']==2), 'vac_sequence'] = 8\n",
    "df.loc[(df['vac_1']==1) & (df['vac_2']==2) & (df['vac_3']==2), 'vac_sequence'] = 9\n",
    "df.loc[(df['vac_1']==1) & (df['vac_2']==2) & (df['vac_3']==1), 'vac_sequence'] = 10\n",
    "df.loc[(df['vac_1']==2) & (df['vac_2']==2) & (df['vac_3']==2), 'vac_sequence'] = 11\n",
    "df.loc[(df['vac_1']==2) & (df['vac_2']==2) & (df['vac_3']==1), 'vac_sequence'] = 12\n",
    "df.loc[(df['vac_1']==2) & (df['vac_2']==1) & (df['vac_3']==1), 'vac_sequence'] = 13\n",
    "df.loc[(df['vac_1']==2) & (df['vac_2']==1) & (df['vac_3']==2), 'vac_sequence'] = 14\n",
    "df['vac_sequence'].fillna(0,inplace=True)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 7,
   "id": "b0d5cf5e",
   "metadata": {},
   "outputs": [],
   "source": [
    "df.to_csv('vaccine_clean_5.csv',index=False,encoding='utf-8-sig')"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "id": "b2e6ef65",
   "metadata": {},
   "outputs": [],
   "source": []
  }
 ],
 "metadata": {
  "kernelspec": {
   "display_name": "Python 3 (ipykernel)",
   "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.10.7"
  }
 },
 "nbformat": 4,
 "nbformat_minor": 5
}