--- /dev/null	Thu Jan 01 00:00:00 1970 +0000
+++ b/asist_dynamic.ipynb	Sat Jun 26 07:27:53 2021 +0000
@@ -0,0 +1,310 @@
+{
+ "cells": [
+  {
+   "cell_type": "code",
+   "execution_count": 1309,
+   "id": "27cfc66f",
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "#ASIST program for phenotype based on Antibiotics profile\n",
+    "# create a profile based on selected antibiotics only\n",
+    "# rakesh4osdd@gmail.com, 14-June-2021"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": 1,
+   "id": "75a352b7",
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "import pandas as pd\n",
+    "import sys\n",
+    "import os\n",
+    "from collections import Counter"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": 176,
+   "id": "d66ec0d2",
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "input_file=sys.argv[1]\n",
+    "output_file=sys.argv[2]\n",
+    "#input_file='test-data/asist_input.csv'\n",
+    "#output_file='test-data/asist_output.csv'"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": 177,
+   "id": "bf24c946",
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "# strain_profile to phenotype condition\n",
+    "def s_phen(sus,res,intm,na,pb_sus):\n",
+    "    if (sus>0 and res==0 and na>=0):\n",
+    "        #print('Possible Susceptible')\n",
+    "        phen='Possible Susceptible'\n",
+    "    elif (sus>=0 and 3<=res<7 and na>=0 and pb_sus==0):\n",
+    "        #print('Possible MDR')\n",
+    "        phen='Possible MDR'\n",
+    "    elif (sus>=0 and 7<=res<9 and na>=0 and pb_sus==0):\n",
+    "        #print('Possible XDR')\n",
+    "        phen='Possible XDR'\n",
+    "    #special cases\n",
+    "    elif (sus>=1 and res>0 and na>=0 and pb_sus==1):\n",
+    "        #print('Possible XDR')\n",
+    "        phen='Possible XDR'\n",
+    "    #special cases\n",
+    "    elif (sus>0 and res==9 and na>=0):\n",
+    "        #print('Possible XDR')\n",
+    "        phen='Possible XDR'\n",
+    "    elif (sus==0 and res==9 and na>=0):\n",
+    "        #print('Possible TDR')\n",
+    "        phen='Possible TDR'\n",
+    "    else:\n",
+    "        #print('Strain could not be classified')\n",
+    "        phen='Strain could not be classified ('+ str(intm)+' | ' + str(na) +')'\n",
+    "    return(phen)\n",
+    "\n",
+    "#print(s_phen(1,9,0,0))"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": 178,
+   "id": "8bad7d9d",
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "# define Antibiotic groups as per antibiotic of CLSI breakpoints MIC\n",
+    "#Aminoglycoside\n",
+    "cat1=['Amikacin','Tobramycin','Gentamycin','Netilmicin']\n",
+    "#Beta-lactams- Carbapenems\n",
+    "cat2=['Imipenem','Meropenam','Doripenem']\n",
+    "#Fluoroquinolone\n",
+    "cat3=['Ciprofloxacin','Levofloxacin']\n",
+    "#Beta-lactam inhibitor\n",
+    "cat4=['Piperacillin/tazobactam','Ticarcillin/clavulanicacid']\n",
+    "#Cephalosporin\n",
+    "cat5=['Cefotaxime','Ceftriaxone','Ceftazidime','Cefepime']\n",
+    "#Sulfonamides\n",
+    "cat6=['Trimethoprim/sulfamethoxazole']\n",
+    "#Penicillins/beta-lactamase\n",
+    "cat7=['Ampicillin/sulbactam']\n",
+    "#Polymyxins\n",
+    "cat8=['Colistin','Polymyxinb']\n",
+    "#Tetracycline\n",
+    "cat9=['Tetracycline','Doxicycline','Minocycline']\n",
+    "\n",
+    "def s_profiler(pd_series):\n",
+    "    #print(type(pd_series),'\\n', pd_series)\n",
+    "    #create a dictionary of dataframe series\n",
+    "    cats={'s1':cat1,'s2':cat2,'s3':cat3,'s4':cat4,'s5':cat5,'s6':cat6,'s7':cat7,'s8':cat8,'s9':cat9}\n",
+    "    # find the antibiotics name in input series\n",
+    "    for cat in cats:\n",
+    "        #print(cats[cat])\n",
+    "        cats[cat]=pd_series.filter(cats[cat])\n",
+    "        #print(cats[cat])\n",
+    "    #define res,sus,intm,na,pb_sus\n",
+    "    res=0\n",
+    "    sus=0\n",
+    "    intm=0\n",
+    "    na=0\n",
+    "    pb_sus=0\n",
+    "    # special case of 'Polymyxin b' for its value\n",
+    "    if 'Polymyxinb' in pd_series:\n",
+    "        ctp=cats['s8']['Polymyxinb'].strip().lower()\n",
+    "        if ctp == 'susceptible':\n",
+    "            pb_sus=1\n",
+    "        #print((ctp,p_sus))\n",
+    "    # check all categories\n",
+    "    for cat in cats:\n",
+    "        #ctp=cats['s8'].iloc[i:i+1].stack().value_counts().to_dict()\n",
+    "        #print(ctp)\n",
+    "        # Pandas series\n",
+    "        ct=cats[cat].value_counts().to_dict()\n",
+    "        #print(ct)\n",
+    "        # remove whitespace and convert to lowercase words\n",
+    "        ct =  {k.strip().lower(): v for k, v in ct.items()}\n",
+    "        #print(ct)\n",
+    "        k=Counter(ct)\n",
+    "        #j=Counter(ct)+Counter(j)\n",
+    "        #print(j)\n",
+    "        # category wise marking\n",
+    "        if k['resistant']>=1:\n",
+    "            res=res+1\n",
+    "        if k['susceptible']>=1:\n",
+    "            sus=sus+1\n",
+    "        if k['intermediate']>=1:\n",
+    "            intm=intm+1\n",
+    "        if k['na']>=1:\n",
+    "            na=na+1\n",
+    "    #print(sus,res,intm,na,pb_sus)\n",
+    "    #print(s_phen(sus,res,intm,na,pb_sus))\n",
+    "    return(s_phen(sus,res,intm,na,pb_sus))"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": 179,
+   "id": "7629fc10",
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "#input_file='input2.csv_table.csv'\n",
+    "#output_file=input_file+'_output.txt'\n",
+    "strain_profile=pd.read_csv(input_file, sep=',',na_filter=False,skipinitialspace = True)"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": 180,
+   "id": "bed1abba",
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "old_strain_name=strain_profile.columns[0]\n",
+    "new_strain_name=old_strain_name.capitalize().strip().replace(' ', '')"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": 181,
+   "id": "a64b5022",
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "# make header capitalization, remove leading,lagging, and multiple whitespace for comparision\n",
+    "strain_profile.columns=strain_profile.columns.str.capitalize().str.strip().str.replace('\\s+', '', regex=True)\n",
+    "#print(strain_profile.columns)\n",
+    "#strain_profile.head()\n",
+    "#strain_profile.columns"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": 182,
+   "id": "caac57d7",
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "# add new column in dataframe on second position\n",
+    "strain_profile.insert(1, 'Strain phenotype','')\n",
+    "#strain_profile.head()"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": 183,
+   "id": "eb4b0c4d",
+   "metadata": {
+    "scrolled": true
+   },
+   "outputs": [],
+   "source": [
+    "strain_profile['Strain phenotype'] = strain_profile.apply(lambda x: (s_profiler(x)), axis=1)"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": 184,
+   "id": "86441c0f",
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "#strain_profile.head()"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": 185,
+   "id": "75698be5",
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "#rename headers for old name\n",
+    "strain_profile=strain_profile.rename(columns = {new_strain_name:old_strain_name, 'Ticarcillin/clavulanicacid':'Ticarcillin/ clavulanic acid','Piperacillin/tazobactam':'Piperacillin/ tazobactam','Trimethoprim/sulfamethoxazole': 'Trimethoprim/ sulfamethoxazole','Ampicillin/sulbactam':'Ampicillin/ sulbactam', 'Polymyxinb': 'Polymyxin B'} )"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": 186,
+   "id": "c14a13eb",
+   "metadata": {
+    "scrolled": true
+   },
+   "outputs": [],
+   "source": [
+    "#strain_profile.columns"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": 187,
+   "id": "ff484767",
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "#strain_profile"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": 188,
+   "id": "5ab72211",
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "strain_profile.to_csv(output_file,na_rep='NA',index=False)"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": 189,
+   "id": "020dbe85",
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "# Open a file with access mode 'a'\n",
+    "with open(output_file, \"a\") as file_object:\n",
+    "    # Append 'hello' at the end of file\n",
+    "    file_object.write(\"Note: \\n1. 'MDR': Multidrug-resistant, 'XDR': Extensively drug-resistant, 'TDR':totally drug resistant, NA': Data Not Available.\\n2. 'Strain could not be classified' numbers follow the format as ('Number of antibiotics categories count as Intermediate' | 'Number of antibiotics categories count as NA')\")"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "id": "9c17e66a",
+   "metadata": {},
+   "outputs": [],
+   "source": []
+  }
+ ],
+ "metadata": {
+  "kernelspec": {
+   "display_name": "Python 3",
+   "language": "python",
+   "name": "python3"
+  },
+  "language_info": {
+   "codemirror_mode": {
+    "name": "ipython",
+    "version": 3
+   },
+   "file_extension": ".py",
+   "mimetype": "text/x-python",
+   "name": "python",
+   "nbconvert_exporter": "python",
+   "pygments_lexer": "ipython3",
+   "version": "3.7.10"
+  }
+ },
+ "nbformat": 4,
+ "nbformat_minor": 5
+}