content based recommendation_detailed.ipynb

{
 "cells": [
  {
   "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>Unnamed: 0</th>\n",
       "      <th>Title</th>\n",
       "      <th>Year</th>\n",
       "      <th>Rated</th>\n",
       "      <th>Released</th>\n",
       "      <th>Runtime</th>\n",
       "      <th>Genre</th>\n",
       "      <th>Director</th>\n",
       "      <th>Writer</th>\n",
       "      <th>Actors</th>\n",
       "      <th>Plot</th>\n",
       "      <th>Language</th>\n",
       "      <th>Country</th>\n",
       "      <th>Awards</th>\n",
       "      <th>Poster</th>\n",
       "      <th>Ratings.Source</th>\n",
       "      <th>Ratings.Value</th>\n",
       "      <th>Metascore</th>\n",
       "      <th>imdbRating</th>\n",
       "      <th>imdbVotes</th>\n",
       "      <th>imdbID</th>\n",
       "      <th>Type</th>\n",
       "      <th>tomatoMeter</th>\n",
       "      <th>tomatoImage</th>\n",
       "      <th>tomatoRating</th>\n",
       "      <th>tomatoReviews</th>\n",
       "      <th>tomatoFresh</th>\n",
       "      <th>tomatoRotten</th>\n",
       "      <th>tomatoConsensus</th>\n",
       "      <th>tomatoUserMeter</th>\n",
       "      <th>tomatoUserRating</th>\n",
       "      <th>tomatoUserReviews</th>\n",
       "      <th>tomatoURL</th>\n",
       "      <th>DVD</th>\n",
       "      <th>BoxOffice</th>\n",
       "      <th>Production</th>\n",
       "      <th>Website</th>\n",
       "      <th>Response</th>\n",
       "    </tr>\n",
       "  </thead>\n",
       "  <tbody>\n",
       "    <tr>\n",
       "      <th>0</th>\n",
       "      <td>1</td>\n",
       "      <td>The Shawshank Redemption</td>\n",
       "      <td>1994</td>\n",
       "      <td>R</td>\n",
       "      <td>14 Oct 1994</td>\n",
       "      <td>142 min</td>\n",
       "      <td>Crime, Drama</td>\n",
       "      <td>Frank Darabont</td>\n",
       "      <td>Stephen King (short story \"Rita Hayworth and S...</td>\n",
       "      <td>Tim Robbins, Morgan Freeman, Bob Gunton, Willi...</td>\n",
       "      <td>Two imprisoned men bond over a number of years...</td>\n",
       "      <td>English</td>\n",
       "      <td>USA</td>\n",
       "      <td>Nominated for 7 Oscars. Another 19 wins &amp; 30 n...</td>\n",
       "      <td>https://images-na.ssl-images-amazon.com/images...</td>\n",
       "      <td>Internet Movie Database</td>\n",
       "      <td>9.3/10</td>\n",
       "      <td>80.0</td>\n",
       "      <td>9.3</td>\n",
       "      <td>1,825,626</td>\n",
       "      <td>tt0111161</td>\n",
       "      <td>movie</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>NaN</td>\n",
       "      <td>http://www.rottentomatoes.com/m/shawshank_rede...</td>\n",
       "      <td>27 Jan 1998</td>\n",
       "      <td>NaN</td>\n",
       "      <td>Columbia Pictures</td>\n",
       "      <td>NaN</td>\n",
       "      <td>True</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>1</th>\n",
       "      <td>2</td>\n",
       "      <td>The Godfather</td>\n",
       "      <td>1972</td>\n",
       "      <td>R</td>\n",
       "      <td>24 Mar 1972</td>\n",
       "      <td>175 min</td>\n",
       "      <td>Crime, Drama</td>\n",
       "      <td>Francis Ford Coppola</td>\n",
       "      <td>Mario Puzo (screenplay), Francis Ford Coppola ...</td>\n",
       "      <td>Marlon Brando, Al Pacino, James Caan, Richard ...</td>\n",
       "      <td>The aging patriarch of an organized crime dyna...</td>\n",
       "      <td>English, Italian, Latin</td>\n",
       "      <td>USA</td>\n",
       "      <td>Won 3 Oscars. Another 23 wins &amp; 27 nominations.</td>\n",
       "      <td>https://images-na.ssl-images-amazon.com/images...</td>\n",
       "      <td>Internet Movie Database</td>\n",
       "      <td>9.2/10</td>\n",
       "      <td>100.0</td>\n",
       "      <td>9.2</td>\n",
       "      <td>1,243,444</td>\n",
       "      <td>tt0068646</td>\n",
       "      <td>movie</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>NaN</td>\n",
       "      <td>http://www.rottentomatoes.com/m/godfather/</td>\n",
       "      <td>09 Oct 2001</td>\n",
       "      <td>NaN</td>\n",
       "      <td>Paramount Pictures</td>\n",
       "      <td>http://www.thegodfather.com</td>\n",
       "      <td>True</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>2</th>\n",
       "      <td>3</td>\n",
       "      <td>The Godfather: Part II</td>\n",
       "      <td>1974</td>\n",
       "      <td>R</td>\n",
       "      <td>20 Dec 1974</td>\n",
       "      <td>202 min</td>\n",
       "      <td>Crime, Drama</td>\n",
       "      <td>Francis Ford Coppola</td>\n",
       "      <td>Francis Ford Coppola (screenplay), Mario Puzo ...</td>\n",
       "      <td>Al Pacino, Robert Duvall, Diane Keaton, Robert...</td>\n",
       "      <td>The early life and career of Vito Corleone in ...</td>\n",
       "      <td>English, Italian, Spanish, Latin, Sicilian</td>\n",
       "      <td>USA</td>\n",
       "      <td>Won 6 Oscars. Another 10 wins &amp; 20 nominations.</td>\n",
       "      <td>https://images-na.ssl-images-amazon.com/images...</td>\n",
       "      <td>Internet Movie Database</td>\n",
       "      <td>9.0/10</td>\n",
       "      <td>85.0</td>\n",
       "      <td>9.0</td>\n",
       "      <td>856,870</td>\n",
       "      <td>tt0071562</td>\n",
       "      <td>movie</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>NaN</td>\n",
       "      <td>http://www.rottentomatoes.com/m/godfather_part...</td>\n",
       "      <td>24 May 2005</td>\n",
       "      <td>NaN</td>\n",
       "      <td>Paramount Pictures</td>\n",
       "      <td>http://www.thegodfather.com/</td>\n",
       "      <td>True</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>3</th>\n",
       "      <td>4</td>\n",
       "      <td>The Dark Knight</td>\n",
       "      <td>2008</td>\n",
       "      <td>PG-13</td>\n",
       "      <td>18 Jul 2008</td>\n",
       "      <td>152 min</td>\n",
       "      <td>Action, Crime, Drama</td>\n",
       "      <td>Christopher Nolan</td>\n",
       "      <td>Jonathan Nolan (screenplay), Christopher Nolan...</td>\n",
       "      <td>Christian Bale, Heath Ledger, Aaron Eckhart, M...</td>\n",
       "      <td>When the menace known as the Joker emerges fro...</td>\n",
       "      <td>English, Mandarin</td>\n",
       "      <td>USA, UK</td>\n",
       "      <td>Won 2 Oscars. Another 151 wins &amp; 153 nominations.</td>\n",
       "      <td>https://images-na.ssl-images-amazon.com/images...</td>\n",
       "      <td>Internet Movie Database</td>\n",
       "      <td>9.0/10</td>\n",
       "      <td>82.0</td>\n",
       "      <td>9.0</td>\n",
       "      <td>1,802,351</td>\n",
       "      <td>tt0468569</td>\n",
       "      <td>movie</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>NaN</td>\n",
       "      <td>http://www.rottentomatoes.com/m/the_dark_knight/</td>\n",
       "      <td>09 Dec 2008</td>\n",
       "      <td>$533,316,061</td>\n",
       "      <td>Warner Bros. Pictures/Legendary</td>\n",
       "      <td>http://thedarkknight.warnerbros.com/</td>\n",
       "      <td>True</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>4</th>\n",
       "      <td>5</td>\n",
       "      <td>12 Angry Men</td>\n",
       "      <td>1957</td>\n",
       "      <td>APPROVED</td>\n",
       "      <td>01 Apr 1957</td>\n",
       "      <td>96 min</td>\n",
       "      <td>Crime, Drama</td>\n",
       "      <td>Sidney Lumet</td>\n",
       "      <td>Reginald Rose (story), Reginald Rose (screenplay)</td>\n",
       "      <td>Martin Balsam, John Fiedler, Lee J. Cobb, E.G....</td>\n",
       "      <td>A jury holdout attempts to prevent a miscarria...</td>\n",
       "      <td>English</td>\n",
       "      <td>USA</td>\n",
       "      <td>Nominated for 3 Oscars. Another 16 wins &amp; 8 no...</td>\n",
       "      <td>https://images-na.ssl-images-amazon.com/images...</td>\n",
       "      <td>Internet Movie Database</td>\n",
       "      <td>8.9/10</td>\n",
       "      <td>96.0</td>\n",
       "      <td>8.9</td>\n",
       "      <td>494,215</td>\n",
       "      <td>tt0050083</td>\n",
       "      <td>movie</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>NaN</td>\n",
       "      <td>http://www.rottentomatoes.com/m/1000013-12_ang...</td>\n",
       "      <td>06 Mar 2001</td>\n",
       "      <td>NaN</td>\n",
       "      <td>Criterion Collection</td>\n",
       "      <td>http://www.criterion.com/films/27871-12-angry-men</td>\n",
       "      <td>True</td>\n",
       "    </tr>\n",
       "  </tbody>\n",
       "</table>\n",
       "</div>"
      ],
      "text/plain": [
       "   Unnamed: 0                     Title  Year     Rated     Released  Runtime  \\\n",
       "0           1  The Shawshank Redemption  1994         R  14 Oct 1994  142 min   \n",
       "1           2             The Godfather  1972         R  24 Mar 1972  175 min   \n",
       "2           3    The Godfather: Part II  1974         R  20 Dec 1974  202 min   \n",
       "3           4           The Dark Knight  2008     PG-13  18 Jul 2008  152 min   \n",
       "4           5              12 Angry Men  1957  APPROVED  01 Apr 1957   96 min   \n",
       "\n",
       "                  Genre              Director  \\\n",
       "0          Crime, Drama        Frank Darabont   \n",
       "1          Crime, Drama  Francis Ford Coppola   \n",
       "2          Crime, Drama  Francis Ford Coppola   \n",
       "3  Action, Crime, Drama     Christopher Nolan   \n",
       "4          Crime, Drama          Sidney Lumet   \n",
       "\n",
       "                                              Writer  \\\n",
       "0  Stephen King (short story \"Rita Hayworth and S...   \n",
       "1  Mario Puzo (screenplay), Francis Ford Coppola ...   \n",
       "2  Francis Ford Coppola (screenplay), Mario Puzo ...   \n",
       "3  Jonathan Nolan (screenplay), Christopher Nolan...   \n",
       "4  Reginald Rose (story), Reginald Rose (screenplay)   \n",
       "\n",
       "                                              Actors  \\\n",
       "0  Tim Robbins, Morgan Freeman, Bob Gunton, Willi...   \n",
       "1  Marlon Brando, Al Pacino, James Caan, Richard ...   \n",
       "2  Al Pacino, Robert Duvall, Diane Keaton, Robert...   \n",
       "3  Christian Bale, Heath Ledger, Aaron Eckhart, M...   \n",
       "4  Martin Balsam, John Fiedler, Lee J. Cobb, E.G....   \n",
       "\n",
       "                                                Plot  \\\n",
       "0  Two imprisoned men bond over a number of years...   \n",
       "1  The aging patriarch of an organized crime dyna...   \n",
       "2  The early life and career of Vito Corleone in ...   \n",
       "3  When the menace known as the Joker emerges fro...   \n",
       "4  A jury holdout attempts to prevent a miscarria...   \n",
       "\n",
       "                                     Language  Country  \\\n",
       "0                                     English      USA   \n",
       "1                     English, Italian, Latin      USA   \n",
       "2  English, Italian, Spanish, Latin, Sicilian      USA   \n",
       "3                           English, Mandarin  USA, UK   \n",
       "4                                     English      USA   \n",
       "\n",
       "                                              Awards  \\\n",
       "0  Nominated for 7 Oscars. Another 19 wins & 30 n...   \n",
       "1    Won 3 Oscars. Another 23 wins & 27 nominations.   \n",
       "2    Won 6 Oscars. Another 10 wins & 20 nominations.   \n",
       "3  Won 2 Oscars. Another 151 wins & 153 nominations.   \n",
       "4  Nominated for 3 Oscars. Another 16 wins & 8 no...   \n",
       "\n",
       "                                              Poster           Ratings.Source  \\\n",
       "0  https://images-na.ssl-images-amazon.com/images...  Internet Movie Database   \n",
       "1  https://images-na.ssl-images-amazon.com/images...  Internet Movie Database   \n",
       "2  https://images-na.ssl-images-amazon.com/images...  Internet Movie Database   \n",
       "3  https://images-na.ssl-images-amazon.com/images...  Internet Movie Database   \n",
       "4  https://images-na.ssl-images-amazon.com/images...  Internet Movie Database   \n",
       "\n",
       "  Ratings.Value  Metascore  imdbRating  imdbVotes     imdbID   Type  \\\n",
       "0        9.3/10       80.0         9.3  1,825,626  tt0111161  movie   \n",
       "1        9.2/10      100.0         9.2  1,243,444  tt0068646  movie   \n",
       "2        9.0/10       85.0         9.0    856,870  tt0071562  movie   \n",
       "3        9.0/10       82.0         9.0  1,802,351  tt0468569  movie   \n",
       "4        8.9/10       96.0         8.9    494,215  tt0050083  movie   \n",
       "\n",
       "   tomatoMeter  tomatoImage  tomatoRating  tomatoReviews  tomatoFresh  \\\n",
       "0          NaN          NaN           NaN            NaN          NaN   \n",
       "1          NaN          NaN           NaN            NaN          NaN   \n",
       "2          NaN          NaN           NaN            NaN          NaN   \n",
       "3          NaN          NaN           NaN            NaN          NaN   \n",
       "4          NaN          NaN           NaN            NaN          NaN   \n",
       "\n",
       "   tomatoRotten  tomatoConsensus  tomatoUserMeter  tomatoUserRating  \\\n",
       "0           NaN              NaN              NaN               NaN   \n",
       "1           NaN              NaN              NaN               NaN   \n",
       "2           NaN              NaN              NaN               NaN   \n",
       "3           NaN              NaN              NaN               NaN   \n",
       "4           NaN              NaN              NaN               NaN   \n",
       "\n",
       "   tomatoUserReviews                                          tomatoURL  \\\n",
       "0                NaN  http://www.rottentomatoes.com/m/shawshank_rede...   \n",
       "1                NaN         http://www.rottentomatoes.com/m/godfather/   \n",
       "2                NaN  http://www.rottentomatoes.com/m/godfather_part...   \n",
       "3                NaN   http://www.rottentomatoes.com/m/the_dark_knight/   \n",
       "4                NaN  http://www.rottentomatoes.com/m/1000013-12_ang...   \n",
       "\n",
       "           DVD     BoxOffice                       Production  \\\n",
       "0  27 Jan 1998           NaN                Columbia Pictures   \n",
       "1  09 Oct 2001           NaN               Paramount Pictures   \n",
       "2  24 May 2005           NaN               Paramount Pictures   \n",
       "3  09 Dec 2008  $533,316,061  Warner Bros. Pictures/Legendary   \n",
       "4  06 Mar 2001           NaN             Criterion Collection   \n",
       "\n",
       "                                             Website  Response  \n",
       "0                                                NaN      True  \n",
       "1                        http://www.thegodfather.com      True  \n",
       "2                       http://www.thegodfather.com/      True  \n",
       "3               http://thedarkknight.warnerbros.com/      True  \n",
       "4  http://www.criterion.com/films/27871-12-angry-men      True  "
      ]
     },
     "execution_count": 2,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "import pandas as pd\n",
    "import numpy as np\n",
    "from sklearn.metrics.pairwise import cosine_similarity\n",
    "from sklearn.feature_extraction.text import CountVectorizer\n",
    "\n",
    "pd.set_option('display.max_columns', 100)\n",
    "df = pd.read_csv('https://query.data.world/s/uikepcpffyo2nhig52xxeevdialfl7')\n",
    "df.head()"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 3,
   "metadata": {},
   "outputs": [
    {
     "data": {
      "text/plain": [
       "(250, 38)"
      ]
     },
     "execution_count": 3,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "df.shape"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 4,
   "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>Title</th>\n",
       "      <th>Genre</th>\n",
       "      <th>Director</th>\n",
       "      <th>Actors</th>\n",
       "      <th>Plot</th>\n",
       "    </tr>\n",
       "  </thead>\n",
       "  <tbody>\n",
       "    <tr>\n",
       "      <th>0</th>\n",
       "      <td>The Shawshank Redemption</td>\n",
       "      <td>Crime, Drama</td>\n",
       "      <td>Frank Darabont</td>\n",
       "      <td>Tim Robbins, Morgan Freeman, Bob Gunton, Willi...</td>\n",
       "      <td>Two imprisoned men bond over a number of years...</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>1</th>\n",
       "      <td>The Godfather</td>\n",
       "      <td>Crime, Drama</td>\n",
       "      <td>Francis Ford Coppola</td>\n",
       "      <td>Marlon Brando, Al Pacino, James Caan, Richard ...</td>\n",
       "      <td>The aging patriarch of an organized crime dyna...</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>2</th>\n",
       "      <td>The Godfather: Part II</td>\n",
       "      <td>Crime, Drama</td>\n",
       "      <td>Francis Ford Coppola</td>\n",
       "      <td>Al Pacino, Robert Duvall, Diane Keaton, Robert...</td>\n",
       "      <td>The early life and career of Vito Corleone in ...</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>3</th>\n",
       "      <td>The Dark Knight</td>\n",
       "      <td>Action, Crime, Drama</td>\n",
       "      <td>Christopher Nolan</td>\n",
       "      <td>Christian Bale, Heath Ledger, Aaron Eckhart, M...</td>\n",
       "      <td>When the menace known as the Joker emerges fro...</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>4</th>\n",
       "      <td>12 Angry Men</td>\n",
       "      <td>Crime, Drama</td>\n",
       "      <td>Sidney Lumet</td>\n",
       "      <td>Martin Balsam, John Fiedler, Lee J. Cobb, E.G....</td>\n",
       "      <td>A jury holdout attempts to prevent a miscarria...</td>\n",
       "    </tr>\n",
       "  </tbody>\n",
       "</table>\n",
       "</div>"
      ],
      "text/plain": [
       "                      Title                 Genre              Director  \\\n",
       "0  The Shawshank Redemption          Crime, Drama        Frank Darabont   \n",
       "1             The Godfather          Crime, Drama  Francis Ford Coppola   \n",
       "2    The Godfather: Part II          Crime, Drama  Francis Ford Coppola   \n",
       "3           The Dark Knight  Action, Crime, Drama     Christopher Nolan   \n",
       "4              12 Angry Men          Crime, Drama          Sidney Lumet   \n",
       "\n",
       "                                              Actors  \\\n",
       "0  Tim Robbins, Morgan Freeman, Bob Gunton, Willi...   \n",
       "1  Marlon Brando, Al Pacino, James Caan, Richard ...   \n",
       "2  Al Pacino, Robert Duvall, Diane Keaton, Robert...   \n",
       "3  Christian Bale, Heath Ledger, Aaron Eckhart, M...   \n",
       "4  Martin Balsam, John Fiedler, Lee J. Cobb, E.G....   \n",
       "\n",
       "                                                Plot  \n",
       "0  Two imprisoned men bond over a number of years...  \n",
       "1  The aging patriarch of an organized crime dyna...  \n",
       "2  The early life and career of Vito Corleone in ...  \n",
       "3  When the menace known as the Joker emerges fro...  \n",
       "4  A jury holdout attempts to prevent a miscarria...  "
      ]
     },
     "execution_count": 4,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "df = df[['Title','Genre','Director','Actors','Plot']]\n",
    "df.head()"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 5,
   "metadata": {},
   "outputs": [],
   "source": [
    "# discarding the commas between the actors' full names and getting only the first three names\n",
    "df['Actors'] = df['Actors'].map(lambda x: x.split(',')[:3])"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 6,
   "metadata": {},
   "outputs": [
    {
     "data": {
      "text/plain": [
       "0            [Tim Robbins,  Morgan Freeman,  Bob Gunton]\n",
       "1               [Marlon Brando,  Al Pacino,  James Caan]\n",
       "2             [Al Pacino,  Robert Duvall,  Diane Keaton]\n",
       "3        [Christian Bale,  Heath Ledger,  Aaron Eckhart]\n",
       "4           [Martin Balsam,  John Fiedler,  Lee J. Cobb]\n",
       "5           [Liam Neeson,  Ben Kingsley,  Ralph Fiennes]\n",
       "6                [Noel Appleby,  Ali Astin,  Sean Astin]\n",
       "7           [Tim Roth,  Amanda Plummer,  Laura Lovelace]\n",
       "8                [Edward Norton,  Brad Pitt,  Meat Loaf]\n",
       "9              [Alan Howard,  Noel Appleby,  Sean Astin]\n",
       "10          [Tom Hanks,  Rebecca Williams,  Sally Field]\n",
       "11         [Mark Hamill,  Harrison Ford,  Carrie Fisher]\n",
       "12     [Leonardo DiCaprio,  Joseph Gordon-Levitt,  El...\n",
       "13             [Bruce Allpress,  Sean Astin,  John Bach]\n",
       "14     [Michael Berryman,  Peter Brocco,  Dean R. Bro...\n",
       "15             [Robert De Niro,  Ray Liotta,  Joe Pesci]\n",
       "16     [Keanu Reeves,  Laurence Fishburne,  Carrie-An...\n",
       "17         [Mark Hamill,  Harrison Ford,  Carrie Fisher]\n",
       "18     [Morgan Freeman,  Andrew Kevin Walker,  Kevin ...\n",
       "19       [James Stewart,  Donna Reed,  Lionel Barrymore]\n",
       "20     [Jodie Foster,  Lawrence A. Bonney,  Kasi Lemm...\n",
       "21     [Stephen Baldwin,  Gabriel Byrne,  Benicio Del...\n",
       "22           [Jean Reno,  Gary Oldman,  Natalie Portman]\n",
       "23             [Tom Hanks,  Tom Sizemore,  Edward Burns]\n",
       "24      [Virginia Cherrill,  Florence Lee,  Harry Myers]\n",
       "25     [Ellen Burstyn,  Matthew McConaughey,  Mackenz...\n",
       "26     [Edward Norton,  Edward Furlong,  Beverly D'An...\n",
       "27     [Charles Chaplin,  Paulette Goddard,  Henry Be...\n",
       "28     [Humphrey Bogart,  Ingrid Bergman,  Paul Henreid]\n",
       "29     [Tom Hanks,  David Morse,  Michael Clarke Duncan]\n",
       "                             ...                        \n",
       "220         [Cary Grant,  Ingrid Bergman,  Claude Rains]\n",
       "221    [Wallace Ford,  William H. Lynn,  Victoria Horne]\n",
       "222       [Matt Damon,  Jessica Chastain,  Kristen Wiig]\n",
       "223        [Ellen Burstyn,  Max von Sydow,  Lee J. Cobb]\n",
       "224            [John Wayne,  Dean Martin,  Ricky Nelson]\n",
       "225     [Cary Grant,  Katharine Hepburn,  James Stewart]\n",
       "226         [John Dall,  Farley Granger,  Edith Evanson]\n",
       "227     [Humphrey Bogart,  Lauren Bacall,  John Ridgely]\n",
       "228    [Bob Geldof,  Christine Hargreaves,  James Lau...\n",
       "229    [Colin Firth,  Helena Bonham Carter,  Derek Ja...\n",
       "230    [Melinda Dillon,  Darren McGavin,  Peter Billi...\n",
       "231    [Anne Bancroft,  Dustin Hoffman,  Katharine Ross]\n",
       "232      [Sally Kirkland,  Anthony Ramirez,  Ray LePere]\n",
       "233    [Billy Bob Thornton,  Dwight Yoakam,  J.T. Walsh]\n",
       "234    [Leonardo DiCaprio,  Djimon Hounsou,  Jennifer...\n",
       "235        [Pat Healy,  Genevieve Zweig,  Mark Flanagan]\n",
       "236       [Dustin Hoffman,  Tom Cruise,  Valeria Golino]\n",
       "237    [Leonardo DiCaprio,  Tom Hardy,  Domhnall Glee...\n",
       "238    [Danny Elfman,  Chris Sarandon,  Catherine O'H...\n",
       "239      [Frank Sinatra,  Laurence Harvey,  Janet Leigh]\n",
       "240             [Ryan Reynolds,  Karan Soni,  Ed Skrein]\n",
       "241     [William Holden,  Ernest Borgnine,  Robert Ryan]\n",
       "242      [Scott Weinger,  Robin Williams,  Linda Larkin]\n",
       "243       [Ewan McGregor,  Albert Finney,  Billy Crudup]\n",
       "244      [George C. Scott,  Karl Malden,  Stephen Young]\n",
       "245           [Ray Milland,  Jane Wyman,  Phillip Terry]\n",
       "246    [Brie Larson,  John Gallagher Jr.,  Stephanie ...\n",
       "247      [Cary Grant,  Rosalind Russell,  Ralph Bellamy]\n",
       "248    [Sissy Spacek,  Jane Galloway Heitz,  Joseph A...\n",
       "249           [Dev Patel,  Saurabh Shukla,  Anil Kapoor]\n",
       "Name: Actors, Length: 250, dtype: object"
      ]
     },
     "execution_count": 6,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "df['Actors']"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 7,
   "metadata": {},
   "outputs": [],
   "source": [
    "# putting the genres in a list of words\n",
    "df['Genre'] = df['Genre'].map(lambda x: x.lower().split(','))"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 8,
   "metadata": {},
   "outputs": [
    {
     "data": {
      "text/plain": [
       "0                       [crime,  drama]\n",
       "1                       [crime,  drama]\n",
       "2                       [crime,  drama]\n",
       "3              [action,  crime,  drama]\n",
       "4                       [crime,  drama]\n",
       "5         [biography,  drama,  history]\n",
       "6         [adventure,  drama,  fantasy]\n",
       "7                       [crime,  drama]\n",
       "8                               [drama]\n",
       "9         [adventure,  drama,  fantasy]\n",
       "10           [comedy,  drama,  romance]\n",
       "11       [action,  adventure,  fantasy]\n",
       "12        [action,  adventure,  sci-fi]\n",
       "13        [adventure,  drama,  fantasy]\n",
       "14                              [drama]\n",
       "15                      [crime,  drama]\n",
       "16                    [action,  sci-fi]\n",
       "17       [action,  adventure,  fantasy]\n",
       "18            [crime,  drama,  mystery]\n",
       "19           [drama,  family,  fantasy]\n",
       "20           [crime,  drama,  thriller]\n",
       "21            [crime,  drama,  mystery]\n",
       "22           [crime,  drama,  thriller]\n",
       "23                        [drama,  war]\n",
       "24           [comedy,  drama,  romance]\n",
       "25         [adventure,  drama,  sci-fi]\n",
       "26                      [crime,  drama]\n",
       "27            [comedy,  drama,  family]\n",
       "28              [drama,  romance,  war]\n",
       "29            [crime,  drama,  fantasy]\n",
       "                     ...               \n",
       "220       [drama,  film-noir,  romance]\n",
       "221          [comedy,  drama,  fantasy]\n",
       "222        [adventure,  drama,  sci-fi]\n",
       "223                            [horror]\n",
       "224          [action,  drama,  western]\n",
       "225                  [comedy,  romance]\n",
       "226          [crime,  drama,  thriller]\n",
       "227       [crime,  film-noir,  mystery]\n",
       "228       [animation,  drama,  fantasy]\n",
       "229                 [biography,  drama]\n",
       "230                   [comedy,  family]\n",
       "231                    [comedy,  drama]\n",
       "232        [drama,  history,  thriller]\n",
       "233                             [drama]\n",
       "234      [adventure,  drama,  thriller]\n",
       "235                             [drama]\n",
       "236                             [drama]\n",
       "237      [adventure,  drama,  thriller]\n",
       "238      [animation,  family,  fantasy]\n",
       "239                  [drama,  thriller]\n",
       "240       [action,  adventure,  comedy]\n",
       "241      [action,  adventure,  western]\n",
       "242    [animation,  adventure,  comedy]\n",
       "243       [adventure,  drama,  fantasy]\n",
       "244           [biography,  drama,  war]\n",
       "245                 [drama,  film-noir]\n",
       "246                             [drama]\n",
       "247          [comedy,  drama,  romance]\n",
       "248                 [biography,  drama]\n",
       "249                             [drama]\n",
       "Name: Genre, Length: 250, dtype: object"
      ]
     },
     "execution_count": 8,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "df['Genre']"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 9,
   "metadata": {},
   "outputs": [],
   "source": [
    "df['Director'] = df['Director'].map(lambda x: x.split(' '))"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 10,
   "metadata": {},
   "outputs": [
    {
     "data": {
      "text/plain": [
       "0                         [Frank, Darabont]\n",
       "1                  [Francis, Ford, Coppola]\n",
       "2                  [Francis, Ford, Coppola]\n",
       "3                      [Christopher, Nolan]\n",
       "4                           [Sidney, Lumet]\n",
       "5                       [Steven, Spielberg]\n",
       "6                          [Peter, Jackson]\n",
       "7                      [Quentin, Tarantino]\n",
       "8                          [David, Fincher]\n",
       "9                          [Peter, Jackson]\n",
       "10                       [Robert, Zemeckis]\n",
       "11                        [Irvin, Kershner]\n",
       "12                     [Christopher, Nolan]\n",
       "13                         [Peter, Jackson]\n",
       "14                          [Milos, Forman]\n",
       "15                       [Martin, Scorsese]\n",
       "16     [Lana, Wachowski,, Lilly, Wachowski]\n",
       "17                          [George, Lucas]\n",
       "18                         [David, Fincher]\n",
       "19                           [Frank, Capra]\n",
       "20                        [Jonathan, Demme]\n",
       "21                          [Bryan, Singer]\n",
       "22                            [Luc, Besson]\n",
       "23                      [Steven, Spielberg]\n",
       "24                       [Charles, Chaplin]\n",
       "25                     [Christopher, Nolan]\n",
       "26                             [Tony, Kaye]\n",
       "27                       [Charles, Chaplin]\n",
       "28                        [Michael, Curtiz]\n",
       "29                        [Frank, Darabont]\n",
       "                       ...                 \n",
       "220                     [Alfred, Hitchcock]\n",
       "221                         [Henry, Koster]\n",
       "222                         [Ridley, Scott]\n",
       "223                     [William, Friedkin]\n",
       "224                         [Howard, Hawks]\n",
       "225                         [George, Cukor]\n",
       "226                     [Alfred, Hitchcock]\n",
       "227                         [Howard, Hawks]\n",
       "228                          [Alan, Parker]\n",
       "229                           [Tom, Hooper]\n",
       "230                            [Bob, Clark]\n",
       "231                         [Mike, Nichols]\n",
       "232                         [Oliver, Stone]\n",
       "233                  [Billy, Bob, Thornton]\n",
       "234                         [Edward, Zwick]\n",
       "235                [Paul, Thomas, Anderson]\n",
       "236                       [Barry, Levinson]\n",
       "237         [Alejandro, González, Iñárritu]\n",
       "238                         [Henry, Selick]\n",
       "239                   [John, Frankenheimer]\n",
       "240                           [Tim, Miller]\n",
       "241                        [Sam, Peckinpah]\n",
       "242          [Ron, Clements,, John, Musker]\n",
       "243                           [Tim, Burton]\n",
       "244               [Franklin, J., Schaffner]\n",
       "245                         [Billy, Wilder]\n",
       "246               [Destin, Daniel, Cretton]\n",
       "247                         [Howard, Hawks]\n",
       "248                          [David, Lynch]\n",
       "249       [Danny, Boyle,, Loveleen, Tandan]\n",
       "Name: Director, Length: 250, dtype: object"
      ]
     },
     "execution_count": 10,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "df['Director']"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 11,
   "metadata": {},
   "outputs": [],
   "source": [
    "for index, row in df.iterrows():\n",
    "    row['Actors'] = [x.lower().replace(' ','') for x in row['Actors']]\n",
    "    row['Director'] = ''.join(row['Director']).lower()"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 12,
   "metadata": {},
   "outputs": [
    {
     "data": {
      "text/plain": [
       "0                 [timrobbins, morganfreeman, bobgunton]\n",
       "1                    [marlonbrando, alpacino, jamescaan]\n",
       "2                  [alpacino, robertduvall, dianekeaton]\n",
       "3             [christianbale, heathledger, aaroneckhart]\n",
       "4                 [martinbalsam, johnfiedler, leej.cobb]\n",
       "5                [liamneeson, benkingsley, ralphfiennes]\n",
       "6                     [noelappleby, aliastin, seanastin]\n",
       "7                [timroth, amandaplummer, lauralovelace]\n",
       "8                     [edwardnorton, bradpitt, meatloaf]\n",
       "9                   [alanhoward, noelappleby, seanastin]\n",
       "10               [tomhanks, rebeccawilliams, sallyfield]\n",
       "11              [markhamill, harrisonford, carriefisher]\n",
       "12     [leonardodicaprio, josephgordon-levitt, ellenp...\n",
       "13                  [bruceallpress, seanastin, johnbach]\n",
       "14          [michaelberryman, peterbrocco, deanr.brooks]\n",
       "15                   [robertdeniro, rayliotta, joepesci]\n",
       "16     [keanureeves, laurencefishburne, carrie-annemoss]\n",
       "17              [markhamill, harrisonford, carriefisher]\n",
       "18       [morganfreeman, andrewkevinwalker, kevinspacey]\n",
       "19            [jamesstewart, donnareed, lionelbarrymore]\n",
       "20          [jodiefoster, lawrencea.bonney, kasilemmons]\n",
       "21        [stephenbaldwin, gabrielbyrne, beniciodeltoro]\n",
       "22                [jeanreno, garyoldman, natalieportman]\n",
       "23                  [tomhanks, tomsizemore, edwardburns]\n",
       "24           [virginiacherrill, florencelee, harrymyers]\n",
       "25      [ellenburstyn, matthewmcconaughey, mackenziefoy]\n",
       "26        [edwardnorton, edwardfurlong, beverlyd'angelo]\n",
       "27       [charleschaplin, paulettegoddard, henrybergman]\n",
       "28          [humphreybogart, ingridbergman, paulhenreid]\n",
       "29           [tomhanks, davidmorse, michaelclarkeduncan]\n",
       "                             ...                        \n",
       "220              [carygrant, ingridbergman, clauderains]\n",
       "221          [wallaceford, williamh.lynn, victoriahorne]\n",
       "222            [mattdamon, jessicachastain, kristenwiig]\n",
       "223               [ellenburstyn, maxvonsydow, leej.cobb]\n",
       "224                 [johnwayne, deanmartin, rickynelson]\n",
       "225          [carygrant, katharinehepburn, jamesstewart]\n",
       "226              [johndall, farleygranger, edithevanson]\n",
       "227          [humphreybogart, laurenbacall, johnridgely]\n",
       "228     [bobgeldof, christinehargreaves, jameslaurenson]\n",
       "229        [colinfirth, helenabonhamcarter, derekjacobi]\n",
       "230     [melindadillon, darrenmcgavin, peterbillingsley]\n",
       "231         [annebancroft, dustinhoffman, katharineross]\n",
       "232           [sallykirkland, anthonyramirez, raylepere]\n",
       "233          [billybobthornton, dwightyoakam, j.t.walsh]\n",
       "234    [leonardodicaprio, djimonhounsou, jenniferconn...\n",
       "235             [pathealy, genevievezweig, markflanagan]\n",
       "236            [dustinhoffman, tomcruise, valeriagolino]\n",
       "237        [leonardodicaprio, tomhardy, domhnallgleeson]\n",
       "238        [dannyelfman, chrissarandon, catherineo'hara]\n",
       "239           [franksinatra, laurenceharvey, janetleigh]\n",
       "240                  [ryanreynolds, karansoni, edskrein]\n",
       "241          [williamholden, ernestborgnine, robertryan]\n",
       "242           [scottweinger, robinwilliams, lindalarkin]\n",
       "243            [ewanmcgregor, albertfinney, billycrudup]\n",
       "244            [georgec.scott, karlmalden, stephenyoung]\n",
       "245                [raymilland, janewyman, phillipterry]\n",
       "246     [brielarson, johngallagherjr., stephaniebeatriz]\n",
       "247           [carygrant, rosalindrussell, ralphbellamy]\n",
       "248    [sissyspacek, janegallowayheitz, josepha.carpe...\n",
       "249                [devpatel, saurabhshukla, anilkapoor]\n",
       "Name: Actors, Length: 250, dtype: object"
      ]
     },
     "execution_count": 12,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "df['Actors']"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 13,
   "metadata": {},
   "outputs": [
    {
     "data": {
      "text/plain": [
       "0                     frankdarabont\n",
       "1                francisfordcoppola\n",
       "2                francisfordcoppola\n",
       "3                  christophernolan\n",
       "4                       sidneylumet\n",
       "5                   stevenspielberg\n",
       "6                      peterjackson\n",
       "7                  quentintarantino\n",
       "8                      davidfincher\n",
       "9                      peterjackson\n",
       "10                   robertzemeckis\n",
       "11                    irvinkershner\n",
       "12                 christophernolan\n",
       "13                     peterjackson\n",
       "14                      milosforman\n",
       "15                   martinscorsese\n",
       "16     lanawachowski,lillywachowski\n",
       "17                      georgelucas\n",
       "18                     davidfincher\n",
       "19                       frankcapra\n",
       "20                    jonathandemme\n",
       "21                      bryansinger\n",
       "22                        lucbesson\n",
       "23                  stevenspielberg\n",
       "24                   charleschaplin\n",
       "25                 christophernolan\n",
       "26                         tonykaye\n",
       "27                   charleschaplin\n",
       "28                    michaelcurtiz\n",
       "29                    frankdarabont\n",
       "                   ...             \n",
       "220                 alfredhitchcock\n",
       "221                     henrykoster\n",
       "222                     ridleyscott\n",
       "223                 williamfriedkin\n",
       "224                     howardhawks\n",
       "225                     georgecukor\n",
       "226                 alfredhitchcock\n",
       "227                     howardhawks\n",
       "228                      alanparker\n",
       "229                       tomhooper\n",
       "230                        bobclark\n",
       "231                     mikenichols\n",
       "232                     oliverstone\n",
       "233                billybobthornton\n",
       "234                     edwardzwick\n",
       "235              paulthomasanderson\n",
       "236                   barrylevinson\n",
       "237       alejandrogonzáleziñárritu\n",
       "238                     henryselick\n",
       "239               johnfrankenheimer\n",
       "240                       timmiller\n",
       "241                    sampeckinpah\n",
       "242          ronclements,johnmusker\n",
       "243                       timburton\n",
       "244             franklinj.schaffner\n",
       "245                     billywilder\n",
       "246             destindanielcretton\n",
       "247                     howardhawks\n",
       "248                      davidlynch\n",
       "249       dannyboyle,loveleentandan\n",
       "Name: Director, Length: 250, dtype: object"
      ]
     },
     "execution_count": 13,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "df['Director']"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 15,
   "metadata": {},
   "outputs": [],
   "source": [
    "from rake_nltk import Rake"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 16,
   "metadata": {},
   "outputs": [],
   "source": [
    "# initializing the new column\n",
    "df['Key_words'] = \"\"\n",
    "\n",
    "for index, row in df.iterrows():\n",
    "    plot = row['Plot']\n",
    "    \n",
    "    # instantiating Rake, by default is uses english stopwords from NLTK\n",
    "    # and discard all puntuation characters\n",
    "    r = Rake()\n",
    "\n",
    "    # extracting the words by passing the text\n",
    "    r.extract_keywords_from_text(plot)\n",
    "\n",
    "    # getting the dictionary whith key words and their scores\n",
    "    key_words_dict_scores = r.get_word_degrees()\n",
    "    \n",
    "    # assigning the key words to the new column\n",
    "    row['Key_words'] = list(key_words_dict_scores.keys())\n",
    "\n",
    "# dropping the Plot column\n",
    "df.drop(columns = ['Plot'], inplace = True)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 20,
   "metadata": {},
   "outputs": [
    {
     "data": {
      "text/plain": [
       "defaultdict(<function rake_nltk.rake.Rake._build_word_co_occurance_graph.<locals>.<lambda>()>,\n",
       "            {'cheating': 1,\n",
       "             'slums': 1,\n",
       "             'indian': 2,\n",
       "             'version': 2,\n",
       "             'wants': 1,\n",
       "             'millionaire': 2,\n",
       "             '?\"': 2,\n",
       "             'upbringing': 1,\n",
       "             'mumbai': 3,\n",
       "             'teen': 3,\n",
       "             'reflects': 3,\n",
       "             'accused': 1})"
      ]
     },
     "execution_count": 20,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "key_words_dict_scores"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 21,
   "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>Genre</th>\n",
       "      <th>Director</th>\n",
       "      <th>Actors</th>\n",
       "      <th>Key_words</th>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>Title</th>\n",
       "      <th></th>\n",
       "      <th></th>\n",
       "      <th></th>\n",
       "      <th></th>\n",
       "    </tr>\n",
       "  </thead>\n",
       "  <tbody>\n",
       "    <tr>\n",
       "      <th>The Shawshank Redemption</th>\n",
       "      <td>[crime,  drama]</td>\n",
       "      <td>frankdarabont</td>\n",
       "      <td>[timrobbins, morganfreeman, bobgunton]</td>\n",
       "      <td>[eventual, redemption, years, two, imprisoned,...</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>The Godfather</th>\n",
       "      <td>[crime,  drama]</td>\n",
       "      <td>francisfordcoppola</td>\n",
       "      <td>[marlonbrando, alpacino, jamescaan]</td>\n",
       "      <td>[clandestine, empire, aging, patriarch, reluct...</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>The Godfather: Part II</th>\n",
       "      <td>[crime,  drama]</td>\n",
       "      <td>francisfordcoppola</td>\n",
       "      <td>[alpacino, robertduvall, dianekeaton]</td>\n",
       "      <td>[son, tightens, expands, grip, early, life, 19...</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>The Dark Knight</th>\n",
       "      <td>[action,  crime,  drama]</td>\n",
       "      <td>christophernolan</td>\n",
       "      <td>[christianbale, heathledger, aaroneckhart]</td>\n",
       "      <td>[wreaks, havoc, gotham, physical, tests, myste...</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>12 Angry Men</th>\n",
       "      <td>[crime,  drama]</td>\n",
       "      <td>sidneylumet</td>\n",
       "      <td>[martinbalsam, johnfiedler, leej.cobb]</td>\n",
       "      <td>[reconsider, forcing, evidence, prevent, justi...</td>\n",
       "    </tr>\n",
       "  </tbody>\n",
       "</table>\n",
       "</div>"
      ],
      "text/plain": [
       "                                             Genre            Director  \\\n",
       "Title                                                                    \n",
       "The Shawshank Redemption           [crime,  drama]       frankdarabont   \n",
       "The Godfather                      [crime,  drama]  francisfordcoppola   \n",
       "The Godfather: Part II             [crime,  drama]  francisfordcoppola   \n",
       "The Dark Knight           [action,  crime,  drama]    christophernolan   \n",
       "12 Angry Men                       [crime,  drama]         sidneylumet   \n",
       "\n",
       "                                                              Actors  \\\n",
       "Title                                                                  \n",
       "The Shawshank Redemption      [timrobbins, morganfreeman, bobgunton]   \n",
       "The Godfather                    [marlonbrando, alpacino, jamescaan]   \n",
       "The Godfather: Part II         [alpacino, robertduvall, dianekeaton]   \n",
       "The Dark Knight           [christianbale, heathledger, aaroneckhart]   \n",
       "12 Angry Men                  [martinbalsam, johnfiedler, leej.cobb]   \n",
       "\n",
       "                                                                  Key_words  \n",
       "Title                                                                        \n",
       "The Shawshank Redemption  [eventual, redemption, years, two, imprisoned,...  \n",
       "The Godfather             [clandestine, empire, aging, patriarch, reluct...  \n",
       "The Godfather: Part II    [son, tightens, expands, grip, early, life, 19...  \n",
       "The Dark Knight           [wreaks, havoc, gotham, physical, tests, myste...  \n",
       "12 Angry Men              [reconsider, forcing, evidence, prevent, justi...  "
      ]
     },
     "execution_count": 21,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "df.set_index('Title', inplace = True)\n",
    "df.head()"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 22,
   "metadata": {},
   "outputs": [],
   "source": [
    "df['bag_of_words'] = ''\n",
    "columns = df.columns\n",
    "for index, row in df.iterrows():\n",
    "    words = ''\n",
    "    for col in columns:\n",
    "        if col != 'Director':\n",
    "            words = words + ' '.join(row[col])+ ' '\n",
    "        else:\n",
    "            words = words + row[col]+ ' '\n",
    "    row['bag_of_words'] = words\n",
    "    \n",
    "df.drop(columns = [col for col in df.columns if col!= 'bag_of_words'], inplace = True)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 23,
   "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>bag_of_words</th>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>Title</th>\n",
       "      <th></th>\n",
       "    </tr>\n",
       "  </thead>\n",
       "  <tbody>\n",
       "    <tr>\n",
       "      <th>The Shawshank Redemption</th>\n",
       "      <td>crime  drama frankdarabont timrobbins morganfr...</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>The Godfather</th>\n",
       "      <td>crime  drama francisfordcoppola marlonbrando a...</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>The Godfather: Part II</th>\n",
       "      <td>crime  drama francisfordcoppola alpacino rober...</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>The Dark Knight</th>\n",
       "      <td>action  crime  drama christophernolan christia...</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>12 Angry Men</th>\n",
       "      <td>crime  drama sidneylumet martinbalsam johnfied...</td>\n",
       "    </tr>\n",
       "  </tbody>\n",
       "</table>\n",
       "</div>"
      ],
      "text/plain": [
       "                                                               bag_of_words\n",
       "Title                                                                      \n",
       "The Shawshank Redemption  crime  drama frankdarabont timrobbins morganfr...\n",
       "The Godfather             crime  drama francisfordcoppola marlonbrando a...\n",
       "The Godfather: Part II    crime  drama francisfordcoppola alpacino rober...\n",
       "The Dark Knight           action  crime  drama christophernolan christia...\n",
       "12 Angry Men              crime  drama sidneylumet martinbalsam johnfied..."
      ]
     },
     "execution_count": 23,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "df.head()"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 24,
   "metadata": {},
   "outputs": [
    {
     "data": {
      "text/plain": [
       "0    The Shawshank Redemption\n",
       "1               The Godfather\n",
       "2      The Godfather: Part II\n",
       "3             The Dark Knight\n",
       "4                12 Angry Men\n",
       "Name: Title, dtype: object"
      ]
     },
     "execution_count": 24,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "# instantiating and generating the count matrix\n",
    "count = CountVectorizer()\n",
    "count_matrix = count.fit_transform(df['bag_of_words'])\n",
    "\n",
    "# creating a Series for the movie titles so they are associated to an ordered numerical\n",
    "# list I will use later to match the indexes\n",
    "indices = pd.Series(df.index)\n",
    "indices[:5]"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 25,
   "metadata": {},
   "outputs": [
    {
     "data": {
      "text/plain": [
       "<250x2961 sparse matrix of type '<class 'numpy.int64'>'\n",
       "\twith 5342 stored elements in Compressed Sparse Row format>"
      ]
     },
     "execution_count": 25,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "count_matrix "
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 26,
   "metadata": {},
   "outputs": [],
   "source": [
    "c=count_matrix.todense()"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 27,
   "metadata": {},
   "outputs": [
    {
     "data": {
      "text/plain": [
       "matrix([[0, 0, 0, ..., 0, 0, 0],\n",
       "        [0, 0, 0, ..., 0, 0, 0],\n",
       "        [0, 0, 0, ..., 0, 0, 0],\n",
       "        ...,\n",
       "        [0, 0, 0, ..., 0, 0, 0],\n",
       "        [0, 0, 0, ..., 0, 0, 0],\n",
       "        [0, 0, 0, ..., 0, 0, 0]], dtype=int64)"
      ]
     },
     "execution_count": 27,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "c"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 28,
   "metadata": {},
   "outputs": [
    {
     "data": {
      "text/plain": [
       "scipy.sparse.csr.csr_matrix"
      ]
     },
     "execution_count": 28,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "type(count_matrix)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 30,
   "metadata": {},
   "outputs": [
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "  (0, 655)\t1\n",
      "  (0, 519)\t1\n",
      "  (0, 2481)\t1\n",
      "  (0, 969)\t1\n",
      "  (0, 1899)\t1\n",
      "  (0, 59)\t1\n",
      "  (0, 311)\t1\n",
      "  (0, 1733)\t1\n",
      "  (0, 1269)\t1\n",
      "  (0, 2765)\t1\n",
      "  (0, 2950)\t1\n",
      "  (0, 2174)\t1\n",
      "  (0, 888)\t1\n",
      "  (0, 306)\t1\n",
      "  (0, 1810)\t1\n",
      "  (0, 2678)\t1\n",
      "  (0, 1011)\t1\n",
      "  (0, 768)\t1\n",
      "  (0, 584)\t1\n"
     ]
    }
   ],
   "source": [
    "print(count_matrix[0,:])"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 34,
   "metadata": {},
   "outputs": [
    {
     "data": {
      "text/plain": [
       "{'crime': 584,\n",
       " 'drama': 768,\n",
       " 'frankdarabont': 1011,\n",
       " 'timrobbins': 2678,\n",
       " 'morganfreeman': 1810,\n",
       " 'bobgunton': 306,\n",
       " 'eventual': 888,\n",
       " 'redemption': 2174,\n",
       " 'years': 2950,\n",
       " 'two': 2765,\n",
       " 'imprisoned': 1269,\n",
       " 'men': 1733,\n",
       " 'bond': 311,\n",
       " 'acts': 59,\n",
       " 'number': 1899,\n",
       " 'finding': 969,\n",
       " 'solace': 2481,\n",
       " 'common': 519,\n",
       " 'decency': 655,\n",
       " 'francisfordcoppola': 1007,\n",
       " 'marlonbrando': 1671,\n",
       " 'alpacino': 118,\n",
       " 'jamescaan': 1355,\n",
       " 'clandestine': 475,\n",
       " 'empire': 846,\n",
       " 'aging': 82,\n",
       " 'patriarch': 1972,\n",
       " 'reluctant': 2193,\n",
       " 'son': 2492,\n",
       " 'organized': 1932,\n",
       " 'dynasty': 793,\n",
       " 'transfers': 2730,\n",
       " 'control': 553,\n",
       " 'robertduvall': 2264,\n",
       " 'dianekeaton': 710,\n",
       " 'tightens': 2670,\n",
       " 'expands': 908,\n",
       " 'grip': 1131,\n",
       " 'early': 795,\n",
       " 'life': 1575,\n",
       " '1920s': 8,\n",
       " 'new': 1870,\n",
       " 'york': 2953,\n",
       " 'portrayed': 2057,\n",
       " 'career': 391,\n",
       " 'michael': 1750,\n",
       " 'vito': 2854,\n",
       " 'corleone': 563,\n",
       " 'family': 934,\n",
       " 'syndicate': 2608,\n",
       " 'action': 56,\n",
       " 'christophernolan': 464,\n",
       " 'christianbale': 458,\n",
       " 'heathledger': 1179,\n",
       " 'aaroneckhart': 31,\n",
       " 'wreaks': 2943,\n",
       " 'havoc': 1173,\n",
       " 'gotham': 1113,\n",
       " 'physical': 2024,\n",
       " 'tests': 2646,\n",
       " 'mysterious': 1844,\n",
       " 'past': 1966,\n",
       " 'ability': 37,\n",
       " 'chaos': 430,\n",
       " 'joker': 1444,\n",
       " 'emerges': 836,\n",
       " 'menace': 1734,\n",
       " 'known': 1512,\n",
       " 'dark': 625,\n",
       " 'knight': 1508,\n",
       " 'must': 1841,\n",
       " 'accept': 41,\n",
       " 'one': 1924,\n",
       " 'fight': 959,\n",
       " 'injustice': 1298,\n",
       " 'greatest': 1127,\n",
       " 'psychological': 2119,\n",
       " 'people': 1991,\n",
       " 'sidneylumet': 2443,\n",
       " 'martinbalsam': 1678,\n",
       " 'johnfiedler': 1424,\n",
       " 'leej': 1556,\n",
       " 'cobb': 492,\n",
       " 'reconsider': 2168,\n",
       " 'forcing': 991,\n",
       " 'evidence': 891,\n",
       " 'prevent': 2079,\n",
       " 'justice': 1472,\n",
       " 'colleagues': 499,\n",
       " 'jury': 1471,\n",
       " 'holdout': 1213,\n",
       " 'attempts': 212,\n",
       " 'miscarriage': 1784,\n",
       " 'biography': 290,\n",
       " 'history': 1206,\n",
       " 'stevenspielberg': 2545,\n",
       " 'liamneeson': 1573,\n",
       " 'benkingsley': 277,\n",
       " 'ralphfiennes': 2144,\n",
       " 'german': 1082,\n",
       " 'persecution': 1997,\n",
       " 'nazi': 1856,\n",
       " 'germans': 1083,\n",
       " 'world': 2935,\n",
       " 'war': 2870,\n",
       " 'ii': 1260,\n",
       " 'witnessing': 2920,\n",
       " 'jewish': 1399,\n",
       " 'workforce': 2932,\n",
       " 'oskar': 1938,\n",
       " 'schindler': 2354,\n",
       " 'gradually': 1117,\n",
       " 'becomes': 257,\n",
       " 'concerned': 532,\n",
       " 'occupied': 1910,\n",
       " 'poland': 2048,\n",
       " 'adventure': 71,\n",
       " 'fantasy': 938,\n",
       " 'peterjackson': 2011,\n",
       " 'noelappleby': 1888,\n",
       " 'aliastin': 106,\n",
       " 'seanastin': 2369,\n",
       " 'sam': 2328,\n",
       " 'draw': 769,\n",
       " 'sauron': 2341,\n",
       " 'gaze': 1061,\n",
       " 'frodo': 1032,\n",
       " 'approach': 172,\n",
       " 'mount': 1816,\n",
       " 'doom': 757,\n",
       " 'ring': 2250,\n",
       " 'gandalf': 1048,\n",
       " 'army': 180,\n",
       " 'aragorn': 174,\n",
       " 'lead': 1544,\n",
       " 'quentintarantino': 2131,\n",
       " 'timroth': 2679,\n",
       " 'amandaplummer': 124,\n",
       " 'lauralovelace': 1534,\n",
       " 'diner': 716,\n",
       " 'bandits': 232,\n",
       " 'intertwine': 1316,\n",
       " 'violence': 2847,\n",
       " 'pair': 1950,\n",
       " 'boxer': 330,\n",
       " 'mob': 1795,\n",
       " 'hit': 1207,\n",
       " 'gangster': 1051,\n",
       " 'four': 1003,\n",
       " 'tales': 2614,\n",
       " 'wife': 2896,\n",
       " 'lives': 1592,\n",
       " 'davidfincher': 631,\n",
       " 'edwardnorton': 816,\n",
       " 'bradpitt': 335,\n",
       " 'meatloaf': 1715,\n",
       " 'underground': 2783,\n",
       " 'club': 488,\n",
       " 'looking': 1607,\n",
       " 'much': 1826,\n",
       " 'something': 2489,\n",
       " 'care': 390,\n",
       " 'soap': 2477,\n",
       " 'maker': 1644,\n",
       " 'may': 1707,\n",
       " 'forming': 998,\n",
       " 'evolves': 893,\n",
       " 'insomniac': 1304,\n",
       " 'office': 1914,\n",
       " 'worker': 2931,\n",
       " 'crosses': 594,\n",
       " 'paths': 1969,\n",
       " 'change': 428,\n",
       " 'way': 2877,\n",
       " 'devil': 702,\n",
       " 'alanhoward': 90,\n",
       " 'journey': 1462,\n",
       " 'destroy': 689,\n",
       " 'save': 2343,\n",
       " 'middle': 1768,\n",
       " 'earth': 796,\n",
       " 'lord': 1609,\n",
       " 'powerful': 2069,\n",
       " 'eight': 822,\n",
       " 'companions': 522,\n",
       " 'set': 2406,\n",
       " 'meek': 1722,\n",
       " 'hobbit': 1210,\n",
       " 'shire': 2432,\n",
       " 'comedy': 511,\n",
       " 'romance': 2289,\n",
       " 'robertzemeckis': 2275,\n",
       " 'tomhanks': 2686,\n",
       " 'rebeccawilliams': 2164,\n",
       " 'sallyfield': 2326,\n",
       " 'accidentally': 45,\n",
       " 'true': 2751,\n",
       " 'love': 1614,\n",
       " 'jenny': 1390,\n",
       " 'curran': 600,\n",
       " 'forrest': 1000,\n",
       " 'gump': 1141,\n",
       " 'present': 2077,\n",
       " 'intelligent': 1311,\n",
       " 'many': 1658,\n",
       " 'historic': 1205,\n",
       " 'moments': 1800,\n",
       " 'eludes': 832,\n",
       " 'irvinkershner': 1334,\n",
       " 'markhamill': 1668,\n",
       " 'harrisonford': 1167,\n",
       " 'carriefisher': 399,\n",
       " 'capture': 387,\n",
       " 'luke': 1622,\n",
       " 'overpowered': 1944,\n",
       " 'plan': 2033,\n",
       " 'skywalker': 2466,\n",
       " 'begins': 264,\n",
       " 'jedi': 1385,\n",
       " 'training': 2728,\n",
       " 'friends': 1030,\n",
       " 'shelter': 2427,\n",
       " 'questionable': 2133,\n",
       " 'ally': 115,\n",
       " 'rebels': 2165,\n",
       " 'master': 1689,\n",
       " 'yoda': 2952,\n",
       " 'newly': 1872,\n",
       " 'established': 880,\n",
       " 'base': 245,\n",
       " 'darth': 628,\n",
       " 'vader': 2815,\n",
       " 'hunts': 1250,\n",
       " 'sci': 2357,\n",
       " 'fi': 956,\n",
       " 'leonardodicaprio': 1566,\n",
       " 'josephgordon': 1457,\n",
       " 'levitt': 1569,\n",
       " 'ellenpage': 831,\n",
       " 'dream': 771,\n",
       " 'idea': 1255,\n",
       " 'planting': 2038,\n",
       " 'ceo': 423,\n",
       " 'inverse': 1318,\n",
       " 'task': 2619,\n",
       " 'steals': 2535,\n",
       " 'corporate': 564,\n",
       " 'secrets': 2382,\n",
       " 'sharing': 2419,\n",
       " 'technology': 2625,\n",
       " 'mind': 1782,\n",
       " 'given': 1097,\n",
       " 'use': 2810,\n",
       " 'thief': 2654,\n",
       " 'bruceallpress': 355,\n",
       " 'johnbach': 1418,\n",
       " 'shifty': 2430,\n",
       " 'gollum': 1109,\n",
       " 'divided': 741,\n",
       " 'fellowship': 953,\n",
       " 'makes': 1645,\n",
       " 'mordor': 1809,\n",
       " 'saruman': 2339,\n",
       " 'isengard': 1335,\n",
       " 'edge': 802,\n",
       " 'closer': 486,\n",
       " 'hordes': 1224,\n",
       " 'stand': 2522,\n",
       " 'help': 1187,\n",
       " 'milosforman': 1781,\n",
       " 'michaelberryman': 1753,\n",
       " 'peterbrocco': 2009,\n",
       " 'deanr': 646,\n",
       " 'brooks': 351,\n",
       " 'criminal': 586,\n",
       " 'pleads': 2042,\n",
       " 'insanity': 1301,\n",
       " 'mental': 1736,\n",
       " 'institution': 1307,\n",
       " 'rallies': 2142,\n",
       " 'scared': 2349,\n",
       " 'patients': 1971,\n",
       " 'trouble': 2749,\n",
       " 'getting': 1089,\n",
       " 'oppressive': 1929,\n",
       " 'nurse': 1900,\n",
       " 'martinscorsese': 1680,\n",
       " 'robertdeniro': 2262,\n",
       " 'rayliotta': 2154,\n",
       " 'joepesci': 1415,\n",
       " 'mafia': 1634,\n",
       " 'partners': 1963,\n",
       " 'jimmy': 1401,\n",
       " 'conway': 559,\n",
       " 'karen': 1476,\n",
       " 'hill': 1202,\n",
       " 'tommy': 2694,\n",
       " 'devitto': 703,\n",
       " 'teen': 2626,\n",
       " 'henry': 1191,\n",
       " 'relationship': 2188,\n",
       " 'italian': 1339,\n",
       " 'american': 127,\n",
       " 'story': 2551,\n",
       " 'covering': 574,\n",
       " 'lanawachowski': 1520,\n",
       " 'lillywachowski': 1579,\n",
       " 'keanureeves': 1484,\n",
       " 'laurencefishburne': 1536,\n",
       " 'carrie': 397,\n",
       " 'annemoss': 151,\n",
       " 'nature': 1853,\n",
       " 'controllers': 554,\n",
       " 'reality': 2160,\n",
       " 'role': 2287,\n",
       " 'computer': 530,\n",
       " 'hacker': 1151,\n",
       " 'learns': 1550,\n",
       " 'georgelucas': 1076,\n",
       " 'rescue': 2207,\n",
       " 'princess': 2085,\n",
       " 'leia': 1564,\n",
       " 'also': 119,\n",
       " 'attempting': 211,\n",
       " 'cocky': 493,\n",
       " 'pilot': 2027,\n",
       " 'joins': 1443,\n",
       " 'forces': 990,\n",
       " 'droids': 778,\n",
       " 'destroying': 690,\n",
       " 'battle': 250,\n",
       " 'wookiee': 2929,\n",
       " 'station': 2532,\n",
       " 'evil': 892,\n",
       " 'galaxy': 1043,\n",
       " 'mystery': 1845,\n",
       " 'andrewkevinwalker': 138,\n",
       " 'kevinspacey': 1493,\n",
       " 'detectives': 696,\n",
       " 'seven': 2410,\n",
       " 'deadly': 642,\n",
       " 'sins': 2457,\n",
       " 'hunt': 1247,\n",
       " 'rookie': 2296,\n",
       " 'veteran': 2827,\n",
       " 'serial': 2402,\n",
       " 'killer': 1497,\n",
       " 'motives': 1815,\n",
       " 'uses': 2811,\n",
       " 'frankcapra': 1010,\n",
       " 'jamesstewart': 1364,\n",
       " 'donnareed': 754,\n",
       " 'lionelbarrymore': 1588,\n",
       " 'like': 1576,\n",
       " 'showing': 2439,\n",
       " 'angel': 141,\n",
       " 'sent': 2398,\n",
       " 'heaven': 1180,\n",
       " 'never': 1869,\n",
       " 'existed': 906,\n",
       " 'would': 2941,\n",
       " 'desperately': 686,\n",
       " 'frustrated': 1034,\n",
       " 'businessman': 370,\n",
       " 'thriller': 2666,\n",
       " 'jonathandemme': 1446,\n",
       " 'jodiefoster': 1411,\n",
       " 'lawrencea': 1542,\n",
       " 'bonney': 312,\n",
       " 'kasilemmons': 1480,\n",
       " 'young': 2954,\n",
       " 'catching': 413,\n",
       " 'another': 155,\n",
       " 'receive': 2166,\n",
       " 'cadet': 374,\n",
       " 'confide': 537,\n",
       " 'skins': 2465,\n",
       " 'victims': 2831,\n",
       " 'incarcerated': 1274,\n",
       " 'manipulative': 1654,\n",
       " 'bryansinger': 359,\n",
       " 'stephenbaldwin': 2538,\n",
       " 'gabrielbyrne': 1042,\n",
       " 'beniciodeltoro': 276,\n",
       " 'seemingly': 2387,\n",
       " 'random': 2147,\n",
       " 'police': 2049,\n",
       " 'lineup': 1586,\n",
       " 'began': 262,\n",
       " 'horrific': 1225,\n",
       " 'gun': 1142,\n",
       " 'boat': 302,\n",
       " 'five': 976,\n",
       " 'criminals': 588,\n",
       " 'met': 1746,\n",
       " 'twisty': 2764,\n",
       " 'events': 887,\n",
       " 'leading': 1546,\n",
       " 'sole': 2485,\n",
       " 'survivor': 2600,\n",
       " 'tells': 2631,\n",
       " 'lucbesson': 1620,\n",
       " 'jeanreno': 1384,\n",
       " 'garyoldman': 1056,\n",
       " 'natalieportman': 1852,\n",
       " 'unusual': 2799,\n",
       " 'year': 2949,\n",
       " '12': 3,\n",
       " 'reluctantly': 2194,\n",
       " 'taken': 2610,\n",
       " 'protégée': 2115,\n",
       " 'léon': 1626,\n",
       " 'murdered': 1831,\n",
       " 'old': 1920,\n",
       " 'girl': 1095,\n",
       " 'professional': 2101,\n",
       " 'assassin': 195,\n",
       " 'trade': 2722,\n",
       " 'mathilda': 1695,\n",
       " 'form': 996,\n",
       " 'tomsizemore': 2696,\n",
       " 'edwardburns': 812,\n",
       " 'killed': 1496,\n",
       " 'normandy': 1891,\n",
       " 'landings': 1523,\n",
       " 'paratrooper': 1956,\n",
       " 'whose': 2894,\n",
       " 'brothers': 353,\n",
       " 'following': 984,\n",
       " 'group': 1133,\n",
       " 'retrieve': 2218,\n",
       " 'soldiers': 2484,\n",
       " 'go': 1104,\n",
       " 'behind': 266,\n",
       " 'enemy': 853,\n",
       " 'lines': 1585,\n",
       " 'charleschaplin': 437,\n",
       " 'virginiacherrill': 2849,\n",
       " 'florencelee': 978,\n",
       " 'harrymyers': 1170,\n",
       " 'eyed': 919,\n",
       " 'tramp': 2729,\n",
       " 'medically': 1720,\n",
       " 'fallen': 930,\n",
       " 'able': 38,\n",
       " 'wealthy': 2879,\n",
       " 'erratic': 875,\n",
       " 'tippler': 2680,\n",
       " 'dewy': 706,\n",
       " 'sightless': 2445,\n",
       " 'flower': 979,\n",
       " 'accumulates': 48,\n",
       " 'money': 1802,\n",
       " 'aid': 86,\n",
       " 'ellenburstyn': 829,\n",
       " 'matthewmcconaughey': 1701,\n",
       " 'mackenziefoy': 1628,\n",
       " 'ensure': 865,\n",
       " 'humanity': 1241,\n",
       " 'team': 2623,\n",
       " 'explorers': 915,\n",
       " 'travel': 2735,\n",
       " 'wormhole': 2936,\n",
       " 'attempt': 210,\n",
       " 'survival': 2596,\n",
       " 'space': 2499,\n",
       " 'tonykaye': 2701,\n",
       " 'edwardfurlong': 814,\n",
       " 'beverlyd': 282,\n",
       " 'angelo': 143,\n",
       " 'skinhead': 2464,\n",
       " 'tries': 2746,\n",
       " 'former': 997,\n",
       " 'neo': 1866,\n",
       " 'wrong': 2946,\n",
       " 'path': 1967,\n",
       " 'going': 1107,\n",
       " 'younger': 2955,\n",
       " 'brother': 352,\n",
       " 'paulettegoddard': 1977,\n",
       " 'henrybergman': 1192,\n",
       " 'live': 1591,\n",
       " 'modern': 1796,\n",
       " 'industrial': 1281,\n",
       " 'society': 2480,\n",
       " 'homeless': 1220,\n",
       " 'woman': 2924,\n",
       " 'struggles': 2564,\n",
       " 'michaelcurtiz': 1758,\n",
       " 'humphreybogart': 1245,\n",
       " 'ingridbergman': 1296,\n",
       " 'paulhenreid': 1980,\n",
       " 'casablanca': 407,\n",
       " 'unforeseen': 2786,\n",
       " 'complications': 529,\n",
       " 'lover': 1616,\n",
       " 'december': 654,\n",
       " '1941': 11,\n",
       " 'cynical': 607,\n",
       " 'expatriate': 909,\n",
       " 'encounters': 850,\n",
       " 'davidmorse': 634,\n",
       " 'michaelclarkeduncan': 1757,\n",
       " 'gift': 1093,\n",
       " 'rape': 2148,\n",
       " 'black': 295,\n",
       " 'man': 1648,\n",
       " 'accused': 50,\n",
       " 'child': 447,\n",
       " 'murder': 1830,\n",
       " 'guards': 1138,\n",
       " 'charges': 434,\n",
       " 'yet': 2951,\n",
       " 'affected': 75,\n",
       " 'death': 647,\n",
       " 'row': 2300,\n",
       " 'horror': 1226,\n",
       " 'alfredhitchcock': 104,\n",
       " 'anthonyperkins': 160,\n",
       " 'veramiles': 2822,\n",
       " 'johngavin': 1429,\n",
       " 'checks': 444,\n",
       " 'employer': 847,\n",
       " 'client': 481,\n",
       " 'goes': 1106,\n",
       " '000': 0,\n",
       " 'run': 2306,\n",
       " 'domination': 751,\n",
       " 'mother': 1814,\n",
       " 'phoenix': 2021,\n",
       " 'secretary': 2381,\n",
       " 'embezzles': 835,\n",
       " 'remote': 2199,\n",
       " 'motel': 1813,\n",
       " '40': 28,\n",
       " 'karenallen': 1477,\n",
       " 'paulfreeman': 1978,\n",
       " 'ark': 179,\n",
       " 'find': 968,\n",
       " 'adventurer': 72,\n",
       " 'indiana': 1279,\n",
       " 'jones': 1448,\n",
       " 'covenant': 572,\n",
       " 'nazis': 1857,\n",
       " 'hired': 1203,\n",
       " 'government': 1114,\n",
       " 'archaeologist': 175,\n",
       " 'music': 1837,\n",
       " 'romanpolanski': 2290,\n",
       " 'adrienbrody': 68,\n",
       " 'emiliafox': 838,\n",
       " 'michalzebrowski': 1764,\n",
       " 'destruction': 692,\n",
       " 'survive': 2597,\n",
       " 'polish': 2051,\n",
       " 'musician': 1839,\n",
       " 'warsaw': 2874,\n",
       " 'ghetto': 1090,\n",
       " 'gracekelly': 1116,\n",
       " 'wendellcorey': 2887,\n",
       " 'bound': 324,\n",
       " 'photographer': 2022,\n",
       " 'spies': 2509,\n",
       " 'apartment': 169,\n",
       " 'window': 2912,\n",
       " 'convinced': 558,\n",
       " 'neighbours': 1863,\n",
       " 'committed': 518,\n",
       " 'wheelchair': 2891,\n",
       " 'mattdamon': 1696,\n",
       " 'jacknicholson': 1346,\n",
       " 'irish': 1332,\n",
       " 'gang': 1050,\n",
       " 'mole': 1798,\n",
       " 'undercover': 2781,\n",
       " 'cop': 560,\n",
       " 'south': 2498,\n",
       " 'boston': 322,\n",
       " 'infiltrating': 1287,\n",
       " 'identify': 1257,\n",
       " 'damienchazelle': 614,\n",
       " 'milesteller': 1774,\n",
       " 'simmons': 2451,\n",
       " 'paulreiser': 1982,\n",
       " 'mentored': 1739,\n",
       " 'greatness': 1128,\n",
       " 'stop': 2549,\n",
       " 'nothing': 1895,\n",
       " 'potential': 2065,\n",
       " 'realize': 2161,\n",
       " 'student': 2568,\n",
       " 'cut': 603,\n",
       " 'promising': 2108,\n",
       " 'drummer': 781,\n",
       " 'enrolls': 862,\n",
       " 'dreams': 772,\n",
       " 'instructor': 1308,\n",
       " 'throat': 2667,\n",
       " 'conservatory': 549,\n",
       " 'jamescameron': 1356,\n",
       " 'arnoldschwarzenegger': 181,\n",
       " 'lindahamilton': 1581,\n",
       " 'ten': 2635,\n",
       " 'cyborg': 605,\n",
       " 'kill': 1495,\n",
       " 'sarah': 2338,\n",
       " 'connor': 547,\n",
       " 'identical': 1256,\n",
       " 'advanced': 70,\n",
       " 'failed': 927,\n",
       " 'protect': 2114,\n",
       " 'john': 1417,\n",
       " 'michaelj': 1760,\n",
       " 'fox': 1004,\n",
       " 'christopherlloyd': 463,\n",
       " 'leathompson': 1551,\n",
       " '30': 27,\n",
       " 'marty': 1683,\n",
       " 'mcfly': 1711,\n",
       " 'close': 485,\n",
       " 'friend': 1029,\n",
       " '17': 4,\n",
       " 'traveling': 2736,\n",
       " 'delorean': 672,\n",
       " 'invented': 1317,\n",
       " 'maverick': 1704,\n",
       " 'scientist': 2358,\n",
       " 'doc': 744,\n",
       " 'brown': 354,\n",
       " 'time': 2673,\n",
       " 'high': 1200,\n",
       " 'school': 2355,\n",
       " 'ridleyscott': 2246,\n",
       " 'russellcrowe': 2312,\n",
       " 'joaquinphoenix': 1409,\n",
       " 'connienielsen': 546,\n",
       " 'seek': 2384,\n",
       " 'revenge': 2226,\n",
       " 'comes': 512,\n",
       " 'roman': 2288,\n",
       " 'general': 1064,\n",
       " 'corrupt': 565,\n",
       " 'betrayed': 280,\n",
       " 'emperor': 845,\n",
       " 'rome': 2291,\n",
       " 'gladiator': 1099,\n",
       " 'animation': 148,\n",
       " 'rogerallers': 2284,\n",
       " 'robminkoff': 2278,\n",
       " 'rowanatkinson': 2301,\n",
       " 'matthewbroderick': 1699,\n",
       " 'niketacalame': 1884,\n",
       " 'please': 2043,\n",
       " 'others': 1939,\n",
       " 'testing': 2645,\n",
       " 'identity': 1258,\n",
       " 'penchant': 1988,\n",
       " 'future': 1040,\n",
       " 'king': 1502,\n",
       " 'simba': 2450,\n",
       " 'searches': 2376,\n",
       " 'lion': 1587,\n",
       " 'cub': 597,\n",
       " 'eagerness': 794,\n",
       " 'boundaries': 325,\n",
       " 'sometimes': 2490,\n",
       " 'gets': 1088,\n",
       " 'hughjackman': 1238,\n",
       " 'michaelcaine': 1755,\n",
       " 'competitive': 526,\n",
       " 'stage': 2520,\n",
       " 'magicians': 1639,\n",
       " 'engage': 854,\n",
       " 'create': 579,\n",
       " 'upmanship': 2803,\n",
       " 'ultimate': 2771,\n",
       " 'illusion': 1262,\n",
       " 'martinsheen': 1681,\n",
       " 'cambodia': 379,\n",
       " 'renegade': 2200,\n",
       " 'colonel': 505,\n",
       " 'god': 1105,\n",
       " 'among': 131,\n",
       " 'captain': 384,\n",
       " 'willard': 2899,\n",
       " 'assassinate': 196,\n",
       " 'dangerous': 617,\n",
       " 'mission': 1789,\n",
       " 'vietnam': 2840,\n",
       " 'local': 1597,\n",
       " 'tribe': 2742,\n",
       " 'guypearce': 1149,\n",
       " 'joepantoliano': 1414,\n",
       " 'obstacle': 1907,\n",
       " 'short': 2435,\n",
       " 'term': 2637,\n",
       " 'memory': 1732,\n",
       " 'loss': 1611,\n",
       " 'keeping': 1487,\n",
       " 'murderer': 1832,\n",
       " 'juggles': 1467,\n",
       " 'searching': 2377,\n",
       " 'jackoakie': 1347,\n",
       " 'reginaldgardiner': 2184,\n",
       " 'hynkel': 1254,\n",
       " 'expand': 907,\n",
       " 'regime': 2183,\n",
       " 'avoid': 222,\n",
       " 'poor': 2055,\n",
       " 'barber': 239,\n",
       " 'dictator': 712,\n",
       " 'adenoid': 64,\n",
       " 'film': 965,\n",
       " 'noir': 1889,\n",
       " 'billywilder': 289,\n",
       " 'williamholden': 2904,\n",
       " 'gloriaswanson': 1103,\n",
       " 'erichvonstroheim': 872,\n",
       " 'screenwriter': 2365,\n",
       " 'faded': 925,\n",
       " 'silent': 2448,\n",
       " 'star': 2527,\n",
       " 'script': 2366,\n",
       " 'rework': 2228,\n",
       " 'developing': 701,\n",
       " 'tomskerritt': 2697,\n",
       " 'sigourneyweaver': 2447,\n",
       " 'veronicacartwright': 2824,\n",
       " 'source': 2497,\n",
       " 'moon': 1806,\n",
       " 'finds': 970,\n",
       " 'merely': 1743,\n",
       " 'begun': 265,\n",
       " 'soon': 2493,\n",
       " 'landing': 1522,\n",
       " 'cycle': 606,\n",
       " 'crew': 583,\n",
       " 'attacked': 208,\n",
       " 'merchant': 1741,\n",
       " 'vessel': 2826,\n",
       " 'perceives': 1993,\n",
       " 'distress': 737,\n",
       " 'call': 377,\n",
       " 'unknown': 2793,\n",
       " 'transmission': 2732,\n",
       " 'stanleykubrick': 2526,\n",
       " 'petersellers': 2013,\n",
       " 'georgec': 1073,\n",
       " 'scott': 2360,\n",
       " 'sterlinghayden': 2541,\n",
       " 'nuclear': 1898,\n",
       " 'holocaust': 1217,\n",
       " 'insane': 1300,\n",
       " 'triggers': 2747,\n",
       " 'generals': 1065,\n",
       " 'frantically': 1017,\n",
       " 'try': 2754,\n",
       " 'politicians': 2053,\n",
       " 'room': 2297,\n",
       " 'full': 1036,\n",
       " 'kirkdouglas': 1505,\n",
       " 'ralphmeeker': 2146,\n",
       " 'adolphemenjou': 66,\n",
       " 'position': 2059,\n",
       " 'refusing': 2181,\n",
       " 'cowardice': 575,\n",
       " 'commanding': 515,\n",
       " 'officer': 1915,\n",
       " 'defend': 665,\n",
       " 'accuses': 51,\n",
       " 'attack': 207,\n",
       " 'western': 2890,\n",
       " 'jamiefoxx': 1366,\n",
       " 'christophwaltz': 467,\n",
       " 'bounty': 326,\n",
       " 'hunter': 1248,\n",
       " 'freed': 1026,\n",
       " 'slave': 2467,\n",
       " 'sets': 2407,\n",
       " 'brutal': 357,\n",
       " 'mississippi': 1790,\n",
       " 'plantation': 2037,\n",
       " 'owner': 1947,\n",
       " 'shelleyduvall': 2425,\n",
       " 'dannylloyd': 623,\n",
       " 'isolated': 1337,\n",
       " 'hotel': 1232,\n",
       " 'spiritual': 2512,\n",
       " 'presence': 2076,\n",
       " 'influences': 1289,\n",
       " 'winter': 2914,\n",
       " 'heads': 1175,\n",
       " 'father': 944,\n",
       " 'psychic': 2118,\n",
       " 'sees': 2389,\n",
       " 'forebodings': 992,\n",
       " 'tyronepower': 2770,\n",
       " 'marlenedietrich': 1670,\n",
       " 'charleslaughton': 438,\n",
       " 'british': 349,\n",
       " 'barrister': 242,\n",
       " 'surprise': 2595,\n",
       " 'trial': 2741,\n",
       " 'tomhardy': 2687,\n",
       " 'enigmatic': 859,\n",
       " 'selina': 2393,\n",
       " 'anarchy': 133,\n",
       " 'city': 473,\n",
       " 'reign': 2187,\n",
       " 'forced': 989,\n",
       " 'imposed': 1267,\n",
       " 'exile': 904,\n",
       " 'guerrilla': 1139,\n",
       " 'terrorist': 2641,\n",
       " 'bane': 233,\n",
       " 'total': 2710,\n",
       " 'annihilation': 154,\n",
       " 'andrewstanton': 139,\n",
       " 'benburtt': 272,\n",
       " 'elissaknight': 825,\n",
       " 'jeffgarlin': 1387,\n",
       " 'collecting': 500,\n",
       " 'robot': 2279,\n",
       " 'inadvertently': 1273,\n",
       " 'embarks': 834,\n",
       " 'fate': 943,\n",
       " 'distant': 736,\n",
       " 'mankind': 1656,\n",
       " 'small': 2474,\n",
       " 'waste': 2875,\n",
       " 'ultimately': 2772,\n",
       " 'decide': 657,\n",
       " 'sammendes': 2330,\n",
       " 'annettebening': 152,\n",
       " 'thorabirch': 2659,\n",
       " 'best': 279,\n",
       " 'crisis': 590,\n",
       " 'daughter': 629,\n",
       " 'sexually': 2414,\n",
       " 'suburban': 2574,\n",
       " 'mid': 1767,\n",
       " 'becoming': 258,\n",
       " 'infatuated': 1285,\n",
       " 'sergioleone': 2401,\n",
       " 'jameswoods': 1365,\n",
       " 'elizabethmcgovern': 827,\n",
       " 'prohibition': 2107,\n",
       " 'ghosts': 1091,\n",
       " 'era': 870,\n",
       " 'returns': 2222,\n",
       " 'manhattan': 1651,\n",
       " 'lower': 1619,\n",
       " 'east': 797,\n",
       " 'side': 2441,\n",
       " 'confront': 542,\n",
       " 'regrets': 2186,\n",
       " 'thirty': 2657,\n",
       " 'later': 1529,\n",
       " 'carriehenn': 400,\n",
       " 'michaelbiehn': 1754,\n",
       " '1979': 19,\n",
       " 'colonized': 506,\n",
       " 'lost': 1612,\n",
       " 'impressive': 1268,\n",
       " 'firepower': 972,\n",
       " 'alien': 108,\n",
       " 'contact': 552,\n",
       " 'enough': 861,\n",
       " 'orsonwelles': 1937,\n",
       " 'josephcotten': 1456,\n",
       " 'dorothycomingore': 761,\n",
       " 'agnesmoorehead': 83,\n",
       " 'final': 966,\n",
       " 'utterance': 2813,\n",
       " 'news': 1873,\n",
       " 'reporters': 2204,\n",
       " 'scramble': 2362,\n",
       " 'discover': 722,\n",
       " 'publishing': 2123,\n",
       " 'tycoon': 2766,\n",
       " 'meaning': 1712,\n",
       " 'carygrant': 406,\n",
       " 'evamariesaint': 885,\n",
       " 'jamesmason': 1361,\n",
       " 'pursued': 2126,\n",
       " 'across': 55,\n",
       " 'agent': 80,\n",
       " 'hapless': 1159,\n",
       " 'advertising': 74,\n",
       " 'executive': 901,\n",
       " 'country': 568,\n",
       " 'looks': 1608,\n",
       " 'foreign': 993,\n",
       " 'mistaken': 1791,\n",
       " 'kimnovak': 1501,\n",
       " 'barbarabelgeddes': 236,\n",
       " 'acrophobia': 54,\n",
       " 'investigates': 1320,\n",
       " 'strange': 2556,\n",
       " 'activities': 57,\n",
       " 'san': 2334,\n",
       " 'francisco': 1006,\n",
       " 'detective': 695,\n",
       " 'suffering': 2582,\n",
       " 'dangerously': 618,\n",
       " 'obsessed': 1906,\n",
       " 'richardmarquand': 2240,\n",
       " 'endor': 851,\n",
       " 'trap': 2734,\n",
       " 'jabba': 1340,\n",
       " 'han': 1155,\n",
       " 'solo': 2486,\n",
       " 'hutt': 1252,\n",
       " 'back': 226,\n",
       " 'daring': 624,\n",
       " 'dispatch': 733,\n",
       " 'meanwhile': 1714,\n",
       " 'without': 2918,\n",
       " 'falling': 931,\n",
       " 'melgibson': 1727,\n",
       " 'jamesrobinson': 1363,\n",
       " 'seanlawlor': 2372,\n",
       " 'sandynelson': 2335,\n",
       " 'assaulting': 199,\n",
       " 'revolt': 2227,\n",
       " 'england': 856,\n",
       " 'sir': 2458,\n",
       " 'william': 2901,\n",
       " 'wallace': 2862,\n",
       " 'tried': 2745,\n",
       " 'executed': 900,\n",
       " 'edward': 810,\n",
       " 'english': 857,\n",
       " 'soldier': 2483,\n",
       " 'secret': 2380,\n",
       " 'bride': 343,\n",
       " 'harveykeitel': 1171,\n",
       " 'michaelmadsen': 1762,\n",
       " 'informant': 1290,\n",
       " 'suspect': 2601,\n",
       " 'simple': 2452,\n",
       " 'jewelry': 1398,\n",
       " 'heist': 1185,\n",
       " 'terribly': 2639,\n",
       " 'surviving': 2599,\n",
       " 'begin': 263,\n",
       " 'fredmacmurray': 1021,\n",
       " 'barbarastanwyck': 238,\n",
       " 'edwardg': 815,\n",
       " 'robinson': 2276,\n",
       " 'arouses': 183,\n",
       " 'talked': 2616,\n",
       " 'insurance': 1309,\n",
       " 'representative': 2205,\n",
       " 'lets': 1567,\n",
       " 'suspicions': 2603,\n",
       " 'investigator': 1322,\n",
       " 'fraud': 1018,\n",
       " 'scheme': 2353,\n",
       " 'darrenaronofsky': 626,\n",
       " 'jaredleto': 1375,\n",
       " 'jenniferconnelly': 1389,\n",
       " 'shattered': 2423,\n",
       " 'drug': 779,\n",
       " 'coney': 535,\n",
       " 'island': 1336,\n",
       " 'induced': 1280,\n",
       " 'utopias': 2812,\n",
       " 'addictions': 63,\n",
       " 'deep': 661,\n",
       " 'davidlean': 632,\n",
       " 'petero': 2012,\n",
       " 'toole': 2703,\n",
       " 'alecguinness': 99,\n",
       " 'anthonyquinn': 161,\n",
       " 'diverse': 740,\n",
       " 'turks': 2758,\n",
       " 'often': 1917,\n",
       " 'warring': 2871,\n",
       " 'lawrence': 1541,\n",
       " 'arab': 173,\n",
       " 'tribes': 2743,\n",
       " 'order': 1930,\n",
       " 'led': 1555,\n",
       " 'successfully': 2577,\n",
       " 'united': 2789,\n",
       " 'carlmiller': 394,\n",
       " 'ednapurviance': 808,\n",
       " 'jackiecoogan': 1343,\n",
       " 'abandoned': 33,\n",
       " 'jeopardy': 1391,\n",
       " 'put': 2129,\n",
       " 'cares': 392,\n",
       " 'malcolmmcdowell': 1646,\n",
       " 'patrickmagee': 1973,\n",
       " 'michaelbates': 1752,\n",
       " 'charismatic': 435,\n",
       " 'likes': 1577,\n",
       " 'volunteers': 2858,\n",
       " 'experimental': 912,\n",
       " ...}"
      ]
     },
     "execution_count": 34,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "count.vocabulary_"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 35,
   "metadata": {},
   "outputs": [
    {
     "data": {
      "text/plain": [
       "array([[1.        , 0.15789474, 0.13764944, ..., 0.05263158, 0.05263158,\n",
       "        0.05564149],\n",
       "       [0.15789474, 1.        , 0.36706517, ..., 0.05263158, 0.05263158,\n",
       "        0.05564149],\n",
       "       [0.13764944, 0.36706517, 1.        , ..., 0.04588315, 0.04588315,\n",
       "        0.04850713],\n",
       "       ...,\n",
       "       [0.05263158, 0.05263158, 0.04588315, ..., 1.        , 0.05263158,\n",
       "        0.05564149],\n",
       "       [0.05263158, 0.05263158, 0.04588315, ..., 0.05263158, 1.        ,\n",
       "        0.05564149],\n",
       "       [0.05564149, 0.05564149, 0.04850713, ..., 0.05564149, 0.05564149,\n",
       "        1.        ]])"
      ]
     },
     "execution_count": 35,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "# generating the cosine similarity matrix\n",
    "cosine_sim = cosine_similarity(count_matrix, count_matrix)\n",
    "cosine_sim"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 49,
   "metadata": {},
   "outputs": [],
   "source": [
    "# function that takes in movie title as input and returns the top 10 recommended movies\n",
    "def recommendations(title, cosine_sim = cosine_sim):\n",
    "    \n",
    "    recommended_movies = []\n",
    "    \n",
    "    # gettin the index of the movie that matches the title\n",
    "    idx = indices[indices == title].index[0]\n",
    "\n",
    "    # creating a Series with the similarity scores in descending order\n",
    "    score_series = pd.Series(cosine_sim[idx]).sort_values(ascending = False)\n",
    "\n",
    "    # getting the indexes of the 10 most similar movies\n",
    "    top_10_indexes = list(score_series.iloc[1:11].index)\n",
    "    print(top_10_indexes)\n",
    "    \n",
    "    # populating the list with the titles of the best 10 matching movies\n",
    "    for i in top_10_indexes:\n",
    "        recommended_movies.append(list(df.index)[i])\n",
    "        \n",
    "    return recommended_movies"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 50,
   "metadata": {},
   "outputs": [
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[132, 34, 226, 1, 61, 2, 100, 15, 214, 125]\n"
     ]
    },
    {
     "data": {
      "text/plain": [
       "['No Country for Old Men',\n",
       " 'The Departed',\n",
       " 'Rope',\n",
       " 'The Godfather',\n",
       " 'Reservoir Dogs',\n",
       " 'The Godfather: Part II',\n",
       " 'On the Waterfront',\n",
       " 'Goodfellas',\n",
       " 'Arsenic and Old Lace',\n",
       " 'The Big Lebowski']"
      ]
     },
     "execution_count": 50,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "recommendations('Fargo')"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 45,
   "metadata": {},
   "outputs": [
    {
     "data": {
      "text/plain": [
       "128"
      ]
     },
     "execution_count": 45,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "indices[indices == 'Fargo'].index[0]"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 48,
   "metadata": {},
   "outputs": [
    {
     "data": {
      "text/plain": [
       "0      0.132453\n",
       "1      0.220755\n",
       "2      0.192450\n",
       "3      0.105409\n",
       "4      0.140028\n",
       "5      0.039284\n",
       "6      0.040129\n",
       "7      0.123091\n",
       "8      0.037743\n",
       "9      0.039284\n",
       "10     0.043033\n",
       "11     0.000000\n",
       "12     0.000000\n",
       "13     0.038490\n",
       "14     0.044151\n",
       "15     0.178685\n",
       "16     0.000000\n",
       "17     0.000000\n",
       "18     0.132453\n",
       "19     0.043033\n",
       "20     0.164122\n",
       "21     0.153960\n",
       "22     0.150970\n",
       "23     0.041996\n",
       "24     0.039284\n",
       "25     0.046676\n",
       "26     0.136083\n",
       "27     0.044151\n",
       "28     0.045361\n",
       "29     0.120386\n",
       "         ...   \n",
       "220    0.044151\n",
       "221    0.109109\n",
       "222    0.040129\n",
       "223    0.000000\n",
       "224    0.038490\n",
       "225    0.000000\n",
       "226    0.222222\n",
       "227    0.082061\n",
       "228    0.045361\n",
       "229    0.037743\n",
       "230    0.000000\n",
       "231    0.051434\n",
       "232    0.099381\n",
       "233    0.040129\n",
       "234    0.096225\n",
       "235    0.048113\n",
       "236    0.044151\n",
       "237    0.083992\n",
       "238    0.000000\n",
       "239    0.099381\n",
       "240    0.000000\n",
       "241    0.000000\n",
       "242    0.000000\n",
       "243    0.048113\n",
       "244    0.041996\n",
       "245    0.048113\n",
       "246    0.041030\n",
       "247    0.044151\n",
       "248    0.044151\n",
       "249    0.046676\n",
       "Length: 250, dtype: float64"
      ]
     },
     "execution_count": 48,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "pd.Series(cosine_sim[indices[indices == 'Fargo'].index[0]])"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "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.6.5"
  }
 },
 "nbformat": 4,
 "nbformat_minor": 2
}