product_recommender.ipynb

{
 "cells": [
  {
   "cell_type": "code",
   "execution_count": 131,
   "metadata": {},
   "outputs": [],
   "source": [
    "import pandas as pd\n",
    "import numpy as np\n",
    "import seaborn as sns\n",
    "from itertools import combinations, groupby, chain\n",
    "from collections import Counter"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 2,
   "metadata": {},
   "outputs": [],
   "source": [
    "cluster0_item_rules = pd.read_csv('group0_item_rules.csv', index_col = 0)\n",
    "cluster1_item_rules = pd.read_csv('group1_item_rules.csv', index_col = 0)\n",
    "cluster2_item_rules = pd.read_csv('group2_item_rules.csv', index_col = 0)\n",
    "cluster3_item_rules = pd.read_csv('group3_item_rules.csv', index_col = 0)\n",
    "cluster4_item_rules = pd.read_csv('group4_item_rules.csv', index_col = 0)\n",
    "cluster5_item_rules = pd.read_csv('group5_item_rules.csv', index_col = 0)\n",
    "cluster0_aisle_rules = pd.read_csv('group0_aisle_rules.csv', index_col = 0)\n",
    "cluster1_aisle_rules = pd.read_csv('group1_aisle_rules.csv', index_col = 0)\n",
    "cluster2_aisle_rules = pd.read_csv('group2_aisle_rules.csv', index_col = 0)\n",
    "cluster3_aisle_rules = pd.read_csv('group3_aisle_rules.csv', index_col = 0)\n",
    "cluster4_aisle_rules = pd.read_csv('group4_aisle_rules.csv', index_col = 0)\n",
    "cluster5_aisle_rules = pd.read_csv('group5_aisle_rules.csv', index_col = 0)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 3,
   "metadata": {},
   "outputs": [],
   "source": [
    "products = pd.read_csv('products.csv', index_col = 0)\n",
    "aisles = pd.read_csv('aisles.csv', index_col = 0)\n",
    "clusters = pd.read_csv('cluster_df', index_col = 0)\n",
    "#products = products.merge(aisles, on = 'aisle_id', how = 'left')"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 229,
   "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>0</th>\n",
       "      <th>1</th>\n",
       "      <th>2</th>\n",
       "      <th>3</th>\n",
       "      <th>4</th>\n",
       "      <th>5</th>\n",
       "      <th>cluster</th>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>user_id</th>\n",
       "      <th></th>\n",
       "      <th></th>\n",
       "      <th></th>\n",
       "      <th></th>\n",
       "      <th></th>\n",
       "      <th></th>\n",
       "      <th></th>\n",
       "    </tr>\n",
       "  </thead>\n",
       "  <tbody>\n",
       "    <tr>\n",
       "      <th>1</th>\n",
       "      <td>-0.103161</td>\n",
       "      <td>0.048617</td>\n",
       "      <td>-0.031836</td>\n",
       "      <td>-0.005633</td>\n",
       "      <td>-0.053015</td>\n",
       "      <td>-0.055213</td>\n",
       "      <td>2</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>2</th>\n",
       "      <td>-0.077766</td>\n",
       "      <td>0.077983</td>\n",
       "      <td>-0.076149</td>\n",
       "      <td>-0.101670</td>\n",
       "      <td>0.081781</td>\n",
       "      <td>-0.040406</td>\n",
       "      <td>2</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>3</th>\n",
       "      <td>0.057496</td>\n",
       "      <td>0.085719</td>\n",
       "      <td>0.001779</td>\n",
       "      <td>-0.013191</td>\n",
       "      <td>-0.028496</td>\n",
       "      <td>0.113063</td>\n",
       "      <td>5</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>4</th>\n",
       "      <td>-0.055343</td>\n",
       "      <td>0.134731</td>\n",
       "      <td>-0.002392</td>\n",
       "      <td>-0.037708</td>\n",
       "      <td>-0.087808</td>\n",
       "      <td>-0.027363</td>\n",
       "      <td>4</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>5</th>\n",
       "      <td>0.133936</td>\n",
       "      <td>-0.019591</td>\n",
       "      <td>-0.020960</td>\n",
       "      <td>-0.003730</td>\n",
       "      <td>0.057734</td>\n",
       "      <td>0.138895</td>\n",
       "      <td>5</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>6</th>\n",
       "      <td>0.162593</td>\n",
       "      <td>-0.169693</td>\n",
       "      <td>0.006627</td>\n",
       "      <td>0.061952</td>\n",
       "      <td>0.020037</td>\n",
       "      <td>0.090207</td>\n",
       "      <td>0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>7</th>\n",
       "      <td>-0.000091</td>\n",
       "      <td>-0.010104</td>\n",
       "      <td>-0.023335</td>\n",
       "      <td>0.004957</td>\n",
       "      <td>-0.020541</td>\n",
       "      <td>0.027418</td>\n",
       "      <td>5</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>8</th>\n",
       "      <td>0.272457</td>\n",
       "      <td>-0.206370</td>\n",
       "      <td>0.042502</td>\n",
       "      <td>0.041497</td>\n",
       "      <td>-0.006041</td>\n",
       "      <td>-0.047413</td>\n",
       "      <td>0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>9</th>\n",
       "      <td>-0.130975</td>\n",
       "      <td>-0.018146</td>\n",
       "      <td>-0.077137</td>\n",
       "      <td>-0.084070</td>\n",
       "      <td>0.123204</td>\n",
       "      <td>-0.043478</td>\n",
       "      <td>2</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>10</th>\n",
       "      <td>0.061656</td>\n",
       "      <td>-0.034921</td>\n",
       "      <td>-0.017177</td>\n",
       "      <td>0.020925</td>\n",
       "      <td>-0.008407</td>\n",
       "      <td>0.050041</td>\n",
       "      <td>5</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>11</th>\n",
       "      <td>-0.095517</td>\n",
       "      <td>0.010746</td>\n",
       "      <td>0.037512</td>\n",
       "      <td>-0.040126</td>\n",
       "      <td>0.043800</td>\n",
       "      <td>-0.006229</td>\n",
       "      <td>2</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>12</th>\n",
       "      <td>0.083170</td>\n",
       "      <td>-0.055728</td>\n",
       "      <td>0.024143</td>\n",
       "      <td>0.009697</td>\n",
       "      <td>-0.035023</td>\n",
       "      <td>-0.035245</td>\n",
       "      <td>0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>13</th>\n",
       "      <td>0.080558</td>\n",
       "      <td>-0.138144</td>\n",
       "      <td>-0.007634</td>\n",
       "      <td>0.013985</td>\n",
       "      <td>0.028689</td>\n",
       "      <td>-0.023335</td>\n",
       "      <td>0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>14</th>\n",
       "      <td>-0.044074</td>\n",
       "      <td>-0.069167</td>\n",
       "      <td>-0.027507</td>\n",
       "      <td>0.029800</td>\n",
       "      <td>-0.027399</td>\n",
       "      <td>-0.002945</td>\n",
       "      <td>2</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>15</th>\n",
       "      <td>-0.153422</td>\n",
       "      <td>-0.035306</td>\n",
       "      <td>-0.013711</td>\n",
       "      <td>0.036386</td>\n",
       "      <td>-0.064137</td>\n",
       "      <td>-0.050130</td>\n",
       "      <td>2</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>16</th>\n",
       "      <td>0.130385</td>\n",
       "      <td>-0.031641</td>\n",
       "      <td>0.001189</td>\n",
       "      <td>0.024568</td>\n",
       "      <td>-0.004197</td>\n",
       "      <td>0.109780</td>\n",
       "      <td>0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>17</th>\n",
       "      <td>-0.118188</td>\n",
       "      <td>0.000701</td>\n",
       "      <td>-0.022280</td>\n",
       "      <td>-0.007853</td>\n",
       "      <td>-0.042402</td>\n",
       "      <td>-0.018177</td>\n",
       "      <td>2</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>18</th>\n",
       "      <td>0.035986</td>\n",
       "      <td>0.059658</td>\n",
       "      <td>0.023618</td>\n",
       "      <td>0.013015</td>\n",
       "      <td>0.026583</td>\n",
       "      <td>0.274647</td>\n",
       "      <td>5</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>19</th>\n",
       "      <td>-0.122734</td>\n",
       "      <td>-0.049328</td>\n",
       "      <td>-0.051573</td>\n",
       "      <td>0.012025</td>\n",
       "      <td>-0.023170</td>\n",
       "      <td>-0.010374</td>\n",
       "      <td>2</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>20</th>\n",
       "      <td>0.018976</td>\n",
       "      <td>0.315233</td>\n",
       "      <td>0.000487</td>\n",
       "      <td>0.061689</td>\n",
       "      <td>-0.034785</td>\n",
       "      <td>-0.086830</td>\n",
       "      <td>4</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>21</th>\n",
       "      <td>-0.063147</td>\n",
       "      <td>0.085705</td>\n",
       "      <td>0.002685</td>\n",
       "      <td>-0.048406</td>\n",
       "      <td>0.004326</td>\n",
       "      <td>-0.012627</td>\n",
       "      <td>2</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>22</th>\n",
       "      <td>0.165547</td>\n",
       "      <td>-0.049237</td>\n",
       "      <td>0.012727</td>\n",
       "      <td>-0.004899</td>\n",
       "      <td>0.001815</td>\n",
       "      <td>-0.018923</td>\n",
       "      <td>0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>23</th>\n",
       "      <td>-0.084061</td>\n",
       "      <td>-0.040314</td>\n",
       "      <td>-0.066604</td>\n",
       "      <td>-0.051135</td>\n",
       "      <td>0.147089</td>\n",
       "      <td>0.070539</td>\n",
       "      <td>2</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>24</th>\n",
       "      <td>-0.094222</td>\n",
       "      <td>0.032029</td>\n",
       "      <td>-0.028414</td>\n",
       "      <td>0.080351</td>\n",
       "      <td>-0.002072</td>\n",
       "      <td>0.032683</td>\n",
       "      <td>2</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>25</th>\n",
       "      <td>-0.060687</td>\n",
       "      <td>-0.078775</td>\n",
       "      <td>-0.038270</td>\n",
       "      <td>0.032244</td>\n",
       "      <td>-0.026695</td>\n",
       "      <td>-0.008720</td>\n",
       "      <td>2</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>26</th>\n",
       "      <td>-0.169667</td>\n",
       "      <td>-0.039696</td>\n",
       "      <td>0.182831</td>\n",
       "      <td>-0.027160</td>\n",
       "      <td>-0.003216</td>\n",
       "      <td>0.027595</td>\n",
       "      <td>1</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>27</th>\n",
       "      <td>-0.118738</td>\n",
       "      <td>0.021549</td>\n",
       "      <td>0.011101</td>\n",
       "      <td>-0.121372</td>\n",
       "      <td>0.145781</td>\n",
       "      <td>-0.031379</td>\n",
       "      <td>2</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>28</th>\n",
       "      <td>0.173673</td>\n",
       "      <td>-0.091887</td>\n",
       "      <td>0.011302</td>\n",
       "      <td>0.005506</td>\n",
       "      <td>0.023483</td>\n",
       "      <td>0.008463</td>\n",
       "      <td>0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>29</th>\n",
       "      <td>-0.141629</td>\n",
       "      <td>-0.050801</td>\n",
       "      <td>-0.018189</td>\n",
       "      <td>0.009059</td>\n",
       "      <td>-0.037895</td>\n",
       "      <td>-0.006142</td>\n",
       "      <td>2</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>30</th>\n",
       "      <td>-0.228467</td>\n",
       "      <td>-0.021273</td>\n",
       "      <td>0.443406</td>\n",
       "      <td>-0.032353</td>\n",
       "      <td>0.049723</td>\n",
       "      <td>0.005172</td>\n",
       "      <td>1</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>...</th>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>206180</th>\n",
       "      <td>0.118065</td>\n",
       "      <td>0.050599</td>\n",
       "      <td>0.053969</td>\n",
       "      <td>0.005989</td>\n",
       "      <td>-0.020349</td>\n",
       "      <td>0.024375</td>\n",
       "      <td>5</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>206181</th>\n",
       "      <td>0.033571</td>\n",
       "      <td>-0.085424</td>\n",
       "      <td>-0.025355</td>\n",
       "      <td>-0.003371</td>\n",
       "      <td>0.024423</td>\n",
       "      <td>-0.008229</td>\n",
       "      <td>5</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>206182</th>\n",
       "      <td>-0.103252</td>\n",
       "      <td>-0.062499</td>\n",
       "      <td>-0.039217</td>\n",
       "      <td>0.002367</td>\n",
       "      <td>-0.009444</td>\n",
       "      <td>-0.040043</td>\n",
       "      <td>2</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>206183</th>\n",
       "      <td>0.124315</td>\n",
       "      <td>0.117831</td>\n",
       "      <td>0.000942</td>\n",
       "      <td>-0.025077</td>\n",
       "      <td>-0.061440</td>\n",
       "      <td>-0.001258</td>\n",
       "      <td>4</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>206184</th>\n",
       "      <td>0.108213</td>\n",
       "      <td>-0.105641</td>\n",
       "      <td>-0.007615</td>\n",
       "      <td>0.030982</td>\n",
       "      <td>-0.012060</td>\n",
       "      <td>0.037068</td>\n",
       "      <td>0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>206185</th>\n",
       "      <td>-0.110057</td>\n",
       "      <td>-0.013545</td>\n",
       "      <td>-0.048883</td>\n",
       "      <td>0.004643</td>\n",
       "      <td>-0.035784</td>\n",
       "      <td>0.011322</td>\n",
       "      <td>2</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>206186</th>\n",
       "      <td>-0.069416</td>\n",
       "      <td>-0.089934</td>\n",
       "      <td>0.061557</td>\n",
       "      <td>0.002463</td>\n",
       "      <td>-0.036717</td>\n",
       "      <td>-0.032196</td>\n",
       "      <td>2</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>206187</th>\n",
       "      <td>0.225091</td>\n",
       "      <td>-0.136182</td>\n",
       "      <td>0.046195</td>\n",
       "      <td>0.011636</td>\n",
       "      <td>-0.015577</td>\n",
       "      <td>-0.082668</td>\n",
       "      <td>0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>206188</th>\n",
       "      <td>0.014667</td>\n",
       "      <td>0.122048</td>\n",
       "      <td>-0.027902</td>\n",
       "      <td>-0.023798</td>\n",
       "      <td>-0.038616</td>\n",
       "      <td>0.013988</td>\n",
       "      <td>4</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>206189</th>\n",
       "      <td>-0.109030</td>\n",
       "      <td>0.018626</td>\n",
       "      <td>0.040297</td>\n",
       "      <td>-0.008008</td>\n",
       "      <td>-0.038375</td>\n",
       "      <td>-0.015350</td>\n",
       "      <td>2</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>206190</th>\n",
       "      <td>0.094096</td>\n",
       "      <td>-0.010271</td>\n",
       "      <td>-0.000792</td>\n",
       "      <td>0.005745</td>\n",
       "      <td>-0.032442</td>\n",
       "      <td>0.021697</td>\n",
       "      <td>5</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>206191</th>\n",
       "      <td>0.090761</td>\n",
       "      <td>-0.048457</td>\n",
       "      <td>0.022075</td>\n",
       "      <td>0.006139</td>\n",
       "      <td>-0.019522</td>\n",
       "      <td>-0.009650</td>\n",
       "      <td>0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>206192</th>\n",
       "      <td>-0.090736</td>\n",
       "      <td>-0.013101</td>\n",
       "      <td>-0.042613</td>\n",
       "      <td>0.013043</td>\n",
       "      <td>-0.038746</td>\n",
       "      <td>0.020027</td>\n",
       "      <td>2</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>206193</th>\n",
       "      <td>0.120594</td>\n",
       "      <td>-0.018295</td>\n",
       "      <td>0.003315</td>\n",
       "      <td>-0.006145</td>\n",
       "      <td>-0.019431</td>\n",
       "      <td>0.000788</td>\n",
       "      <td>0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>206194</th>\n",
       "      <td>0.092770</td>\n",
       "      <td>-0.006795</td>\n",
       "      <td>0.016130</td>\n",
       "      <td>0.002082</td>\n",
       "      <td>-0.027934</td>\n",
       "      <td>0.027662</td>\n",
       "      <td>5</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>206195</th>\n",
       "      <td>-0.011678</td>\n",
       "      <td>0.044205</td>\n",
       "      <td>-0.029063</td>\n",
       "      <td>-0.009212</td>\n",
       "      <td>-0.029639</td>\n",
       "      <td>0.005311</td>\n",
       "      <td>5</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>206196</th>\n",
       "      <td>0.118525</td>\n",
       "      <td>-0.060037</td>\n",
       "      <td>0.023729</td>\n",
       "      <td>0.011800</td>\n",
       "      <td>-0.022086</td>\n",
       "      <td>0.009534</td>\n",
       "      <td>0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>206197</th>\n",
       "      <td>-0.003363</td>\n",
       "      <td>-0.090393</td>\n",
       "      <td>-0.021594</td>\n",
       "      <td>0.037079</td>\n",
       "      <td>-0.004549</td>\n",
       "      <td>0.042953</td>\n",
       "      <td>5</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>206198</th>\n",
       "      <td>0.156617</td>\n",
       "      <td>-0.168687</td>\n",
       "      <td>0.045201</td>\n",
       "      <td>0.033473</td>\n",
       "      <td>-0.012215</td>\n",
       "      <td>-0.046276</td>\n",
       "      <td>0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>206199</th>\n",
       "      <td>-0.090888</td>\n",
       "      <td>-0.037290</td>\n",
       "      <td>-0.043918</td>\n",
       "      <td>-0.033467</td>\n",
       "      <td>0.037187</td>\n",
       "      <td>-0.039444</td>\n",
       "      <td>2</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>206200</th>\n",
       "      <td>0.089677</td>\n",
       "      <td>-0.012174</td>\n",
       "      <td>0.036927</td>\n",
       "      <td>-0.002099</td>\n",
       "      <td>-0.023412</td>\n",
       "      <td>0.020383</td>\n",
       "      <td>5</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>206201</th>\n",
       "      <td>-0.087776</td>\n",
       "      <td>-0.050358</td>\n",
       "      <td>-0.012275</td>\n",
       "      <td>0.025023</td>\n",
       "      <td>-0.033518</td>\n",
       "      <td>-0.002956</td>\n",
       "      <td>2</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>206202</th>\n",
       "      <td>0.001792</td>\n",
       "      <td>0.041479</td>\n",
       "      <td>0.040981</td>\n",
       "      <td>-0.030241</td>\n",
       "      <td>-0.020695</td>\n",
       "      <td>-0.008188</td>\n",
       "      <td>5</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>206203</th>\n",
       "      <td>0.036284</td>\n",
       "      <td>-0.057194</td>\n",
       "      <td>0.014533</td>\n",
       "      <td>0.012839</td>\n",
       "      <td>-0.025754</td>\n",
       "      <td>-0.029678</td>\n",
       "      <td>5</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>206204</th>\n",
       "      <td>-0.001979</td>\n",
       "      <td>-0.058562</td>\n",
       "      <td>-0.029888</td>\n",
       "      <td>0.010035</td>\n",
       "      <td>-0.004878</td>\n",
       "      <td>0.009821</td>\n",
       "      <td>5</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>206205</th>\n",
       "      <td>-0.021674</td>\n",
       "      <td>0.081783</td>\n",
       "      <td>-0.073480</td>\n",
       "      <td>-0.088436</td>\n",
       "      <td>0.155964</td>\n",
       "      <td>-0.022700</td>\n",
       "      <td>5</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>206206</th>\n",
       "      <td>-0.040316</td>\n",
       "      <td>-0.036658</td>\n",
       "      <td>-0.024052</td>\n",
       "      <td>0.010058</td>\n",
       "      <td>-0.051022</td>\n",
       "      <td>0.000835</td>\n",
       "      <td>2</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>206207</th>\n",
       "      <td>-0.022110</td>\n",
       "      <td>-0.032613</td>\n",
       "      <td>-0.009468</td>\n",
       "      <td>-0.015799</td>\n",
       "      <td>0.021500</td>\n",
       "      <td>0.020459</td>\n",
       "      <td>5</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>206208</th>\n",
       "      <td>-0.002217</td>\n",
       "      <td>-0.010633</td>\n",
       "      <td>-0.037621</td>\n",
       "      <td>-0.007752</td>\n",
       "      <td>0.028734</td>\n",
       "      <td>0.044838</td>\n",
       "      <td>5</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>206209</th>\n",
       "      <td>-0.072488</td>\n",
       "      <td>0.017514</td>\n",
       "      <td>-0.043592</td>\n",
       "      <td>-0.003232</td>\n",
       "      <td>-0.010478</td>\n",
       "      <td>0.027641</td>\n",
       "      <td>2</td>\n",
       "    </tr>\n",
       "  </tbody>\n",
       "</table>\n",
       "<p>206209 rows × 7 columns</p>\n",
       "</div>"
      ],
      "text/plain": [
       "                0         1         2         3         4         5  cluster\n",
       "user_id                                                                     \n",
       "1       -0.103161  0.048617 -0.031836 -0.005633 -0.053015 -0.055213        2\n",
       "2       -0.077766  0.077983 -0.076149 -0.101670  0.081781 -0.040406        2\n",
       "3        0.057496  0.085719  0.001779 -0.013191 -0.028496  0.113063        5\n",
       "4       -0.055343  0.134731 -0.002392 -0.037708 -0.087808 -0.027363        4\n",
       "5        0.133936 -0.019591 -0.020960 -0.003730  0.057734  0.138895        5\n",
       "6        0.162593 -0.169693  0.006627  0.061952  0.020037  0.090207        0\n",
       "7       -0.000091 -0.010104 -0.023335  0.004957 -0.020541  0.027418        5\n",
       "8        0.272457 -0.206370  0.042502  0.041497 -0.006041 -0.047413        0\n",
       "9       -0.130975 -0.018146 -0.077137 -0.084070  0.123204 -0.043478        2\n",
       "10       0.061656 -0.034921 -0.017177  0.020925 -0.008407  0.050041        5\n",
       "11      -0.095517  0.010746  0.037512 -0.040126  0.043800 -0.006229        2\n",
       "12       0.083170 -0.055728  0.024143  0.009697 -0.035023 -0.035245        0\n",
       "13       0.080558 -0.138144 -0.007634  0.013985  0.028689 -0.023335        0\n",
       "14      -0.044074 -0.069167 -0.027507  0.029800 -0.027399 -0.002945        2\n",
       "15      -0.153422 -0.035306 -0.013711  0.036386 -0.064137 -0.050130        2\n",
       "16       0.130385 -0.031641  0.001189  0.024568 -0.004197  0.109780        0\n",
       "17      -0.118188  0.000701 -0.022280 -0.007853 -0.042402 -0.018177        2\n",
       "18       0.035986  0.059658  0.023618  0.013015  0.026583  0.274647        5\n",
       "19      -0.122734 -0.049328 -0.051573  0.012025 -0.023170 -0.010374        2\n",
       "20       0.018976  0.315233  0.000487  0.061689 -0.034785 -0.086830        4\n",
       "21      -0.063147  0.085705  0.002685 -0.048406  0.004326 -0.012627        2\n",
       "22       0.165547 -0.049237  0.012727 -0.004899  0.001815 -0.018923        0\n",
       "23      -0.084061 -0.040314 -0.066604 -0.051135  0.147089  0.070539        2\n",
       "24      -0.094222  0.032029 -0.028414  0.080351 -0.002072  0.032683        2\n",
       "25      -0.060687 -0.078775 -0.038270  0.032244 -0.026695 -0.008720        2\n",
       "26      -0.169667 -0.039696  0.182831 -0.027160 -0.003216  0.027595        1\n",
       "27      -0.118738  0.021549  0.011101 -0.121372  0.145781 -0.031379        2\n",
       "28       0.173673 -0.091887  0.011302  0.005506  0.023483  0.008463        0\n",
       "29      -0.141629 -0.050801 -0.018189  0.009059 -0.037895 -0.006142        2\n",
       "30      -0.228467 -0.021273  0.443406 -0.032353  0.049723  0.005172        1\n",
       "...           ...       ...       ...       ...       ...       ...      ...\n",
       "206180   0.118065  0.050599  0.053969  0.005989 -0.020349  0.024375        5\n",
       "206181   0.033571 -0.085424 -0.025355 -0.003371  0.024423 -0.008229        5\n",
       "206182  -0.103252 -0.062499 -0.039217  0.002367 -0.009444 -0.040043        2\n",
       "206183   0.124315  0.117831  0.000942 -0.025077 -0.061440 -0.001258        4\n",
       "206184   0.108213 -0.105641 -0.007615  0.030982 -0.012060  0.037068        0\n",
       "206185  -0.110057 -0.013545 -0.048883  0.004643 -0.035784  0.011322        2\n",
       "206186  -0.069416 -0.089934  0.061557  0.002463 -0.036717 -0.032196        2\n",
       "206187   0.225091 -0.136182  0.046195  0.011636 -0.015577 -0.082668        0\n",
       "206188   0.014667  0.122048 -0.027902 -0.023798 -0.038616  0.013988        4\n",
       "206189  -0.109030  0.018626  0.040297 -0.008008 -0.038375 -0.015350        2\n",
       "206190   0.094096 -0.010271 -0.000792  0.005745 -0.032442  0.021697        5\n",
       "206191   0.090761 -0.048457  0.022075  0.006139 -0.019522 -0.009650        0\n",
       "206192  -0.090736 -0.013101 -0.042613  0.013043 -0.038746  0.020027        2\n",
       "206193   0.120594 -0.018295  0.003315 -0.006145 -0.019431  0.000788        0\n",
       "206194   0.092770 -0.006795  0.016130  0.002082 -0.027934  0.027662        5\n",
       "206195  -0.011678  0.044205 -0.029063 -0.009212 -0.029639  0.005311        5\n",
       "206196   0.118525 -0.060037  0.023729  0.011800 -0.022086  0.009534        0\n",
       "206197  -0.003363 -0.090393 -0.021594  0.037079 -0.004549  0.042953        5\n",
       "206198   0.156617 -0.168687  0.045201  0.033473 -0.012215 -0.046276        0\n",
       "206199  -0.090888 -0.037290 -0.043918 -0.033467  0.037187 -0.039444        2\n",
       "206200   0.089677 -0.012174  0.036927 -0.002099 -0.023412  0.020383        5\n",
       "206201  -0.087776 -0.050358 -0.012275  0.025023 -0.033518 -0.002956        2\n",
       "206202   0.001792  0.041479  0.040981 -0.030241 -0.020695 -0.008188        5\n",
       "206203   0.036284 -0.057194  0.014533  0.012839 -0.025754 -0.029678        5\n",
       "206204  -0.001979 -0.058562 -0.029888  0.010035 -0.004878  0.009821        5\n",
       "206205  -0.021674  0.081783 -0.073480 -0.088436  0.155964 -0.022700        5\n",
       "206206  -0.040316 -0.036658 -0.024052  0.010058 -0.051022  0.000835        2\n",
       "206207  -0.022110 -0.032613 -0.009468 -0.015799  0.021500  0.020459        5\n",
       "206208  -0.002217 -0.010633 -0.037621 -0.007752  0.028734  0.044838        5\n",
       "206209  -0.072488  0.017514 -0.043592 -0.003232 -0.010478  0.027641        2\n",
       "\n",
       "[206209 rows x 7 columns]"
      ]
     },
     "execution_count": 229,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "cluster_df = clusters.set_index('user_id')\n",
    "cluster_df"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 155,
   "metadata": {},
   "outputs": [],
   "source": [
    "cluster_item_rules_dic = {}\n",
    "cluster_item_rules_dic[0] = cluster0_item_rules\n",
    "cluster_item_rules_dic[1] = cluster1_item_rules\n",
    "cluster_item_rules_dic[2] = cluster2_item_rules\n",
    "cluster_item_rules_dic[3] = cluster3_item_rules\n",
    "cluster_item_rules_dic[4] = cluster4_item_rules\n",
    "cluster_item_rules_dic[5] = cluster5_item_rules"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 5,
   "metadata": {},
   "outputs": [],
   "source": [
    "cluster_aisle_rules_dic = {}\n",
    "cluster_aisle_rules_dic[0] = cluster0_aisle_rules\n",
    "cluster_aisle_rules_dic[1] = cluster1_aisle_rules\n",
    "cluster_aisle_rules_dic[2] = cluster2_aisle_rules\n",
    "cluster_aisle_rules_dic[3] = cluster3_aisle_rules\n",
    "cluster_aisle_rules_dic[4] = cluster4_aisle_rules\n",
    "cluster_aisle_rules_dic[5] = cluster5_aisle_rules"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 171,
   "metadata": {},
   "outputs": [],
   "source": [
    "# returns top x items associated based on lift\n",
    "\n",
    "def product_to_product(cluster, product_id, item_lift, product_name, num_products):\n",
    "    df = cluster_item_rules_dic[cluster]\n",
    "    df = df[(df['item_A'] == product_id) | (df['item_B'] == product_id)][['product_name_A','item_A','product_name_B','item_B','confidenceAtoB','lift']].sort_values('lift', ascending = False)\n",
    "    df = df[df['lift'] > item_lift]\n",
    "    df = df.sort_values('lift', ascending = False)\n",
    "    df = df.head(n = num_products)\n",
    "    product_to_product_associations = df['product_name_A'].values.tolist()\n",
    "    for x in df['product_name_B'].values.tolist():\n",
    "        product_to_product_associations.append(x)\n",
    "    product_to_product_associations = [x for x in product_to_product_associations if x != product_name]\n",
    "    return product_to_product_associations"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 237,
   "metadata": {},
   "outputs": [],
   "source": [
    "# returns recommended products given inputs\n",
    "\n",
    "def pdp_recommender(user_id, product_id, item_lift, num_products):\n",
    "    product_name = products.at[product_id,'product_name']\n",
    "    aisle_id = products.at[product_id,'aisle_id']\n",
    "    aisle_name = aisles.at[aisle_id,'aisle']\n",
    "    cluster = cluster_df.at[user_id, 'cluster']\n",
    "    print(cluster)\n",
    "    return product_to_product(cluster = cluster, product_id = product_id, item_lift = item_lift, \n",
    "                              product_name = product_name, num_products = num_products)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 270,
   "metadata": {},
   "outputs": [
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "0\n",
      "cluster 0['Thin & Light Tortilla Chips', 'Red Peppers', 'Organic Lemon', 'Organic Cucumber', 'Organic Grape Tomatoes']\n",
      "1\n",
      "cluster 1[]\n",
      "2\n",
      "cluster 2['Real Guacamole', 'Thin & Light Tortilla Chips', 'Original Hummus', 'Organic Reduced Fat 2% Milk', 'Thick & Crispy Tortilla Chips']\n",
      "3\n",
      "cluster 3[]\n",
      "4\n",
      "cluster 4['Organic Raspberries', 'Banana', 'Organic Strawberries', 'Organic Hass Avocado']\n",
      "5\n",
      "cluster 5['Thin & Light Tortilla Chips', 'Organic Large Brown Grade AA Cage Free Eggs', 'Organic Reduced Fat 2% Milk', 'Organic Large Grade AA Brown Eggs', 'Thick & Crispy Tortilla Chips']\n"
     ]
    }
   ],
   "source": [
    "# results for users in the 6 different clusters\n",
    "\n",
    "users = [6,26,1,81,4,7]\n",
    "i = 0\n",
    "for x in users:\n",
    "    print('cluster ' + str(i) + str(pdp_recommender(x,39055,1,5)))\n",
    "    i = i + 1"
   ]
  }
 ],
 "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.3"
  }
 },
 "nbformat": 4,
 "nbformat_minor": 4
}