Q10.ipynb

{
 "cells": [
  {
   "attachments": {},
   "cell_type": "markdown",
   "metadata": {},
   "source": [
    "# Q10: Based on a prediction, what are the sweet spots of the existence of crude oil?\n",
    "    a. Predict the lithology of the sensor readings\n",
    "    b. pick the sandstone lithology with cross over in the permeability and porosity graphs and high resistivity (higher than 10 ohm.m)\n",
    "    b. visualize the sweet spots on a graph\n",
    "\n",
    "**Purpose of the question**: interpret the analysis results to the well log plot\n"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 1,
   "metadata": {},
   "outputs": [],
   "source": [
    "import matplotlib.backends.backend_pdf\n",
    "from chart_studio import plotly as py\n",
    "import plotly\n",
    "import matplotlib.pyplot as plt\n",
    "import numpy as np\n",
    "import math\n",
    "import lasio\n",
    "import pandas as pd\n",
    "from utils import *\n",
    "import warnings\n",
    "\n",
    "warnings.filterwarnings(\"ignore\")"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 2,
   "metadata": {},
   "outputs": [],
   "source": [
    "df = pd.read_csv(\"./Data/labeled_logs.csv\")\n",
    "well = df[df[\"WELL\"] == \"30/6-5\"]"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 3,
   "metadata": {},
   "outputs": [],
   "source": [
    "rename_columns(well)\n",
    "rename_lithology(well)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 4,
   "metadata": {},
   "outputs": [
    {
     "data": {
      "text/plain": [
       "Index(['WELL', 'Measured Depth', 'X_LOC', 'Y_LOC', 'Z_LOC', 'GROUP',\n",
       "       'FORMATION', 'Caliper', 'Shallow Resistivity',\n",
       "       'Medium Deep Resistivity', 'Deep Resistivity', 'Bulk Density',\n",
       "       'Gamma Ray', 'Spectra Gamma Ray', 'Neutron Porosity',\n",
       "       'Photo Electric Factor', 'Compressional waves sonic', 'Self Potential',\n",
       "       'Borehole Size', 'Rate of Penetration', 'Shear wave sonic',\n",
       "       'Differential Caliper', 'Density Correction', 'Weight of Drilling Mud',\n",
       "       'Micro Resistivity', 'Average Rate of Penetration',\n",
       "       'Flushed Zone Resistivity', 'LITHOLOGY', 'CONFIDENCE'],\n",
       "      dtype='object')"
      ]
     },
     "execution_count": 4,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "well.columns"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 5,
   "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>Measured Depth</th>\n",
       "      <th>X_LOC</th>\n",
       "      <th>Y_LOC</th>\n",
       "      <th>Z_LOC</th>\n",
       "      <th>Caliper</th>\n",
       "      <th>Shallow Resistivity</th>\n",
       "      <th>Medium Deep Resistivity</th>\n",
       "      <th>Deep Resistivity</th>\n",
       "      <th>Bulk Density</th>\n",
       "      <th>Gamma Ray</th>\n",
       "      <th>...</th>\n",
       "      <th>Borehole Size</th>\n",
       "      <th>Rate of Penetration</th>\n",
       "      <th>Shear wave sonic</th>\n",
       "      <th>Differential Caliper</th>\n",
       "      <th>Density Correction</th>\n",
       "      <th>Weight of Drilling Mud</th>\n",
       "      <th>Micro Resistivity</th>\n",
       "      <th>Average Rate of Penetration</th>\n",
       "      <th>Flushed Zone Resistivity</th>\n",
       "      <th>CONFIDENCE</th>\n",
       "    </tr>\n",
       "  </thead>\n",
       "  <tbody>\n",
       "    <tr>\n",
       "      <th>count</th>\n",
       "      <td>21636.000000</td>\n",
       "      <td>21636.000000</td>\n",
       "      <td>2.163600e+04</td>\n",
       "      <td>21636.000000</td>\n",
       "      <td>21636.000000</td>\n",
       "      <td>21543.000000</td>\n",
       "      <td>21636.000000</td>\n",
       "      <td>21636.000000</td>\n",
       "      <td>17396.000000</td>\n",
       "      <td>21636.000000</td>\n",
       "      <td>...</td>\n",
       "      <td>0.0</td>\n",
       "      <td>5428.000000</td>\n",
       "      <td>0.0</td>\n",
       "      <td>0.0</td>\n",
       "      <td>17396.000000</td>\n",
       "      <td>0.0</td>\n",
       "      <td>0.0</td>\n",
       "      <td>0.0</td>\n",
       "      <td>21543.000000</td>\n",
       "      <td>21634.000000</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>mean</th>\n",
       "      <td>1901.789128</td>\n",
       "      <td>497435.690403</td>\n",
       "      <td>6.728314e+06</td>\n",
       "      <td>-1876.742554</td>\n",
       "      <td>15.023619</td>\n",
       "      <td>4.406523</td>\n",
       "      <td>3.249195</td>\n",
       "      <td>2.770618</td>\n",
       "      <td>2.235966</td>\n",
       "      <td>60.485826</td>\n",
       "      <td>...</td>\n",
       "      <td>NaN</td>\n",
       "      <td>626.793661</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>-0.019565</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>4.406743</td>\n",
       "      <td>1.110798</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>std</th>\n",
       "      <td>950.076651</td>\n",
       "      <td>1.603602</td>\n",
       "      <td>1.772950e+00</td>\n",
       "      <td>950.036494</td>\n",
       "      <td>3.567134</td>\n",
       "      <td>15.309442</td>\n",
       "      <td>8.684753</td>\n",
       "      <td>3.984296</td>\n",
       "      <td>0.287440</td>\n",
       "      <td>28.672819</td>\n",
       "      <td>...</td>\n",
       "      <td>NaN</td>\n",
       "      <td>652.197972</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>0.162474</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>15.015536</td>\n",
       "      <td>0.400585</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>min</th>\n",
       "      <td>256.922400</td>\n",
       "      <td>497432.656250</td>\n",
       "      <td>6.728312e+06</td>\n",
       "      <td>-3522.748779</td>\n",
       "      <td>8.293892</td>\n",
       "      <td>0.144376</td>\n",
       "      <td>0.489001</td>\n",
       "      <td>0.408959</td>\n",
       "      <td>1.058234</td>\n",
       "      <td>13.831209</td>\n",
       "      <td>...</td>\n",
       "      <td>NaN</td>\n",
       "      <td>14.740163</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>-1.740271</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>0.151241</td>\n",
       "      <td>1.000000</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>25%</th>\n",
       "      <td>1079.052400</td>\n",
       "      <td>497434.187500</td>\n",
       "      <td>6.728313e+06</td>\n",
       "      <td>-2699.891785</td>\n",
       "      <td>13.062982</td>\n",
       "      <td>1.053047</td>\n",
       "      <td>1.005875</td>\n",
       "      <td>1.010638</td>\n",
       "      <td>1.975110</td>\n",
       "      <td>39.670696</td>\n",
       "      <td>...</td>\n",
       "      <td>NaN</td>\n",
       "      <td>148.326778</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>-0.010969</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>1.055680</td>\n",
       "      <td>1.000000</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>50%</th>\n",
       "      <td>1901.182400</td>\n",
       "      <td>497435.968750</td>\n",
       "      <td>6.728314e+06</td>\n",
       "      <td>-1876.156800</td>\n",
       "      <td>14.308678</td>\n",
       "      <td>1.827821</td>\n",
       "      <td>1.608696</td>\n",
       "      <td>1.549330</td>\n",
       "      <td>2.271871</td>\n",
       "      <td>54.092361</td>\n",
       "      <td>...</td>\n",
       "      <td>NaN</td>\n",
       "      <td>365.188965</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>-0.002281</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>1.822853</td>\n",
       "      <td>1.000000</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>75%</th>\n",
       "      <td>2724.984400</td>\n",
       "      <td>497437.218750</td>\n",
       "      <td>6.728316e+06</td>\n",
       "      <td>-1054.043274</td>\n",
       "      <td>18.167191</td>\n",
       "      <td>5.564113</td>\n",
       "      <td>4.066304</td>\n",
       "      <td>3.253166</td>\n",
       "      <td>2.489623</td>\n",
       "      <td>72.815086</td>\n",
       "      <td>...</td>\n",
       "      <td>NaN</td>\n",
       "      <td>943.401611</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>0.009558</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>5.616581</td>\n",
       "      <td>1.000000</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>max</th>\n",
       "      <td>3547.874400</td>\n",
       "      <td>497437.937500</td>\n",
       "      <td>6.728318e+06</td>\n",
       "      <td>-231.922302</td>\n",
       "      <td>24.139614</td>\n",
       "      <td>1122.207642</td>\n",
       "      <td>598.500000</td>\n",
       "      <td>128.508682</td>\n",
       "      <td>2.787034</td>\n",
       "      <td>236.665665</td>\n",
       "      <td>...</td>\n",
       "      <td>NaN</td>\n",
       "      <td>4837.952148</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>0.401935</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>1077.067993</td>\n",
       "      <td>3.000000</td>\n",
       "    </tr>\n",
       "  </tbody>\n",
       "</table>\n",
       "<p>8 rows × 25 columns</p>\n",
       "</div>"
      ],
      "text/plain": [
       "       Measured Depth          X_LOC         Y_LOC         Z_LOC  \\\n",
       "count    21636.000000   21636.000000  2.163600e+04  21636.000000   \n",
       "mean      1901.789128  497435.690403  6.728314e+06  -1876.742554   \n",
       "std        950.076651       1.603602  1.772950e+00    950.036494   \n",
       "min        256.922400  497432.656250  6.728312e+06  -3522.748779   \n",
       "25%       1079.052400  497434.187500  6.728313e+06  -2699.891785   \n",
       "50%       1901.182400  497435.968750  6.728314e+06  -1876.156800   \n",
       "75%       2724.984400  497437.218750  6.728316e+06  -1054.043274   \n",
       "max       3547.874400  497437.937500  6.728318e+06   -231.922302   \n",
       "\n",
       "            Caliper  Shallow Resistivity  Medium Deep Resistivity  \\\n",
       "count  21636.000000         21543.000000             21636.000000   \n",
       "mean      15.023619             4.406523                 3.249195   \n",
       "std        3.567134            15.309442                 8.684753   \n",
       "min        8.293892             0.144376                 0.489001   \n",
       "25%       13.062982             1.053047                 1.005875   \n",
       "50%       14.308678             1.827821                 1.608696   \n",
       "75%       18.167191             5.564113                 4.066304   \n",
       "max       24.139614          1122.207642               598.500000   \n",
       "\n",
       "       Deep Resistivity  Bulk Density     Gamma Ray  ...  Borehole Size  \\\n",
       "count      21636.000000  17396.000000  21636.000000  ...            0.0   \n",
       "mean           2.770618      2.235966     60.485826  ...            NaN   \n",
       "std            3.984296      0.287440     28.672819  ...            NaN   \n",
       "min            0.408959      1.058234     13.831209  ...            NaN   \n",
       "25%            1.010638      1.975110     39.670696  ...            NaN   \n",
       "50%            1.549330      2.271871     54.092361  ...            NaN   \n",
       "75%            3.253166      2.489623     72.815086  ...            NaN   \n",
       "max          128.508682      2.787034    236.665665  ...            NaN   \n",
       "\n",
       "       Rate of Penetration  Shear wave sonic  Differential Caliper  \\\n",
       "count          5428.000000               0.0                   0.0   \n",
       "mean            626.793661               NaN                   NaN   \n",
       "std             652.197972               NaN                   NaN   \n",
       "min              14.740163               NaN                   NaN   \n",
       "25%             148.326778               NaN                   NaN   \n",
       "50%             365.188965               NaN                   NaN   \n",
       "75%             943.401611               NaN                   NaN   \n",
       "max            4837.952148               NaN                   NaN   \n",
       "\n",
       "       Density Correction  Weight of Drilling Mud  Micro Resistivity  \\\n",
       "count        17396.000000                     0.0                0.0   \n",
       "mean            -0.019565                     NaN                NaN   \n",
       "std              0.162474                     NaN                NaN   \n",
       "min             -1.740271                     NaN                NaN   \n",
       "25%             -0.010969                     NaN                NaN   \n",
       "50%             -0.002281                     NaN                NaN   \n",
       "75%              0.009558                     NaN                NaN   \n",
       "max              0.401935                     NaN                NaN   \n",
       "\n",
       "       Average Rate of Penetration  Flushed Zone Resistivity    CONFIDENCE  \n",
       "count                          0.0              21543.000000  21634.000000  \n",
       "mean                           NaN                  4.406743      1.110798  \n",
       "std                            NaN                 15.015536      0.400585  \n",
       "min                            NaN                  0.151241      1.000000  \n",
       "25%                            NaN                  1.055680      1.000000  \n",
       "50%                            NaN                  1.822853      1.000000  \n",
       "75%                            NaN                  5.616581      1.000000  \n",
       "max                            NaN               1077.067993      3.000000  \n",
       "\n",
       "[8 rows x 25 columns]"
      ]
     },
     "execution_count": 5,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "well.describe()"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 6,
   "metadata": {},
   "outputs": [],
   "source": [
    "def is_crossover(row):\n",
    "    if math.isnan(row[\"Neutron Porosity2\"]) or math.isnan(row[\"Bulk Density2\"]):\n",
    "        return False\n",
    "    if row[\"Neutron Porosity2\"] + row[\"Bulk Density2\"] > 100:\n",
    "        return False\n",
    "    else:\n",
    "        return True"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 7,
   "metadata": {},
   "outputs": [],
   "source": [
    "def translate(value, leftMin, leftMax, rightMin, rightMax):\n",
    "    leftSpan = leftMax - leftMin\n",
    "    rightSpan = rightMax - rightMin\n",
    "\n",
    "    valueScaled = value.apply(lambda x: float(x - leftMin) / float(leftSpan))\n",
    "\n",
    "    return rightMin + (valueScaled * rightSpan)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 8,
   "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>WELL</th>\n",
       "      <th>Measured Depth</th>\n",
       "      <th>X_LOC</th>\n",
       "      <th>Y_LOC</th>\n",
       "      <th>Z_LOC</th>\n",
       "      <th>GROUP</th>\n",
       "      <th>FORMATION</th>\n",
       "      <th>Caliper</th>\n",
       "      <th>Shallow Resistivity</th>\n",
       "      <th>Medium Deep Resistivity</th>\n",
       "      <th>...</th>\n",
       "      <th>Rate of Penetration</th>\n",
       "      <th>Shear wave sonic</th>\n",
       "      <th>Differential Caliper</th>\n",
       "      <th>Density Correction</th>\n",
       "      <th>Weight of Drilling Mud</th>\n",
       "      <th>Micro Resistivity</th>\n",
       "      <th>Average Rate of Penetration</th>\n",
       "      <th>Flushed Zone Resistivity</th>\n",
       "      <th>LITHOLOGY</th>\n",
       "      <th>CONFIDENCE</th>\n",
       "    </tr>\n",
       "  </thead>\n",
       "  <tbody>\n",
       "    <tr>\n",
       "      <th>517758</th>\n",
       "      <td>30/6-5</td>\n",
       "      <td>256.9224</td>\n",
       "      <td>497437.375</td>\n",
       "      <td>6728312.0</td>\n",
       "      <td>-231.922302</td>\n",
       "      <td>NORDLAND GP.</td>\n",
       "      <td>NaN</td>\n",
       "      <td>21.149618</td>\n",
       "      <td>1.837429</td>\n",
       "      <td>1.539432</td>\n",
       "      <td>...</td>\n",
       "      <td>98.551178</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>1.847653</td>\n",
       "      <td>Shale</td>\n",
       "      <td>NaN</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>517759</th>\n",
       "      <td>30/6-5</td>\n",
       "      <td>257.0744</td>\n",
       "      <td>497437.375</td>\n",
       "      <td>6728312.0</td>\n",
       "      <td>-232.074310</td>\n",
       "      <td>NORDLAND GP.</td>\n",
       "      <td>NaN</td>\n",
       "      <td>20.023129</td>\n",
       "      <td>1.950167</td>\n",
       "      <td>1.608084</td>\n",
       "      <td>...</td>\n",
       "      <td>57.606297</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>2.041368</td>\n",
       "      <td>Shale</td>\n",
       "      <td>NaN</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>517760</th>\n",
       "      <td>30/6-5</td>\n",
       "      <td>257.2264</td>\n",
       "      <td>497437.375</td>\n",
       "      <td>6728312.0</td>\n",
       "      <td>-232.226303</td>\n",
       "      <td>NORDLAND GP.</td>\n",
       "      <td>NaN</td>\n",
       "      <td>17.702616</td>\n",
       "      <td>2.133782</td>\n",
       "      <td>1.667697</td>\n",
       "      <td>...</td>\n",
       "      <td>213.779495</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>2.213406</td>\n",
       "      <td>Shale</td>\n",
       "      <td>1.0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>517761</th>\n",
       "      <td>30/6-5</td>\n",
       "      <td>257.3784</td>\n",
       "      <td>497437.375</td>\n",
       "      <td>6728312.0</td>\n",
       "      <td>-232.378311</td>\n",
       "      <td>NORDLAND GP.</td>\n",
       "      <td>NaN</td>\n",
       "      <td>16.079689</td>\n",
       "      <td>2.224826</td>\n",
       "      <td>1.712445</td>\n",
       "      <td>...</td>\n",
       "      <td>190.913376</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>2.231647</td>\n",
       "      <td>Shale</td>\n",
       "      <td>1.0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>517762</th>\n",
       "      <td>30/6-5</td>\n",
       "      <td>257.5304</td>\n",
       "      <td>497437.375</td>\n",
       "      <td>6728312.0</td>\n",
       "      <td>-232.530304</td>\n",
       "      <td>NORDLAND GP.</td>\n",
       "      <td>NaN</td>\n",
       "      <td>14.906407</td>\n",
       "      <td>2.308409</td>\n",
       "      <td>1.786179</td>\n",
       "      <td>...</td>\n",
       "      <td>85.149597</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>2.377096</td>\n",
       "      <td>Shale</td>\n",
       "      <td>1.0</td>\n",
       "    </tr>\n",
       "  </tbody>\n",
       "</table>\n",
       "<p>5 rows × 29 columns</p>\n",
       "</div>"
      ],
      "text/plain": [
       "          WELL  Measured Depth       X_LOC      Y_LOC       Z_LOC  \\\n",
       "517758  30/6-5        256.9224  497437.375  6728312.0 -231.922302   \n",
       "517759  30/6-5        257.0744  497437.375  6728312.0 -232.074310   \n",
       "517760  30/6-5        257.2264  497437.375  6728312.0 -232.226303   \n",
       "517761  30/6-5        257.3784  497437.375  6728312.0 -232.378311   \n",
       "517762  30/6-5        257.5304  497437.375  6728312.0 -232.530304   \n",
       "\n",
       "               GROUP FORMATION    Caliper  Shallow Resistivity  \\\n",
       "517758  NORDLAND GP.       NaN  21.149618             1.837429   \n",
       "517759  NORDLAND GP.       NaN  20.023129             1.950167   \n",
       "517760  NORDLAND GP.       NaN  17.702616             2.133782   \n",
       "517761  NORDLAND GP.       NaN  16.079689             2.224826   \n",
       "517762  NORDLAND GP.       NaN  14.906407             2.308409   \n",
       "\n",
       "        Medium Deep Resistivity  ...  Rate of Penetration  Shear wave sonic  \\\n",
       "517758                 1.539432  ...            98.551178               NaN   \n",
       "517759                 1.608084  ...            57.606297               NaN   \n",
       "517760                 1.667697  ...           213.779495               NaN   \n",
       "517761                 1.712445  ...           190.913376               NaN   \n",
       "517762                 1.786179  ...            85.149597               NaN   \n",
       "\n",
       "        Differential Caliper  Density Correction  Weight of Drilling Mud  \\\n",
       "517758                   NaN                 NaN                     NaN   \n",
       "517759                   NaN                 NaN                     NaN   \n",
       "517760                   NaN                 NaN                     NaN   \n",
       "517761                   NaN                 NaN                     NaN   \n",
       "517762                   NaN                 NaN                     NaN   \n",
       "\n",
       "        Micro Resistivity  Average Rate of Penetration  \\\n",
       "517758                NaN                          NaN   \n",
       "517759                NaN                          NaN   \n",
       "517760                NaN                          NaN   \n",
       "517761                NaN                          NaN   \n",
       "517762                NaN                          NaN   \n",
       "\n",
       "        Flushed Zone Resistivity  LITHOLOGY  CONFIDENCE  \n",
       "517758                  1.847653      Shale         NaN  \n",
       "517759                  2.041368      Shale         NaN  \n",
       "517760                  2.213406      Shale         1.0  \n",
       "517761                  2.231647      Shale         1.0  \n",
       "517762                  2.377096      Shale         1.0  \n",
       "\n",
       "[5 rows x 29 columns]"
      ]
     },
     "execution_count": 8,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "well.head()"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 9,
   "metadata": {},
   "outputs": [],
   "source": [
    "def get_sandstone_position_tuples(log):\n",
    "    log[\"Neutron Porosity\"] = translate(\n",
    "        log[\"Neutron Porosity\"],\n",
    "        log[\"Neutron Porosity\"].min(),\n",
    "        log[\"Neutron Porosity\"].max(),\n",
    "        -0.15,\n",
    "        0.45,\n",
    "    )\n",
    "    log[\"Bulk Density\"] = translate(\n",
    "        log[\"Bulk Density\"],\n",
    "        log[\"Bulk Density\"].min(),\n",
    "        log[\"Bulk Density\"].max(),\n",
    "        1.9,\n",
    "        2.9,\n",
    "    )\n",
    "\n",
    "    log[\"Neutron Porosity2\"] = translate(log[\"Neutron Porosity\"], -0.15, 0.45, 0, 100)\n",
    "    log[\"Bulk Density2\"] = translate(log[\"Bulk Density\"], 1.9, 2.9, 0, 100)\n",
    "\n",
    "    log[\"Crossover\"] = log.apply(is_crossover, axis=1)\n",
    "\n",
    "    log[\"WATER\"] = log[\"Deep Resistivity\"].apply(lambda x: False if x > 10 else True)\n",
    "    sandstone_positions = log[\n",
    "        (log[\"LITHOLOGY\"] == \"Sandstone\")\n",
    "        & (log[\"WATER\"] == False)\n",
    "        & (log[\"Crossover\"] == True)\n",
    "    ][\"Measured Depth\"].tolist()\n",
    "\n",
    "    sandstone_positions = [int(i) for i in sandstone_positions]\n",
    "\n",
    "    sandstone_positions_tuples = []\n",
    "    for i in range(len(sandstone_positions)):\n",
    "        j = i + 1\n",
    "        while (\n",
    "            j < len(sandstone_positions)\n",
    "            and sandstone_positions[j] - sandstone_positions[j - 1] == 1\n",
    "        ):\n",
    "            j += 1\n",
    "        if j == len(sandstone_positions):\n",
    "            sandstone_positions_tuples.append(\n",
    "                (sandstone_positions[i], sandstone_positions[j - 1])\n",
    "            )\n",
    "            break\n",
    "        sandstone_positions_tuples.append(\n",
    "            (sandstone_positions[i], sandstone_positions[j - 1])\n",
    "        )\n",
    "        i = j\n",
    "\n",
    "    return sandstone_positions_tuples"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 10,
   "metadata": {},
   "outputs": [],
   "source": [
    "def add_graph(\n",
    "    ax,\n",
    "    x,\n",
    "    y,\n",
    "    label,\n",
    "    color,\n",
    "    xlim_min,\n",
    "    xlim_max,\n",
    "    spines,\n",
    "    spines_value,\n",
    "    minor_ticks=None,\n",
    "    major_ticks=None,\n",
    "    new_grid=False,\n",
    "    log_scale=False,\n",
    "):\n",
    "    ax_new = ax.twiny()\n",
    "    if log_scale:\n",
    "        ax_new.set_xscale(\"log\")\n",
    "    ax_new.set_xlim(xlim_min, xlim_max)\n",
    "    ax_new.set_xlabel(label, color=color, fontsize=12)\n",
    "    ax_new.plot(\n",
    "        x,\n",
    "        y,\n",
    "        spines,\n",
    "        color=color,\n",
    "        label=label,\n",
    "        linewidth=1,\n",
    "    )\n",
    "    ax_new.tick_params(axis=\"x\", colors=color)\n",
    "    ax_new.spines[\"top\"].set_position((\"outward\", spines_value))\n",
    "    if type(minor_ticks) != type(None) and type(major_ticks) != type(None):\n",
    "        ax_new.set_xticks(major_ticks)\n",
    "        ax_new.set_xticks(minor_ticks, minor=True)\n",
    "    if new_grid:\n",
    "        ax_new.grid(which=\"minor\", alpha=0.5)\n",
    "        ax_new.grid(which=\"major\", alpha=1)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 11,
   "metadata": {},
   "outputs": [],
   "source": [
    "def draw_first_track(ax, log):\n",
    "    ax[0].set_ylabel(\"Depth [m]\", color=\"Black\", fontsize=15)\n",
    "    add_graph(\n",
    "        ax[0],\n",
    "        log[\"Gamma Ray\"],\n",
    "        log[\"Measured Depth\"],\n",
    "        \"Gamma Ray [GAPI]\",\n",
    "        \"Green\",\n",
    "        0,\n",
    "        200,\n",
    "        \"-\",\n",
    "        50,\n",
    "        minor_ticks=np.arange(0, 201, 20),\n",
    "        major_ticks=np.arange(0, 201, 100),\n",
    "        new_grid=True,\n",
    "    )\n",
    "    add_graph(\n",
    "        ax[0],\n",
    "        log[\"Caliper\"],\n",
    "        log[\"Measured Depth\"],\n",
    "        \"Caliper [in]\",\n",
    "        \"Red\",\n",
    "        6,\n",
    "        16,\n",
    "        \"--\",\n",
    "        10,\n",
    "        minor_ticks=np.arange(6, 17, 1),\n",
    "        major_ticks=np.arange(6, 17, 5),\n",
    "    )\n",
    "    "
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 12,
   "metadata": {},
   "outputs": [],
   "source": [
    "def draw_second_track(ax, log):\n",
    "    add_graph(\n",
    "        ax[1],\n",
    "        log[\"Medium Deep Resistivity\"],\n",
    "        log[\"Measured Depth\"],\n",
    "        \"Medium Deep Resistivity [ohm.m]\",\n",
    "        \"Red\",\n",
    "        0.2,\n",
    "        200,\n",
    "        \"--\",\n",
    "        90,\n",
    "        log_scale=True,\n",
    "        new_grid=True,\n",
    "    )\n",
    "    add_graph(\n",
    "        ax[1],\n",
    "        log[\"Deep Resistivity\"],\n",
    "        log[\"Measured Depth\"],\n",
    "        \"Deep Resistivity [ohm.m]\",\n",
    "        \"Black\",\n",
    "        0.2,\n",
    "        200,\n",
    "        \"-\",\n",
    "        10,\n",
    "        log_scale=True,\n",
    "    )\n",
    "    add_graph(\n",
    "        ax[1],\n",
    "        log[\"Shallow Resistivity\"],\n",
    "        log[\"Measured Depth\"],\n",
    "        \"Shallow Resistivity [ohm.m]\",\n",
    "        \"Blue\",\n",
    "        0.2,\n",
    "        200,\n",
    "        \"--\",\n",
    "        50,\n",
    "        log_scale=True,\n",
    "    )"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 13,
   "metadata": {},
   "outputs": [],
   "source": [
    "def draw_third_track(ax, log):\n",
    "    add_graph(\n",
    "        ax[2],\n",
    "        log[\"Neutron Porosity\"],\n",
    "        log[\"Measured Depth\"],\n",
    "        \"Neutron Porosity [decp]\",\n",
    "        \"Blue\",\n",
    "        0.45,\n",
    "        -0.15,\n",
    "        \"--\",\n",
    "        50,\n",
    "        minor_ticks=np.arange(-0.15, 0.45, 0.06),\n",
    "        major_ticks=np.arange(-0.15, 0.45, 0.3),\n",
    "        new_grid=True,\n",
    "    )\n",
    "\n",
    "    add_graph(\n",
    "        ax[2],\n",
    "        log[\"Bulk Density\"],\n",
    "        log[\"Measured Depth\"],\n",
    "        \"Bulk Density [g/cc]\",\n",
    "        \"Red\",\n",
    "        1.9,\n",
    "        2.9,\n",
    "        \"-\",\n",
    "        10,\n",
    "        minor_ticks=np.arange(1.9, 2.9, 0.1),\n",
    "        major_ticks=np.arange(1.9, 2.9, 0.5),\n",
    "    )\n",
    "    "
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 14,
   "metadata": {},
   "outputs": [],
   "source": [
    "def highlight_reservoir(ax, sandstone_positions_tuples, top, bottom):\n",
    "    for start, end in sandstone_positions_tuples:\n",
    "        s = start if start > top else top\n",
    "        e = end if end < bottom else bottom\n",
    "        ax[0].axhspan(s, e, color=\"yellow\", alpha=0.5)\n",
    "        ax[1].axhspan(s, e, color=\"yellow\", alpha=0.5)\n",
    "        ax[2].axhspan(s, e, color=\"yellow\", alpha=0.5)\n",
    "        if e == bottom:\n",
    "            break"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 15,
   "metadata": {},
   "outputs": [],
   "source": [
    "def plot_3_tracks(log):\n",
    "    log = log.copy()\n",
    "    sandstone_positions_tuples = get_sandstone_position_tuples(log)\n",
    "    pdf = matplotlib.backends.backend_pdf.PdfPages(\"wireline_log.pdf\")\n",
    "    o_top = log.iloc[0][\"Measured Depth\"]\n",
    "    o_bottom = log.iloc[-1][\"Measured Depth\"]\n",
    "\n",
    "    # for every 100 readings, plot the log in a new page\n",
    "    i = o_top\n",
    "    while i < o_bottom:\n",
    "        top = i\n",
    "        bottom = i + 100\n",
    "        i = i + 100\n",
    "        fig, ax = plt.subplots(nrows=1, ncols=3, figsize=(20, 25), sharey=True)\n",
    "        fig.suptitle(\"Wireline Log\", fontsize=20)\n",
    "        fig.subplots_adjust(top=0.88)\n",
    "\n",
    "        for axes in ax:\n",
    "            axes.set_ylim(bottom, top)\n",
    "            depth_major_ticks = np.arange(top, bottom, 50)\n",
    "            depth_minor_ticks = np.arange(top, bottom, 10)\n",
    "            axes.set_yticks(depth_major_ticks)\n",
    "            axes.set_yticks(depth_minor_ticks, minor=True)\n",
    "            axes.get_xaxis().set_visible(False)\n",
    "            axes.grid(which=\"minor\", axis=\"y\", alpha=0.5)\n",
    "            axes.grid(which=\"major\", axis=\"y\", alpha=1)\n",
    "\n",
    "        draw_first_track(ax, log)\n",
    "        draw_second_track(ax, log)\n",
    "        draw_third_track(ax, log)\n",
    "\n",
    "        highlight_reservoir(ax, sandstone_positions_tuples, top, bottom)\n",
    "\n",
    "        pdf.savefig(fig)\n",
    "        plt.close()\n",
    "    pdf.close()\n"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 16,
   "metadata": {},
   "outputs": [],
   "source": [
    "plot_3_tracks(well)\n"
   ]
  }
 ],
 "metadata": {
  "kernelspec": {
   "display_name": "base",
   "language": "python",
   "name": "python3"
  },
  "language_info": {
   "codemirror_mode": {
    "name": "ipython",
    "version": 3
   },
   "file_extension": ".py",
   "mimetype": "text/x-python",
   "name": "python",
   "nbconvert_exporter": "python",
   "pygments_lexer": "ipython3",
   "version": "3.10.9"
  },
  "orig_nbformat": 4
 },
 "nbformat": 4,
 "nbformat_minor": 2
}