imdb_web_scraping (1).ipynb

{
  "nbformat": 4,
  "nbformat_minor": 0,
  "metadata": {
    "colab": {
      "provenance": []
    },
    "kernelspec": {
      "name": "python3",
      "display_name": "Python 3"
    },
    "language_info": {
      "name": "python"
    }
  },
  "cells": [
    {
      "cell_type": "code",
      "source": [
        "#importing the libraries needed\n",
        "import requests\n",
        "from bs4 import BeautifulSoup\n",
        "import pandas as pd\n",
        "import numpy as np\n",
        "import csv"
      ],
      "metadata": {
        "id": "7Q2W1lNlBKvJ"
      },
      "execution_count": null,
      "outputs": []
    },
    {
      "cell_type": "code",
      "source": [
        "#creating empty lists to store movie data\n",
        "rating=[]\n",
        "year=[]\n",
        "gross=[]\n",
        "title=[]\n",
        "genre =[]\n",
        "run_time=[]\n",
        "user_rating=[]\n",
        "\n",
        "pages = np.arange(1,1000,100)\n",
        "\n",
        "#for loop to to extract movies details from IMDB\n",
        "for page in pages:\n",
        "    page = requests.get(\"https://www.imdb.com/search/title/?groups=top_250&sort=user_rating,desc\"+str(page)+\"&ref_=adv_nxt\")\n",
        "    soup = BeautifulSoup(page.text, 'html.parser')\n",
        "    results=soup.find_all(\"div\",class_='lister-item mode-advanced')\n",
        "\n",
        "    for res in results:\n",
        "        rating.append(res.find('span',class_='certificate').text.strip())\n",
        "        year.append(res.find('span',class_='lister-item-year text-muted unbold').text.strip('()'))\n",
        "        gross.append(res.find('p', class_='sort-num_votes-visible').text.strip().replace('\\n',''))\n",
        "        title.append(res.find(\"h3\",class_=\"lister-item-header\").text.strip().replace('\\n',''))\n",
        "        genre.append(res.find(\"span\",class_='genre').text.strip())\n",
        "        run_time.append(res.find(\"span\",class_=\"runtime\").text.strip())\n",
        "        user_rating.append(res.find(\"div\", class_=\"inline-block ratings-imdb-rating\").text.strip())"
      ],
      "metadata": {
        "id": "1sCaiBhrK6WA"
      },
      "execution_count": null,
      "outputs": []
    },
    {
      "cell_type": "code",
      "source": [
        "#Creating dataframe\n",
        "df1 = pd.DataFrame({'rating':rating,'year':year,'gross':gross,'title':title,'run_time':run_time,'genre':genre,'user_rating':user_rating})"
      ],
      "metadata": {
        "id": "735HV5gF1mb6"
      },
      "execution_count": null,
      "outputs": []
    },
    {
      "cell_type": "code",
      "source": [
        "df1"
      ],
      "metadata": {
        "id": "wVk8JA64QCxD"
      },
      "execution_count": null,
      "outputs": []
    },
    {
      "cell_type": "code",
      "source": [
        "#Separating gross into three columns(movie_gross,number of votes & imdb ranking)\n",
        "new = df1['gross'].str.split('|', expand= True)\n",
        "num_votes = new[0].str.split(':', expand= True)\n",
        "df1[\"num_votes\"]=num_votes[1]\n",
        "imdb_ranking = new[2].str.split(':', expand= True)\n",
        "df1[\"imdb_ranking\"]=imdb_ranking[1]\n",
        "movie_gross = new[1].str.split(':', expand= True)\n",
        "df1[\"movie_gross\"]=movie_gross[1]"
      ],
      "metadata": {
        "id": "WgdUdEUocsVH"
      },
      "execution_count": null,
      "outputs": []
    },
    {
      "cell_type": "code",
      "source": [
        "#Remove year & number from title column\n",
        "new1 = df1['title'].str.split('(', expand= True)\n",
        "movie_title = new1[0].str.split('.', expand= True)\n",
        "df1['movie_title']=movie_title[1]"
      ],
      "metadata": {
        "id": "dHR40AiCnLpe"
      },
      "execution_count": null,
      "outputs": []
    },
    {
      "cell_type": "code",
      "source": [
        "#Removing title & gross column\n",
        "df1.drop(columns=['title', 'gross'],axis=1, inplace = True)"
      ],
      "metadata": {
        "id": "aASQaN6_T01F"
      },
      "execution_count": null,
      "outputs": []
    },
    {
      "cell_type": "code",
      "source": [
        "#changing columns order\n",
        "\n",
        "df1.iloc[:, [8,1,3,0,7,2,4,5,6]]"
      ],
      "metadata": {
        "colab": {
          "base_uri": "https://localhost:8080/",
          "height": 423
        },
        "id": "DFQbnQJNtf08",
        "outputId": "1d6d6658-b311-4e57-aa61-f0dc33dd8cd3"
      },
      "execution_count": null,
      "outputs": [
        {
          "output_type": "execute_result",
          "data": {
            "text/plain": [
              "                      movie_title  year                      genre     rating  \\\n",
              "0        The Shawshank Redemption  1994                      Drama          R   \n",
              "1                   The Godfather  1972               Crime, Drama          R   \n",
              "2                Schindler's List  1993  Biography, Drama, History          R   \n",
              "3                 The Dark Knight  2008       Action, Crime, Drama      PG-13   \n",
              "4                    12 Angry Men  1957               Crime, Drama   Approved   \n",
              "..                            ...   ...                        ...        ...   \n",
              "495              The Intouchables  2011   Biography, Comedy, Drama          R   \n",
              "496                    Casablanca  1942        Drama, Romance, War         PG   \n",
              "497  Once Upon a Time in the West  1968                    Western      PG-13   \n",
              "498                        Psycho  1960  Horror, Mystery, Thriller          R   \n",
              "499        Grave of the Fireflies  1988      Animation, Drama, War  Not Rated   \n",
              "\n",
              "    movie_gross run_time user_rating  num_votes imdb_ranking  \n",
              "0       $28.34M  142 min         9.3  2,695,712           #1  \n",
              "1      $134.97M  175 min         9.2  1,870,793           #2  \n",
              "2       $96.90M  195 min         9.0  1,363,323           #6  \n",
              "3      $534.86M  152 min         9.0  2,669,777           #3  \n",
              "4        $4.36M   96 min         9.0    796,488           #5  \n",
              "..          ...      ...         ...        ...          ...  \n",
              "495     $13.18M  112 min         8.5    865,666          #46  \n",
              "496      $1.02M  102 min         8.5    575,780          #43  \n",
              "497      $5.32M  165 min         8.5    332,823          #48  \n",
              "498     $32.00M  109 min         8.5    676,778          #33  \n",
              "499         #44   89 min         8.5    281,016         None  \n",
              "\n",
              "[500 rows x 9 columns]"
            ],
            "text/html": [
              "\n",
              "  <div id=\"df-8c2e380a-0499-4ebb-a27d-2cf524fc94d1\">\n",
              "    <div class=\"colab-df-container\">\n",
              "      <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>movie_title</th>\n",
              "      <th>year</th>\n",
              "      <th>genre</th>\n",
              "      <th>rating</th>\n",
              "      <th>movie_gross</th>\n",
              "      <th>run_time</th>\n",
              "      <th>user_rating</th>\n",
              "      <th>num_votes</th>\n",
              "      <th>imdb_ranking</th>\n",
              "    </tr>\n",
              "  </thead>\n",
              "  <tbody>\n",
              "    <tr>\n",
              "      <th>0</th>\n",
              "      <td>The Shawshank Redemption</td>\n",
              "      <td>1994</td>\n",
              "      <td>Drama</td>\n",
              "      <td>R</td>\n",
              "      <td>$28.34M</td>\n",
              "      <td>142 min</td>\n",
              "      <td>9.3</td>\n",
              "      <td>2,695,712</td>\n",
              "      <td>#1</td>\n",
              "    </tr>\n",
              "    <tr>\n",
              "      <th>1</th>\n",
              "      <td>The Godfather</td>\n",
              "      <td>1972</td>\n",
              "      <td>Crime, Drama</td>\n",
              "      <td>R</td>\n",
              "      <td>$134.97M</td>\n",
              "      <td>175 min</td>\n",
              "      <td>9.2</td>\n",
              "      <td>1,870,793</td>\n",
              "      <td>#2</td>\n",
              "    </tr>\n",
              "    <tr>\n",
              "      <th>2</th>\n",
              "      <td>Schindler's List</td>\n",
              "      <td>1993</td>\n",
              "      <td>Biography, Drama, History</td>\n",
              "      <td>R</td>\n",
              "      <td>$96.90M</td>\n",
              "      <td>195 min</td>\n",
              "      <td>9.0</td>\n",
              "      <td>1,363,323</td>\n",
              "      <td>#6</td>\n",
              "    </tr>\n",
              "    <tr>\n",
              "      <th>3</th>\n",
              "      <td>The Dark Knight</td>\n",
              "      <td>2008</td>\n",
              "      <td>Action, Crime, Drama</td>\n",
              "      <td>PG-13</td>\n",
              "      <td>$534.86M</td>\n",
              "      <td>152 min</td>\n",
              "      <td>9.0</td>\n",
              "      <td>2,669,777</td>\n",
              "      <td>#3</td>\n",
              "    </tr>\n",
              "    <tr>\n",
              "      <th>4</th>\n",
              "      <td>12 Angry Men</td>\n",
              "      <td>1957</td>\n",
              "      <td>Crime, Drama</td>\n",
              "      <td>Approved</td>\n",
              "      <td>$4.36M</td>\n",
              "      <td>96 min</td>\n",
              "      <td>9.0</td>\n",
              "      <td>796,488</td>\n",
              "      <td>#5</td>\n",
              "    </tr>\n",
              "    <tr>\n",
              "      <th>...</th>\n",
              "      <td>...</td>\n",
              "      <td>...</td>\n",
              "      <td>...</td>\n",
              "      <td>...</td>\n",
              "      <td>...</td>\n",
              "      <td>...</td>\n",
              "      <td>...</td>\n",
              "      <td>...</td>\n",
              "      <td>...</td>\n",
              "    </tr>\n",
              "    <tr>\n",
              "      <th>495</th>\n",
              "      <td>The Intouchables</td>\n",
              "      <td>2011</td>\n",
              "      <td>Biography, Comedy, Drama</td>\n",
              "      <td>R</td>\n",
              "      <td>$13.18M</td>\n",
              "      <td>112 min</td>\n",
              "      <td>8.5</td>\n",
              "      <td>865,666</td>\n",
              "      <td>#46</td>\n",
              "    </tr>\n",
              "    <tr>\n",
              "      <th>496</th>\n",
              "      <td>Casablanca</td>\n",
              "      <td>1942</td>\n",
              "      <td>Drama, Romance, War</td>\n",
              "      <td>PG</td>\n",
              "      <td>$1.02M</td>\n",
              "      <td>102 min</td>\n",
              "      <td>8.5</td>\n",
              "      <td>575,780</td>\n",
              "      <td>#43</td>\n",
              "    </tr>\n",
              "    <tr>\n",
              "      <th>497</th>\n",
              "      <td>Once Upon a Time in the West</td>\n",
              "      <td>1968</td>\n",
              "      <td>Western</td>\n",
              "      <td>PG-13</td>\n",
              "      <td>$5.32M</td>\n",
              "      <td>165 min</td>\n",
              "      <td>8.5</td>\n",
              "      <td>332,823</td>\n",
              "      <td>#48</td>\n",
              "    </tr>\n",
              "    <tr>\n",
              "      <th>498</th>\n",
              "      <td>Psycho</td>\n",
              "      <td>1960</td>\n",
              "      <td>Horror, Mystery, Thriller</td>\n",
              "      <td>R</td>\n",
              "      <td>$32.00M</td>\n",
              "      <td>109 min</td>\n",
              "      <td>8.5</td>\n",
              "      <td>676,778</td>\n",
              "      <td>#33</td>\n",
              "    </tr>\n",
              "    <tr>\n",
              "      <th>499</th>\n",
              "      <td>Grave of the Fireflies</td>\n",
              "      <td>1988</td>\n",
              "      <td>Animation, Drama, War</td>\n",
              "      <td>Not Rated</td>\n",
              "      <td>#44</td>\n",
              "      <td>89 min</td>\n",
              "      <td>8.5</td>\n",
              "      <td>281,016</td>\n",
              "      <td>None</td>\n",
              "    </tr>\n",
              "  </tbody>\n",
              "</table>\n",
              "<p>500 rows × 9 columns</p>\n",
              "</div>\n",
              "      <button class=\"colab-df-convert\" onclick=\"convertToInteractive('df-8c2e380a-0499-4ebb-a27d-2cf524fc94d1')\"\n",
              "              title=\"Convert this dataframe to an interactive table.\"\n",
              "              style=\"display:none;\">\n",
              "        \n",
              "  <svg xmlns=\"http://www.w3.org/2000/svg\" height=\"24px\"viewBox=\"0 0 24 24\"\n",
              "       width=\"24px\">\n",
              "    <path d=\"M0 0h24v24H0V0z\" fill=\"none\"/>\n",
              "    <path d=\"M18.56 5.44l.94 2.06.94-2.06 2.06-.94-2.06-.94-.94-2.06-.94 2.06-2.06.94zm-11 1L8.5 8.5l.94-2.06 2.06-.94-2.06-.94L8.5 2.5l-.94 2.06-2.06.94zm10 10l.94 2.06.94-2.06 2.06-.94-2.06-.94-.94-2.06-.94 2.06-2.06.94z\"/><path d=\"M17.41 7.96l-1.37-1.37c-.4-.4-.92-.59-1.43-.59-.52 0-1.04.2-1.43.59L10.3 9.45l-7.72 7.72c-.78.78-.78 2.05 0 2.83L4 21.41c.39.39.9.59 1.41.59.51 0 1.02-.2 1.41-.59l7.78-7.78 2.81-2.81c.8-.78.8-2.07 0-2.86zM5.41 20L4 18.59l7.72-7.72 1.47 1.35L5.41 20z\"/>\n",
              "  </svg>\n",
              "      </button>\n",
              "      \n",
              "  <style>\n",
              "    .colab-df-container {\n",
              "      display:flex;\n",
              "      flex-wrap:wrap;\n",
              "      gap: 12px;\n",
              "    }\n",
              "\n",
              "    .colab-df-convert {\n",
              "      background-color: #E8F0FE;\n",
              "      border: none;\n",
              "      border-radius: 50%;\n",
              "      cursor: pointer;\n",
              "      display: none;\n",
              "      fill: #1967D2;\n",
              "      height: 32px;\n",
              "      padding: 0 0 0 0;\n",
              "      width: 32px;\n",
              "    }\n",
              "\n",
              "    .colab-df-convert:hover {\n",
              "      background-color: #E2EBFA;\n",
              "      box-shadow: 0px 1px 2px rgba(60, 64, 67, 0.3), 0px 1px 3px 1px rgba(60, 64, 67, 0.15);\n",
              "      fill: #174EA6;\n",
              "    }\n",
              "\n",
              "    [theme=dark] .colab-df-convert {\n",
              "      background-color: #3B4455;\n",
              "      fill: #D2E3FC;\n",
              "    }\n",
              "\n",
              "    [theme=dark] .colab-df-convert:hover {\n",
              "      background-color: #434B5C;\n",
              "      box-shadow: 0px 1px 3px 1px rgba(0, 0, 0, 0.15);\n",
              "      filter: drop-shadow(0px 1px 2px rgba(0, 0, 0, 0.3));\n",
              "      fill: #FFFFFF;\n",
              "    }\n",
              "  </style>\n",
              "\n",
              "      <script>\n",
              "        const buttonEl =\n",
              "          document.querySelector('#df-8c2e380a-0499-4ebb-a27d-2cf524fc94d1 button.colab-df-convert');\n",
              "        buttonEl.style.display =\n",
              "          google.colab.kernel.accessAllowed ? 'block' : 'none';\n",
              "\n",
              "        async function convertToInteractive(key) {\n",
              "          const element = document.querySelector('#df-8c2e380a-0499-4ebb-a27d-2cf524fc94d1');\n",
              "          const dataTable =\n",
              "            await google.colab.kernel.invokeFunction('convertToInteractive',\n",
              "                                                     [key], {});\n",
              "          if (!dataTable) return;\n",
              "\n",
              "          const docLinkHtml = 'Like what you see? Visit the ' +\n",
              "            '<a target=\"_blank\" href=https://colab.research.google.com/notebooks/data_table.ipynb>data table notebook</a>'\n",
              "            + ' to learn more about interactive tables.';\n",
              "          element.innerHTML = '';\n",
              "          dataTable['output_type'] = 'display_data';\n",
              "          await google.colab.output.renderOutput(dataTable, element);\n",
              "          const docLink = document.createElement('div');\n",
              "          docLink.innerHTML = docLinkHtml;\n",
              "          element.appendChild(docLink);\n",
              "        }\n",
              "      </script>\n",
              "    </div>\n",
              "  </div>\n",
              "  "
            ]
          },
          "metadata": {},
          "execution_count": 9
        }
      ]
    },
    {
      "cell_type": "code",
      "source": [
        "#saving the data in csv format\n",
        "\n",
        "from google.colab import files\n",
        "df1.to_csv('imdb_movies.csv', index = False, encoding = 'utf-8-sig')\n",
        "files.download('imdb_movies.csv')"
      ],
      "metadata": {
        "colab": {
          "base_uri": "https://localhost:8080/",
          "height": 17
        },
        "id": "PvL9lUI1hbJo",
        "outputId": "22608578-0876-4380-956b-e0c4f3320e36"
      },
      "execution_count": null,
      "outputs": [
        {
          "output_type": "display_data",
          "data": {
            "text/plain": [
              "<IPython.core.display.Javascript object>"
            ],
            "application/javascript": [
              "\n",
              "    async function download(id, filename, size) {\n",
              "      if (!google.colab.kernel.accessAllowed) {\n",
              "        return;\n",
              "      }\n",
              "      const div = document.createElement('div');\n",
              "      const label = document.createElement('label');\n",
              "      label.textContent = `Downloading \"${filename}\": `;\n",
              "      div.appendChild(label);\n",
              "      const progress = document.createElement('progress');\n",
              "      progress.max = size;\n",
              "      div.appendChild(progress);\n",
              "      document.body.appendChild(div);\n",
              "\n",
              "      const buffers = [];\n",
              "      let downloaded = 0;\n",
              "\n",
              "      const channel = await google.colab.kernel.comms.open(id);\n",
              "      // Send a message to notify the kernel that we're ready.\n",
              "      channel.send({})\n",
              "\n",
              "      for await (const message of channel.messages) {\n",
              "        // Send a message to notify the kernel that we're ready.\n",
              "        channel.send({})\n",
              "        if (message.buffers) {\n",
              "          for (const buffer of message.buffers) {\n",
              "            buffers.push(buffer);\n",
              "            downloaded += buffer.byteLength;\n",
              "            progress.value = downloaded;\n",
              "          }\n",
              "        }\n",
              "      }\n",
              "      const blob = new Blob(buffers, {type: 'application/binary'});\n",
              "      const a = document.createElement('a');\n",
              "      a.href = window.URL.createObjectURL(blob);\n",
              "      a.download = filename;\n",
              "      div.appendChild(a);\n",
              "      a.click();\n",
              "      div.remove();\n",
              "    }\n",
              "  "
            ]
          },
          "metadata": {}
        },
        {
          "output_type": "display_data",
          "data": {
            "text/plain": [
              "<IPython.core.display.Javascript object>"
            ],
            "application/javascript": [
              "download(\"download_6da34aad-d363-4b61-b598-965d74575f01\", \"imdb_movies.csv\", 41629)"
            ]
          },
          "metadata": {}
        }
      ]
    }
  ]
}