Skip to content

Instantly share code, notes, and snippets.

@salouri

salouri/DA0101EN-Review-Data-Wrangling.ipynb

Created November 12, 2019 12:45
Show Gist options
  • Save salouri/834648884dbbce885c7ca035e32750ec to your computer and use it in GitHub Desktop.
Save salouri/834648884dbbce885c7ca035e32750ec to your computer and use it in GitHub Desktop.
Download ZIP
Created on Cognitive Class Labs
Display the source blob
Display the rendered blob
Raw
DA0101EN-Review-Data-Wrangling.ipynb
{
"cells": [
{
"cell_type": "markdown",
"metadata": {},
"source": [
"<div class=\"alert alert-block alert-info\" style=\"margin-top: 20px\">\n",
" <a href=\"http://cocl.us/DA0101EN_NotbookLink_Top\">\n",
" <img src=\"https://s3-api.us-geo.objectstorage.softlayer.net/cf-courses-data/CognitiveClass/DA0101EN/Images/TopAd.png\" width=\"750\" align=\"center\">\n",
" </a>\n",
"</div>"
]
},
{
"cell_type": "markdown",
"metadata": {},
"source": [
"<a href=\"https://www.bigdatauniversity.com\"><img src = \"https://s3-api.us-geo.objectstorage.softlayer.net/cf-courses-data/CognitiveClass/DA0101EN/Images/CCLog.png\" width = 300, align = \"center\"></a>\n",
"\n",
"<h1 align=center><font size=5>Data Analysis with Python</font></h1>"
]
},
{
"cell_type": "markdown",
"metadata": {},
"source": [
"<h1>Data Wrangling</h1>"
]
},
{
"cell_type": "markdown",
"metadata": {},
"source": [
"<h3>Welcome!</h3>\n",
"\n",
"By the end of this notebook, you will have learned the basics of Data Wrangling! "
]
},
{
"cell_type": "markdown",
"metadata": {},
"source": [
"<h2>Table of content</h2>\n",
"\n",
"<div class=\"alert alert-block alert-info\" style=\"margin-top: 20px\">\n",
"<ul>\n",
" <li><a href=\"#identify_handle_missing_values\">Identify and handle missing values</a>\n",
" <ul>\n",
" <li><a href=\"#identify_missing_values\">Identify missing values</a></li>\n",
" <li><a href=\"#deal_missing_values\">Deal with missing values</a></li>\n",
" <li><a href=\"#correct_data_format\">Correct data format</a></li>\n",
" </ul>\n",
" </li>\n",
" <li><a href=\"#data_standardization\">Data standardization</a></li>\n",
" <li><a href=\"#data_normalization\">Data Normalization (centering/scaling)</a></li>\n",
" <li><a href=\"#binning\">Binning</a></li>\n",
" <li><a href=\"#indicator\">Indicator variable</a></li>\n",
"</ul>\n",
" \n",
"Estimated Time Needed: <strong>30 min</strong>\n",
"</div>\n",
" \n",
"<hr>"
]
},
{
"cell_type": "markdown",
"metadata": {},
"source": [
"<h2>What is the purpose of Data Wrangling?</h2>"
]
},
{
"cell_type": "markdown",
"metadata": {},
"source": [
"Data Wrangling is the process of converting data from the initial format to a format that may be better for analysis."
]
},
{
"cell_type": "markdown",
"metadata": {},
"source": [
"<h3>What is the fuel consumption (L/100k) rate for the diesel car?</h3>"
]
},
{
"cell_type": "markdown",
"metadata": {},
"source": [
"<h3>Import data</h3>\n",
"<p>\n",
"You can find the \"Automobile Data Set\" from the following link: <a href=\"https://archive.ics.uci.edu/ml/machine-learning-databases/autos/imports-85.data\">https://archive.ics.uci.edu/ml/machine-learning-databases/autos/imports-85.data</a>. \n",
"We will be using this data set throughout this course.\n",
"</p>"
]
},
{
"cell_type": "markdown",
"metadata": {},
"source": [
"<h4>Import pandas</h4> "
]
},
{
"cell_type": "code",
"execution_count": 5,
"metadata": {},
"outputs": [],
"source": [
"import pandas as pd\n",
"import matplotlib.pylab as plt"
]
},
{
"cell_type": "markdown",
"metadata": {},
"source": [
"<h2>Reading the data set from the URL and adding the related headers.</h2>"
]
},
{
"cell_type": "markdown",
"metadata": {},
"source": [
"URL of the dataset"
]
},
{
"cell_type": "markdown",
"metadata": {},
"source": [
"This dataset was hosted on IBM Cloud object click <a href=\"https://cocl.us/cognitive_class_DA0101EN_objectstorage\">HERE</a> for free storage "
]
},
{
"cell_type": "code",
"execution_count": 6,
"metadata": {},
"outputs": [],
"source": [
"filename = \"https://s3-api.us-geo.objectstorage.softlayer.net/cf-courses-data/CognitiveClass/DA0101EN/auto.csv\""
]
},
{
"cell_type": "markdown",
"metadata": {},
"source": [
" Python list <b>headers</b> containing name of headers "
]
},
{
"cell_type": "code",
"execution_count": 7,
"metadata": {},
"outputs": [],
"source": [
"headers = [\"symboling\",\"normalized-losses\",\"make\",\"fuel-type\",\"aspiration\", \"num-of-doors\",\"body-style\",\n",
" \"drive-wheels\",\"engine-location\",\"wheel-base\", \"length\",\"width\",\"height\",\"curb-weight\",\"engine-type\",\n",
" \"num-of-cylinders\", \"engine-size\",\"fuel-system\",\"bore\",\"stroke\",\"compression-ratio\",\"horsepower\",\n",
" \"peak-rpm\",\"city-mpg\",\"highway-mpg\",\"price\"]"
]
},
{
"cell_type": "markdown",
"metadata": {},
"source": [
"Use the Pandas method <b>read_csv()</b> to load the data from the web address. Set the parameter \"names\" equal to the Python list \"headers\"."
]
},
{
"cell_type": "code",
"execution_count": 95,
"metadata": {
"collapsed": false,
"jupyter": {
"outputs_hidden": false
}
},
"outputs": [],
"source": [
"df = pd.read_csv(filename, names = headers)"
]
},
{
"cell_type": "markdown",
"metadata": {},
"source": [
" Use the method <b>head()</b> to display the first five rows of the dataframe. "
]
},
{
"cell_type": "code",
"execution_count": 9,
"metadata": {
"collapsed": false,
"jupyter": {
"outputs_hidden": false
}
},
"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>symboling</th>\n",
" <th>normalized-losses</th>\n",
" <th>make</th>\n",
" <th>fuel-type</th>\n",
" <th>aspiration</th>\n",
" <th>num-of-doors</th>\n",
" <th>body-style</th>\n",
" <th>drive-wheels</th>\n",
" <th>engine-location</th>\n",
" <th>wheel-base</th>\n",
" <th>...</th>\n",
" <th>engine-size</th>\n",
" <th>fuel-system</th>\n",
" <th>bore</th>\n",
" <th>stroke</th>\n",
" <th>compression-ratio</th>\n",
" <th>horsepower</th>\n",
" <th>peak-rpm</th>\n",
" <th>city-mpg</th>\n",
" <th>highway-mpg</th>\n",
" <th>price</th>\n",
" </tr>\n",
" </thead>\n",
" <tbody>\n",
" <tr>\n",
" <th>0</th>\n",
" <td>3</td>\n",
" <td>?</td>\n",
" <td>alfa-romero</td>\n",
" <td>gas</td>\n",
" <td>std</td>\n",
" <td>two</td>\n",
" <td>convertible</td>\n",
" <td>rwd</td>\n",
" <td>front</td>\n",
" <td>88.6</td>\n",
" <td>...</td>\n",
" <td>130</td>\n",
" <td>mpfi</td>\n",
" <td>3.47</td>\n",
" <td>2.68</td>\n",
" <td>9.0</td>\n",
" <td>111</td>\n",
" <td>5000</td>\n",
" <td>21</td>\n",
" <td>27</td>\n",
" <td>13495</td>\n",
" </tr>\n",
" <tr>\n",
" <th>1</th>\n",
" <td>3</td>\n",
" <td>?</td>\n",
" <td>alfa-romero</td>\n",
" <td>gas</td>\n",
" <td>std</td>\n",
" <td>two</td>\n",
" <td>convertible</td>\n",
" <td>rwd</td>\n",
" <td>front</td>\n",
" <td>88.6</td>\n",
" <td>...</td>\n",
" <td>130</td>\n",
" <td>mpfi</td>\n",
" <td>3.47</td>\n",
" <td>2.68</td>\n",
" <td>9.0</td>\n",
" <td>111</td>\n",
" <td>5000</td>\n",
" <td>21</td>\n",
" <td>27</td>\n",
" <td>16500</td>\n",
" </tr>\n",
" <tr>\n",
" <th>2</th>\n",
" <td>1</td>\n",
" <td>?</td>\n",
" <td>alfa-romero</td>\n",
" <td>gas</td>\n",
" <td>std</td>\n",
" <td>two</td>\n",
" <td>hatchback</td>\n",
" <td>rwd</td>\n",
" <td>front</td>\n",
" <td>94.5</td>\n",
" <td>...</td>\n",
" <td>152</td>\n",
" <td>mpfi</td>\n",
" <td>2.68</td>\n",
" <td>3.47</td>\n",
" <td>9.0</td>\n",
" <td>154</td>\n",
" <td>5000</td>\n",
" <td>19</td>\n",
" <td>26</td>\n",
" <td>16500</td>\n",
" </tr>\n",
" <tr>\n",
" <th>3</th>\n",
" <td>2</td>\n",
" <td>164</td>\n",
" <td>audi</td>\n",
" <td>gas</td>\n",
" <td>std</td>\n",
" <td>four</td>\n",
" <td>sedan</td>\n",
" <td>fwd</td>\n",
" <td>front</td>\n",
" <td>99.8</td>\n",
" <td>...</td>\n",
" <td>109</td>\n",
" <td>mpfi</td>\n",
" <td>3.19</td>\n",
" <td>3.40</td>\n",
" <td>10.0</td>\n",
" <td>102</td>\n",
" <td>5500</td>\n",
" <td>24</td>\n",
" <td>30</td>\n",
" <td>13950</td>\n",
" </tr>\n",
" <tr>\n",
" <th>4</th>\n",
" <td>2</td>\n",
" <td>164</td>\n",
" <td>audi</td>\n",
" <td>gas</td>\n",
" <td>std</td>\n",
" <td>four</td>\n",
" <td>sedan</td>\n",
" <td>4wd</td>\n",
" <td>front</td>\n",
" <td>99.4</td>\n",
" <td>...</td>\n",
" <td>136</td>\n",
" <td>mpfi</td>\n",
" <td>3.19</td>\n",
" <td>3.40</td>\n",
" <td>8.0</td>\n",
" <td>115</td>\n",
" <td>5500</td>\n",
" <td>18</td>\n",
" <td>22</td>\n",
" <td>17450</td>\n",
" </tr>\n",
" </tbody>\n",
"</table>\n",
"<p>5 rows × 26 columns</p>\n",
"</div>"
],
"text/plain": [
" symboling normalized-losses make fuel-type aspiration num-of-doors \\\n",
"0 3 ? alfa-romero gas std two \n",
"1 3 ? alfa-romero gas std two \n",
"2 1 ? alfa-romero gas std two \n",
"3 2 164 audi gas std four \n",
"4 2 164 audi gas std four \n",
"\n",
" body-style drive-wheels engine-location wheel-base ... engine-size \\\n",
"0 convertible rwd front 88.6 ... 130 \n",
"1 convertible rwd front 88.6 ... 130 \n",
"2 hatchback rwd front 94.5 ... 152 \n",
"3 sedan fwd front 99.8 ... 109 \n",
"4 sedan 4wd front 99.4 ... 136 \n",
"\n",
" fuel-system bore stroke compression-ratio horsepower peak-rpm city-mpg \\\n",
"0 mpfi 3.47 2.68 9.0 111 5000 21 \n",
"1 mpfi 3.47 2.68 9.0 111 5000 21 \n",
"2 mpfi 2.68 3.47 9.0 154 5000 19 \n",
"3 mpfi 3.19 3.40 10.0 102 5500 24 \n",
"4 mpfi 3.19 3.40 8.0 115 5500 18 \n",
"\n",
" highway-mpg price \n",
"0 27 13495 \n",
"1 27 16500 \n",
"2 26 16500 \n",
"3 30 13950 \n",
"4 22 17450 \n",
"\n",
"[5 rows x 26 columns]"
]
},
"execution_count": 9,
"metadata": {},
"output_type": "execute_result"
}
],
"source": [
"# To see what the data set looks like, we'll use the head() method.\n",
"df.head()"
]
},
{
"cell_type": "markdown",
"metadata": {},
"source": [
"As we can see, several question marks appeared in the dataframe; those are missing values which may hinder our further analysis. \n",
"<div>So, how do we identify all those missing values and deal with them?</div> \n",
"\n",
"\n",
"<b>How to work with missing data?</b>\n",
"\n",
"Steps for working with missing data:\n",
"<ol>\n",
" <li>dentify missing data</li>\n",
" <li>deal with missing data</li>\n",
" <li>correct data format</li>\n",
"</ol>"
]
},
{
"cell_type": "markdown",
"metadata": {},
"source": [
"<h2 id=\"identify_handle_missing_values\">Identify and handle missing values</h2>\n",
"\n",
"\n",
"<h3 id=\"identify_missing_values\">Identify missing values</h3>\n",
"<h4>Convert \"?\" to NaN</h4>\n",
"In the car dataset, missing data comes with the question mark \"?\".\n",
"We replace \"?\" with NaN (Not a Number), which is Python's default missing value marker, for reasons of computational speed and convenience. Here we use the function: \n",
" <pre>.replace(A, B, inplace = True) </pre>\n",
"to replace A by B"
]
},
{
"cell_type": "code",
"execution_count": 6,
"metadata": {
"collapsed": false,
"jupyter": {
"outputs_hidden": false
}
},
"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>symboling</th>\n",
" <th>normalized-losses</th>\n",
" <th>make</th>\n",
" <th>fuel-type</th>\n",
" <th>aspiration</th>\n",
" <th>num-of-doors</th>\n",
" <th>body-style</th>\n",
" <th>drive-wheels</th>\n",
" <th>engine-location</th>\n",
" <th>wheel-base</th>\n",
" <th>...</th>\n",
" <th>engine-size</th>\n",
" <th>fuel-system</th>\n",
" <th>bore</th>\n",
" <th>stroke</th>\n",
" <th>compression-ratio</th>\n",
" <th>horsepower</th>\n",
" <th>peak-rpm</th>\n",
" <th>city-mpg</th>\n",
" <th>highway-mpg</th>\n",
" <th>price</th>\n",
" </tr>\n",
" </thead>\n",
" <tbody>\n",
" <tr>\n",
" <th>0</th>\n",
" <td>3</td>\n",
" <td>NaN</td>\n",
" <td>alfa-romero</td>\n",
" <td>gas</td>\n",
" <td>std</td>\n",
" <td>two</td>\n",
" <td>convertible</td>\n",
" <td>rwd</td>\n",
" <td>front</td>\n",
" <td>88.6</td>\n",
" <td>...</td>\n",
" <td>130</td>\n",
" <td>mpfi</td>\n",
" <td>3.47</td>\n",
" <td>2.68</td>\n",
" <td>9.0</td>\n",
" <td>111</td>\n",
" <td>5000</td>\n",
" <td>21</td>\n",
" <td>27</td>\n",
" <td>13495</td>\n",
" </tr>\n",
" <tr>\n",
" <th>1</th>\n",
" <td>3</td>\n",
" <td>NaN</td>\n",
" <td>alfa-romero</td>\n",
" <td>gas</td>\n",
" <td>std</td>\n",
" <td>two</td>\n",
" <td>convertible</td>\n",
" <td>rwd</td>\n",
" <td>front</td>\n",
" <td>88.6</td>\n",
" <td>...</td>\n",
" <td>130</td>\n",
" <td>mpfi</td>\n",
" <td>3.47</td>\n",
" <td>2.68</td>\n",
" <td>9.0</td>\n",
" <td>111</td>\n",
" <td>5000</td>\n",
" <td>21</td>\n",
" <td>27</td>\n",
" <td>16500</td>\n",
" </tr>\n",
" <tr>\n",
" <th>2</th>\n",
" <td>1</td>\n",
" <td>NaN</td>\n",
" <td>alfa-romero</td>\n",
" <td>gas</td>\n",
" <td>std</td>\n",
" <td>two</td>\n",
" <td>hatchback</td>\n",
" <td>rwd</td>\n",
" <td>front</td>\n",
" <td>94.5</td>\n",
" <td>...</td>\n",
" <td>152</td>\n",
" <td>mpfi</td>\n",
" <td>2.68</td>\n",
" <td>3.47</td>\n",
" <td>9.0</td>\n",
" <td>154</td>\n",
" <td>5000</td>\n",
" <td>19</td>\n",
" <td>26</td>\n",
" <td>16500</td>\n",
" </tr>\n",
" <tr>\n",
" <th>3</th>\n",
" <td>2</td>\n",
" <td>164</td>\n",
" <td>audi</td>\n",
" <td>gas</td>\n",
" <td>std</td>\n",
" <td>four</td>\n",
" <td>sedan</td>\n",
" <td>fwd</td>\n",
" <td>front</td>\n",
" <td>99.8</td>\n",
" <td>...</td>\n",
" <td>109</td>\n",
" <td>mpfi</td>\n",
" <td>3.19</td>\n",
" <td>3.40</td>\n",
" <td>10.0</td>\n",
" <td>102</td>\n",
" <td>5500</td>\n",
" <td>24</td>\n",
" <td>30</td>\n",
" <td>13950</td>\n",
" </tr>\n",
" <tr>\n",
" <th>4</th>\n",
" <td>2</td>\n",
" <td>164</td>\n",
" <td>audi</td>\n",
" <td>gas</td>\n",
" <td>std</td>\n",
" <td>four</td>\n",
" <td>sedan</td>\n",
" <td>4wd</td>\n",
" <td>front</td>\n",
" <td>99.4</td>\n",
" <td>...</td>\n",
" <td>136</td>\n",
" <td>mpfi</td>\n",
" <td>3.19</td>\n",
" <td>3.40</td>\n",
" <td>8.0</td>\n",
" <td>115</td>\n",
" <td>5500</td>\n",
" <td>18</td>\n",
" <td>22</td>\n",
" <td>17450</td>\n",
" </tr>\n",
" </tbody>\n",
"</table>\n",
"<p>5 rows × 26 columns</p>\n",
"</div>"
],
"text/plain": [
" symboling normalized-losses make fuel-type aspiration num-of-doors \\\n",
"0 3 NaN alfa-romero gas std two \n",
"1 3 NaN alfa-romero gas std two \n",
"2 1 NaN alfa-romero gas std two \n",
"3 2 164 audi gas std four \n",
"4 2 164 audi gas std four \n",
"\n",
" body-style drive-wheels engine-location wheel-base ... engine-size \\\n",
"0 convertible rwd front 88.6 ... 130 \n",
"1 convertible rwd front 88.6 ... 130 \n",
"2 hatchback rwd front 94.5 ... 152 \n",
"3 sedan fwd front 99.8 ... 109 \n",
"4 sedan 4wd front 99.4 ... 136 \n",
"\n",
" fuel-system bore stroke compression-ratio horsepower peak-rpm city-mpg \\\n",
"0 mpfi 3.47 2.68 9.0 111 5000 21 \n",
"1 mpfi 3.47 2.68 9.0 111 5000 21 \n",
"2 mpfi 2.68 3.47 9.0 154 5000 19 \n",
"3 mpfi 3.19 3.40 10.0 102 5500 24 \n",
"4 mpfi 3.19 3.40 8.0 115 5500 18 \n",
"\n",
" highway-mpg price \n",
"0 27 13495 \n",
"1 27 16500 \n",
"2 26 16500 \n",
"3 30 13950 \n",
"4 22 17450 \n",
"\n",
"[5 rows x 26 columns]"
]
},
"execution_count": 6,
"metadata": {},
"output_type": "execute_result"
}
],
"source": [
"import numpy as np\n",
"\n",
"# replace \"?\" to NaN\n",
"df.replace(\"?\", np.nan, inplace = True)\n",
"df.head(5)"
]
},
{
"cell_type": "markdown",
"metadata": {},
"source": [
"dentify_missing_values\n",
"\n",
"<h4>Evaluating for Missing Data</h4>\n",
"\n",
"The missing values are converted to Python's default. We use Python's built-in functions to identify these missing values. There are two methods to detect missing data:\n",
"<ol>\n",
" <li><b>.isnull()</b></li>\n",
" <li><b>.notnull()</b></li>\n",
"</ol>\n",
"The output is a boolean value indicating whether the value that is passed into the argument is in fact missing data."
]
},
{
"cell_type": "code",
"execution_count": 7,
"metadata": {
"collapsed": false,
"jupyter": {
"outputs_hidden": false
}
},
"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>symboling</th>\n",
" <th>normalized-losses</th>\n",
" <th>make</th>\n",
" <th>fuel-type</th>\n",
" <th>aspiration</th>\n",
" <th>num-of-doors</th>\n",
" <th>body-style</th>\n",
" <th>drive-wheels</th>\n",
" <th>engine-location</th>\n",
" <th>wheel-base</th>\n",
" <th>...</th>\n",
" <th>engine-size</th>\n",
" <th>fuel-system</th>\n",
" <th>bore</th>\n",
" <th>stroke</th>\n",
" <th>compression-ratio</th>\n",
" <th>horsepower</th>\n",
" <th>peak-rpm</th>\n",
" <th>city-mpg</th>\n",
" <th>highway-mpg</th>\n",
" <th>price</th>\n",
" </tr>\n",
" </thead>\n",
" <tbody>\n",
" <tr>\n",
" <th>0</th>\n",
" <td>False</td>\n",
" <td>True</td>\n",
" <td>False</td>\n",
" <td>False</td>\n",
" <td>False</td>\n",
" <td>False</td>\n",
" <td>False</td>\n",
" <td>False</td>\n",
" <td>False</td>\n",
" <td>False</td>\n",
" <td>...</td>\n",
" <td>False</td>\n",
" <td>False</td>\n",
" <td>False</td>\n",
" <td>False</td>\n",
" <td>False</td>\n",
" <td>False</td>\n",
" <td>False</td>\n",
" <td>False</td>\n",
" <td>False</td>\n",
" <td>False</td>\n",
" </tr>\n",
" <tr>\n",
" <th>1</th>\n",
" <td>False</td>\n",
" <td>True</td>\n",
" <td>False</td>\n",
" <td>False</td>\n",
" <td>False</td>\n",
" <td>False</td>\n",
" <td>False</td>\n",
" <td>False</td>\n",
" <td>False</td>\n",
" <td>False</td>\n",
" <td>...</td>\n",
" <td>False</td>\n",
" <td>False</td>\n",
" <td>False</td>\n",
" <td>False</td>\n",
" <td>False</td>\n",
" <td>False</td>\n",
" <td>False</td>\n",
" <td>False</td>\n",
" <td>False</td>\n",
" <td>False</td>\n",
" </tr>\n",
" <tr>\n",
" <th>2</th>\n",
" <td>False</td>\n",
" <td>True</td>\n",
" <td>False</td>\n",
" <td>False</td>\n",
" <td>False</td>\n",
" <td>False</td>\n",
" <td>False</td>\n",
" <td>False</td>\n",
" <td>False</td>\n",
" <td>False</td>\n",
" <td>...</td>\n",
" <td>False</td>\n",
" <td>False</td>\n",
" <td>False</td>\n",
" <td>False</td>\n",
" <td>False</td>\n",
" <td>False</td>\n",
" <td>False</td>\n",
" <td>False</td>\n",
" <td>False</td>\n",
" <td>False</td>\n",
" </tr>\n",
" <tr>\n",
" <th>3</th>\n",
" <td>False</td>\n",
" <td>False</td>\n",
" <td>False</td>\n",
" <td>False</td>\n",
" <td>False</td>\n",
" <td>False</td>\n",
" <td>False</td>\n",
" <td>False</td>\n",
" <td>False</td>\n",
" <td>False</td>\n",
" <td>...</td>\n",
" <td>False</td>\n",
" <td>False</td>\n",
" <td>False</td>\n",
" <td>False</td>\n",
" <td>False</td>\n",
" <td>False</td>\n",
" <td>False</td>\n",
" <td>False</td>\n",
" <td>False</td>\n",
" <td>False</td>\n",
" </tr>\n",
" <tr>\n",
" <th>4</th>\n",
" <td>False</td>\n",
" <td>False</td>\n",
" <td>False</td>\n",
" <td>False</td>\n",
" <td>False</td>\n",
" <td>False</td>\n",
" <td>False</td>\n",
" <td>False</td>\n",
" <td>False</td>\n",
" <td>False</td>\n",
" <td>...</td>\n",
" <td>False</td>\n",
" <td>False</td>\n",
" <td>False</td>\n",
" <td>False</td>\n",
" <td>False</td>\n",
" <td>False</td>\n",
" <td>False</td>\n",
" <td>False</td>\n",
" <td>False</td>\n",
" <td>False</td>\n",
" </tr>\n",
" </tbody>\n",
"</table>\n",
"<p>5 rows × 26 columns</p>\n",
"</div>"
],
"text/plain": [
" symboling normalized-losses make fuel-type aspiration num-of-doors \\\n",
"0 False True False False False False \n",
"1 False True False False False False \n",
"2 False True False False False False \n",
"3 False False False False False False \n",
"4 False False False False False False \n",
"\n",
" body-style drive-wheels engine-location wheel-base ... engine-size \\\n",
"0 False False False False ... False \n",
"1 False False False False ... False \n",
"2 False False False False ... False \n",
"3 False False False False ... False \n",
"4 False False False False ... False \n",
"\n",
" fuel-system bore stroke compression-ratio horsepower peak-rpm \\\n",
"0 False False False False False False \n",
"1 False False False False False False \n",
"2 False False False False False False \n",
"3 False False False False False False \n",
"4 False False False False False False \n",
"\n",
" city-mpg highway-mpg price \n",
"0 False False False \n",
"1 False False False \n",
"2 False False False \n",
"3 False False False \n",
"4 False False False \n",
"\n",
"[5 rows x 26 columns]"
]
},
"execution_count": 7,
"metadata": {},
"output_type": "execute_result"
}
],
"source": [
"missing_data = df.isnull()\n",
"missing_data.head(5)"
]
},
{
"cell_type": "markdown",
"metadata": {},
"source": [
"\"True\" stands for missing value, while \"False\" stands for not missing value."
]
},
{
"cell_type": "markdown",
"metadata": {},
"source": [
"<h4>Count missing values in each column</h4>\n",
"<p>\n",
"Using a for loop in Python, we can quickly figure out the number of missing values in each column. As mentioned above, \"True\" represents a missing value, \"False\" means the value is present in the dataset. In the body of the for loop the method \".value_counts()\" counts the number of \"True\" values. \n",
"</p>"
]
},
{
"cell_type": "code",
"execution_count": 9,
"metadata": {
"collapsed": false,
"jupyter": {
"outputs_hidden": false
}
},
"outputs": [
{
"name": "stdout",
"output_type": "stream",
"text": [
"['symboling', 'normalized-losses', 'make', 'fuel-type', 'aspiration', 'num-of-doors', 'body-style', 'drive-wheels', 'engine-location', 'wheel-base', 'length', 'width', 'height', 'curb-weight', 'engine-type', 'num-of-cylinders', 'engine-size', 'fuel-system', 'bore', 'stroke', 'compression-ratio', 'horsepower', 'peak-rpm', 'city-mpg', 'highway-mpg', 'price']\n",
"symboling\n",
"False 205\n",
"Name: symboling, dtype: int64\n",
"\n",
"normalized-losses\n",
"False 164\n",
"True 41\n",
"Name: normalized-losses, dtype: int64\n",
"\n",
"make\n",
"False 205\n",
"Name: make, dtype: int64\n",
"\n",
"fuel-type\n",
"False 205\n",
"Name: fuel-type, dtype: int64\n",
"\n",
"aspiration\n",
"False 205\n",
"Name: aspiration, dtype: int64\n",
"\n",
"num-of-doors\n",
"False 203\n",
"True 2\n",
"Name: num-of-doors, dtype: int64\n",
"\n",
"body-style\n",
"False 205\n",
"Name: body-style, dtype: int64\n",
"\n",
"drive-wheels\n",
"False 205\n",
"Name: drive-wheels, dtype: int64\n",
"\n",
"engine-location\n",
"False 205\n",
"Name: engine-location, dtype: int64\n",
"\n",
"wheel-base\n",
"False 205\n",
"Name: wheel-base, dtype: int64\n",
"\n",
"length\n",
"False 205\n",
"Name: length, dtype: int64\n",
"\n",
"width\n",
"False 205\n",
"Name: width, dtype: int64\n",
"\n",
"height\n",
"False 205\n",
"Name: height, dtype: int64\n",
"\n",
"curb-weight\n",
"False 205\n",
"Name: curb-weight, dtype: int64\n",
"\n",
"engine-type\n",
"False 205\n",
"Name: engine-type, dtype: int64\n",
"\n",
"num-of-cylinders\n",
"False 205\n",
"Name: num-of-cylinders, dtype: int64\n",
"\n",
"engine-size\n",
"False 205\n",
"Name: engine-size, dtype: int64\n",
"\n",
"fuel-system\n",
"False 205\n",
"Name: fuel-system, dtype: int64\n",
"\n",
"bore\n",
"False 201\n",
"True 4\n",
"Name: bore, dtype: int64\n",
"\n",
"stroke\n",
"False 201\n",
"True 4\n",
"Name: stroke, dtype: int64\n",
"\n",
"compression-ratio\n",
"False 205\n",
"Name: compression-ratio, dtype: int64\n",
"\n",
"horsepower\n",
"False 203\n",
"True 2\n",
"Name: horsepower, dtype: int64\n",
"\n",
"peak-rpm\n",
"False 203\n",
"True 2\n",
"Name: peak-rpm, dtype: int64\n",
"\n",
"city-mpg\n",
"False 205\n",
"Name: city-mpg, dtype: int64\n",
"\n",
"highway-mpg\n",
"False 205\n",
"Name: highway-mpg, dtype: int64\n",
"\n",
"price\n",
"False 201\n",
"True 4\n",
"Name: price, dtype: int64\n",
"\n"
]
}
],
"source": [
"x = missing_data.columns.values.tolist()\n",
"print(x)\n",
"for column in x:\n",
" print(column)\n",
" print (missing_data[column].value_counts())\n",
" print(\"\") "
]
},
{
"cell_type": "markdown",
"metadata": {},
"source": [
"Based on the summary above, each column has 205 rows of data, seven columns containing missing data:\n",
"<ol>\n",
" <li>\"normalized-losses\": 41 missing data</li>\n",
" <li>\"num-of-doors\": 2 missing data</li>\n",
" <li>\"bore\": 4 missing data</li>\n",
" <li>\"stroke\" : 4 missing data</li>\n",
" <li>\"horsepower\": 2 missing data</li>\n",
" <li>\"peak-rpm\": 2 missing data</li>\n",
" <li>\"price\": 4 missing data</li>\n",
"</ol>"
]
},
{
"cell_type": "markdown",
"metadata": {},
"source": [
"<h3 id=\"deal_missing_values\">Deal with missing data</h3>\n",
"<b>How to deal with missing data?</b>\n",
"\n",
"<ol>\n",
" <li>drop data<br>\n",
" a. drop the whole row<br>\n",
" b. drop the whole column\n",
" </li>\n",
" <li>replace data<br>\n",
" a. replace it by mean<br>\n",
" b. replace it by frequency<br>\n",
" c. replace it based on other functions\n",
" </li>\n",
"</ol>"
]
},
{
"cell_type": "markdown",
"metadata": {},
"source": [
"Whole columns should be dropped only if most entries in the column are empty. In our dataset, none of the columns are empty enough to drop entirely.\n",
"We have some freedom in choosing which method to replace data; however, some methods may seem more reasonable than others. We will apply each method to many different columns:\n",
"\n",
"<b>Replace by mean:</b>\n",
"<ul>\n",
" <li>\"normalized-losses\": 41 missing data, replace them with mean</li>\n",
" <li>\"stroke\": 4 missing data, replace them with mean</li>\n",
" <li>\"bore\": 4 missing data, replace them with mean</li>\n",
" <li>\"horsepower\": 2 missing data, replace them with mean</li>\n",
" <li>\"peak-rpm\": 2 missing data, replace them with mean</li>\n",
"</ul>\n",
"\n",
"<b>Replace by frequency:</b>\n",
"<ul>\n",
" <li>\"num-of-doors\": 2 missing data, replace them with \"four\". \n",
" <ul>\n",
" <li>Reason: 84% sedans is four doors. Since four doors is most frequent, it is most likely to occur</li>\n",
" </ul>\n",
" </li>\n",
"</ul>\n",
"\n",
"<b>Drop the whole row:</b>\n",
"<ul>\n",
" <li>\"price\": 4 missing data, simply delete the whole row\n",
" <ul>\n",
" <li>Reason: price is what we want to predict. Any data entry without price data cannot be used for prediction; therefore any row now without price data is not useful to us</li>\n",
" </ul>\n",
" </li>\n",
"</ul>"
]
},
{
"cell_type": "markdown",
"metadata": {},
"source": [
"<h4>Calculate the average of the column </h4>"
]
},
{
"cell_type": "code",
"execution_count": 10,
"metadata": {
"collapsed": false,
"jupyter": {
"outputs_hidden": false
}
},
"outputs": [
{
"name": "stdout",
"output_type": "stream",
"text": [
"Average of normalized-losses: 122.0\n"
]
}
],
"source": [
"avg_norm_loss = df[\"normalized-losses\"].astype(\"float\").mean(axis=0)\n",
"print(\"Average of normalized-losses:\", avg_norm_loss)"
]
},
{
"cell_type": "markdown",
"metadata": {},
"source": [
"<h4>Replace \"NaN\" by mean value in \"normalized-losses\" column</h4>"
]
},
{
"cell_type": "code",
"execution_count": 11,
"metadata": {},
"outputs": [],
"source": [
"df[\"normalized-losses\"].replace(np.nan, avg_norm_loss, inplace=True)"
]
},
{
"cell_type": "markdown",
"metadata": {},
"source": [
"<h4>Calculate the mean value for 'bore' column</h4>"
]
},
{
"cell_type": "code",
"execution_count": 12,
"metadata": {},
"outputs": [
{
"name": "stdout",
"output_type": "stream",
"text": [
"Average of bore: 3.3297512437810943\n"
]
}
],
"source": [
"avg_bore=df['bore'].astype('float').mean(axis=0)\n",
"print(\"Average of bore:\", avg_bore)"
]
},
{
"cell_type": "markdown",
"metadata": {},
"source": [
"<h4>Replace NaN by mean value</h4>"
]
},
{
"cell_type": "code",
"execution_count": 13,
"metadata": {},
"outputs": [],
"source": [
"df[\"bore\"].replace(np.nan, avg_bore, inplace=True)"
]
},
{
"cell_type": "markdown",
"metadata": {},
"source": [
"<div class=\"alert alert-danger alertdanger\" style=\"margin-top: 20px\">\n",
"<h1> Question #1: </h1>\n",
"\n",
"<b>According to the example above, replace NaN in \"stroke\" column by mean.</b>\n",
"</div>"
]
},
{
"cell_type": "code",
"execution_count": 17,
"metadata": {
"collapsed": false,
"jupyter": {
"outputs_hidden": false
}
},
"outputs": [
{
"name": "stdout",
"output_type": "stream",
"text": [
"0 2.68\n",
"1 2.68\n",
"2 3.47\n",
"3 3.40\n",
"4 3.40\n",
" ... \n",
"200 3.15\n",
"201 3.15\n",
"202 2.87\n",
"203 3.40\n",
"204 3.15\n",
"Name: stroke, Length: 205, dtype: object\n"
]
}
],
"source": [
"# Write your code below and press Shift+Enter to execute \n",
"import numpy as np\n",
"print(df[\"stroke\"])\n",
"df[\"stroke\"].replace(\"?\",np.nan, inplace = True )\n",
"stroke_avg = df[\"stroke\"].astype(\"float\").mean(axis=0)\n",
"df[\"stroke\"].replace(np.nan,stroke_avg, inplace = True )"
]
},
{
"cell_type": "markdown",
"metadata": {},
"source": [
"Double-click <b>here</b> for the solution.\n",
"\n",
"<!-- The answer is below:\n",
"\n",
"# calculate the mean vaule for \"stroke\" column\n",
"avg_stroke = df[\"stroke\"].astype(\"float\").mean(axis = 0)\n",
"print(\"Average of stroke:\", avg_stroke)\n",
"\n",
"# replace NaN by mean value in \"stroke\" column\n",
"df[\"stroke\"].replace(np.nan, avg_stroke, inplace = True)\n",
"\n",
"-->\n"
]
},
{
"cell_type": "markdown",
"metadata": {},
"source": [
"<h4>Calculate the mean value for the 'horsepower' column:</h4>"
]
},
{
"cell_type": "code",
"execution_count": 22,
"metadata": {},
"outputs": [
{
"name": "stdout",
"output_type": "stream",
"text": [
"Average horsepower: 104.25615763546799\n"
]
}
],
"source": [
"df['horsepower'].replace(\"?\",np.nan, inplace=True)\n",
"avg_horsepower = df['horsepower'].astype('float').mean(axis=0)\n",
"print(\"Average horsepower:\", avg_horsepower)"
]
},
{
"cell_type": "markdown",
"metadata": {},
"source": [
"<h4>Replace \"NaN\" by mean value:</h4>"
]
},
{
"cell_type": "code",
"execution_count": 23,
"metadata": {},
"outputs": [],
"source": [
"df['horsepower'].replace(np.nan, avg_horsepower, inplace=True)"
]
},
{
"cell_type": "markdown",
"metadata": {},
"source": [
"<h4>Calculate the mean value for 'peak-rpm' column:</h4>"
]
},
{
"cell_type": "code",
"execution_count": 25,
"metadata": {},
"outputs": [
{
"name": "stdout",
"output_type": "stream",
"text": [
"Average peak rpm: 5125.369458128079\n"
]
}
],
"source": [
"df['peak-rpm'].replace('?',np.nan, inplace=True)\n",
"avg_peakrpm=df['peak-rpm'].astype('float').mean(axis=0)\n",
"print(\"Average peak rpm:\", avg_peakrpm)"
]
},
{
"cell_type": "markdown",
"metadata": {},
"source": [
"<h4>Replace NaN by mean value:</h4>"
]
},
{
"cell_type": "code",
"execution_count": 26,
"metadata": {},
"outputs": [],
"source": [
"df['peak-rpm'].replace(np.nan, avg_peakrpm, inplace=True)"
]
},
{
"cell_type": "markdown",
"metadata": {},
"source": [
"To see which values are present in a particular column, we can use the \".value_counts()\" method:"
]
},
{
"cell_type": "code",
"execution_count": 27,
"metadata": {
"collapsed": false,
"jupyter": {
"outputs_hidden": false
}
},
"outputs": [
{
"data": {
"text/plain": [
"four 114\n",
"two 89\n",
"? 2\n",
"Name: num-of-doors, dtype: int64"
]
},
"execution_count": 27,
"metadata": {},
"output_type": "execute_result"
}
],
"source": [
"df['num-of-doors'].value_counts()"
]
},
{
"cell_type": "markdown",
"metadata": {},
"source": [
"We can see that four doors are the most common type. We can also use the \".idxmax()\" method to calculate for us the most common type automatically:"
]
},
{
"cell_type": "code",
"execution_count": 28,
"metadata": {
"collapsed": false,
"jupyter": {
"outputs_hidden": false
}
},
"outputs": [
{
"data": {
"text/plain": [
"'four'"
]
},
"execution_count": 28,
"metadata": {},
"output_type": "execute_result"
}
],
"source": [
"df['num-of-doors'].value_counts().idxmax()"
]
},
{
"cell_type": "markdown",
"metadata": {},
"source": [
"The replacement procedure is very similar to what we have seen previously"
]
},
{
"cell_type": "code",
"execution_count": 29,
"metadata": {
"collapsed": false,
"jupyter": {
"outputs_hidden": false
}
},
"outputs": [],
"source": [
"#replace the missing 'num-of-doors' values by the most frequent \n",
"df[\"num-of-doors\"].replace(np.nan, \"four\", inplace=True)"
]
},
{
"cell_type": "markdown",
"metadata": {},
"source": [
"Finally, let's drop all rows that do not have price data:"
]
},
{
"cell_type": "code",
"execution_count": 42,
"metadata": {},
"outputs": [],
"source": [
"# simply drop whole row with NaN in \"price\" column\n",
"#print(df.describe)\n",
"df.dropna(subset=[\"price\"], axis=0, inplace=True)\n",
"# print(df[\"price\"])\n",
"# reset index, because we droped two rows\n",
"df.reset_index(drop=True, inplace=True)"
]
},
{
"cell_type": "code",
"execution_count": 43,
"metadata": {
"collapsed": false,
"jupyter": {
"outputs_hidden": false
}
},
"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>symboling</th>\n",
" <th>normalized-losses</th>\n",
" <th>make</th>\n",
" <th>fuel-type</th>\n",
" <th>aspiration</th>\n",
" <th>num-of-doors</th>\n",
" <th>body-style</th>\n",
" <th>drive-wheels</th>\n",
" <th>engine-location</th>\n",
" <th>wheel-base</th>\n",
" <th>...</th>\n",
" <th>engine-size</th>\n",
" <th>fuel-system</th>\n",
" <th>bore</th>\n",
" <th>stroke</th>\n",
" <th>compression-ratio</th>\n",
" <th>horsepower</th>\n",
" <th>peak-rpm</th>\n",
" <th>city-mpg</th>\n",
" <th>highway-mpg</th>\n",
" <th>price</th>\n",
" </tr>\n",
" </thead>\n",
" <tbody>\n",
" <tr>\n",
" <th>0</th>\n",
" <td>3</td>\n",
" <td>?</td>\n",
" <td>alfa-romero</td>\n",
" <td>gas</td>\n",
" <td>std</td>\n",
" <td>two</td>\n",
" <td>convertible</td>\n",
" <td>rwd</td>\n",
" <td>front</td>\n",
" <td>88.6</td>\n",
" <td>...</td>\n",
" <td>130</td>\n",
" <td>mpfi</td>\n",
" <td>3.47</td>\n",
" <td>2.68</td>\n",
" <td>9.0</td>\n",
" <td>111</td>\n",
" <td>5000</td>\n",
" <td>21</td>\n",
" <td>27</td>\n",
" <td>13495</td>\n",
" </tr>\n",
" <tr>\n",
" <th>1</th>\n",
" <td>3</td>\n",
" <td>?</td>\n",
" <td>alfa-romero</td>\n",
" <td>gas</td>\n",
" <td>std</td>\n",
" <td>two</td>\n",
" <td>convertible</td>\n",
" <td>rwd</td>\n",
" <td>front</td>\n",
" <td>88.6</td>\n",
" <td>...</td>\n",
" <td>130</td>\n",
" <td>mpfi</td>\n",
" <td>3.47</td>\n",
" <td>2.68</td>\n",
" <td>9.0</td>\n",
" <td>111</td>\n",
" <td>5000</td>\n",
" <td>21</td>\n",
" <td>27</td>\n",
" <td>16500</td>\n",
" </tr>\n",
" <tr>\n",
" <th>2</th>\n",
" <td>1</td>\n",
" <td>?</td>\n",
" <td>alfa-romero</td>\n",
" <td>gas</td>\n",
" <td>std</td>\n",
" <td>two</td>\n",
" <td>hatchback</td>\n",
" <td>rwd</td>\n",
" <td>front</td>\n",
" <td>94.5</td>\n",
" <td>...</td>\n",
" <td>152</td>\n",
" <td>mpfi</td>\n",
" <td>2.68</td>\n",
" <td>3.47</td>\n",
" <td>9.0</td>\n",
" <td>154</td>\n",
" <td>5000</td>\n",
" <td>19</td>\n",
" <td>26</td>\n",
" <td>16500</td>\n",
" </tr>\n",
" <tr>\n",
" <th>3</th>\n",
" <td>2</td>\n",
" <td>164</td>\n",
" <td>audi</td>\n",
" <td>gas</td>\n",
" <td>std</td>\n",
" <td>four</td>\n",
" <td>sedan</td>\n",
" <td>fwd</td>\n",
" <td>front</td>\n",
" <td>99.8</td>\n",
" <td>...</td>\n",
" <td>109</td>\n",
" <td>mpfi</td>\n",
" <td>3.19</td>\n",
" <td>3.40</td>\n",
" <td>10.0</td>\n",
" <td>102</td>\n",
" <td>5500</td>\n",
" <td>24</td>\n",
" <td>30</td>\n",
" <td>13950</td>\n",
" </tr>\n",
" <tr>\n",
" <th>4</th>\n",
" <td>2</td>\n",
" <td>164</td>\n",
" <td>audi</td>\n",
" <td>gas</td>\n",
" <td>std</td>\n",
" <td>four</td>\n",
" <td>sedan</td>\n",
" <td>4wd</td>\n",
" <td>front</td>\n",
" <td>99.4</td>\n",
" <td>...</td>\n",
" <td>136</td>\n",
" <td>mpfi</td>\n",
" <td>3.19</td>\n",
" <td>3.40</td>\n",
" <td>8.0</td>\n",
" <td>115</td>\n",
" <td>5500</td>\n",
" <td>18</td>\n",
" <td>22</td>\n",
" <td>17450</td>\n",
" </tr>\n",
" </tbody>\n",
"</table>\n",
"<p>5 rows × 26 columns</p>\n",
"</div>"
],
"text/plain": [
" symboling normalized-losses make fuel-type aspiration num-of-doors \\\n",
"0 3 ? alfa-romero gas std two \n",
"1 3 ? alfa-romero gas std two \n",
"2 1 ? alfa-romero gas std two \n",
"3 2 164 audi gas std four \n",
"4 2 164 audi gas std four \n",
"\n",
" body-style drive-wheels engine-location wheel-base ... engine-size \\\n",
"0 convertible rwd front 88.6 ... 130 \n",
"1 convertible rwd front 88.6 ... 130 \n",
"2 hatchback rwd front 94.5 ... 152 \n",
"3 sedan fwd front 99.8 ... 109 \n",
"4 sedan 4wd front 99.4 ... 136 \n",
"\n",
" fuel-system bore stroke compression-ratio horsepower peak-rpm city-mpg \\\n",
"0 mpfi 3.47 2.68 9.0 111 5000 21 \n",
"1 mpfi 3.47 2.68 9.0 111 5000 21 \n",
"2 mpfi 2.68 3.47 9.0 154 5000 19 \n",
"3 mpfi 3.19 3.40 10.0 102 5500 24 \n",
"4 mpfi 3.19 3.40 8.0 115 5500 18 \n",
"\n",
" highway-mpg price \n",
"0 27 13495 \n",
"1 27 16500 \n",
"2 26 16500 \n",
"3 30 13950 \n",
"4 22 17450 \n",
"\n",
"[5 rows x 26 columns]"
]
},
"execution_count": 43,
"metadata": {},
"output_type": "execute_result"
}
],
"source": [
"df.head()"
]
},
{
"cell_type": "markdown",
"metadata": {},
"source": [
"<b>Good!</b> Now, we obtain the dataset with no missing values."
]
},
{
"cell_type": "markdown",
"metadata": {},
"source": [
"<h3 id=\"correct_data_format\">Correct data format</h3>\n",
"<b>We are almost there!</b>\n",
"<p>The last step in data cleaning is checking and making sure that all data is in the correct format (int, float, text or other).</p>\n",
"\n",
"In Pandas, we use \n",
"<p><b>.dtype()</b> to check the data type</p>\n",
"<p><b>.astype()</b> to change the data type</p>"
]
},
{
"cell_type": "markdown",
"metadata": {},
"source": [
"<h4>Lets list the data types for each column</h4>"
]
},
{
"cell_type": "code",
"execution_count": 44,
"metadata": {
"collapsed": false,
"jupyter": {
"outputs_hidden": false
}
},
"outputs": [
{
"data": {
"text/plain": [
"symboling int64\n",
"normalized-losses object\n",
"make object\n",
"fuel-type object\n",
"aspiration object\n",
"num-of-doors object\n",
"body-style object\n",
"drive-wheels object\n",
"engine-location object\n",
"wheel-base float64\n",
"length float64\n",
"width float64\n",
"height float64\n",
"curb-weight int64\n",
"engine-type object\n",
"num-of-cylinders object\n",
"engine-size int64\n",
"fuel-system object\n",
"bore object\n",
"stroke object\n",
"compression-ratio float64\n",
"horsepower object\n",
"peak-rpm object\n",
"city-mpg int64\n",
"highway-mpg int64\n",
"price object\n",
"dtype: object"
]
},
"execution_count": 44,
"metadata": {},
"output_type": "execute_result"
}
],
"source": [
"df.dtypes"
]
},
{
"cell_type": "markdown",
"metadata": {},
"source": [
"<p>As we can see above, some columns are not of the correct data type. Numerical variables should have type 'float' or 'int', and variables with strings such as categories should have type 'object'. For example, 'bore' and 'stroke' variables are numerical values that describe the engines, so we should expect them to be of the type 'float' or 'int'; however, they are shown as type 'object'. We have to convert data types into a proper format for each column using the \"astype()\" method.</p> "
]
},
{
"cell_type": "markdown",
"metadata": {},
"source": [
"<h4>Convert data types to proper format</h4>"
]
},
{
"cell_type": "code",
"execution_count": 68,
"metadata": {
"collapsed": false,
"jupyter": {
"outputs_hidden": false
}
},
"outputs": [],
"source": [
"df.loc[:,[\"bore\", \"stroke\"]].replace('?',0, inplace=True)\n",
"df.loc[:,[\"bore\", \"stroke\"]] = df[[\"bore\", \"stroke\"]].astype(\"float\")\n",
"df.loc[:,\"normalized-losses\"].replace(np.nan,0, inplace=True)\n",
"df.loc[:,[\"normalized-losses\"]] = df[[\"normalized-losses\"]].astype(\"int\")\n",
"df.loc[:,[\"price\"]] = df[[\"price\"]].astype(\"float\")\n",
"df.loc[:,[\"peak-rpm\"]] = df[[\"peak-rpm\"]].astype(\"float\")"
]
},
{
"cell_type": "markdown",
"metadata": {},
"source": [
"<h4>Let us list the columns after the conversion</h4>"
]
},
{
"cell_type": "code",
"execution_count": 69,
"metadata": {
"collapsed": false,
"jupyter": {
"outputs_hidden": false
}
},
"outputs": [
{
"data": {
"text/plain": [
"symboling int64\n",
"normalized-losses int64\n",
"make object\n",
"fuel-type object\n",
"aspiration object\n",
"num-of-doors object\n",
"body-style object\n",
"drive-wheels object\n",
"engine-location object\n",
"wheel-base float64\n",
"length float64\n",
"width float64\n",
"height float64\n",
"curb-weight int64\n",
"engine-type object\n",
"num-of-cylinders object\n",
"engine-size int64\n",
"fuel-system object\n",
"bore float64\n",
"stroke float64\n",
"compression-ratio float64\n",
"horsepower object\n",
"peak-rpm float64\n",
"city-mpg int64\n",
"highway-mpg int64\n",
"price float64\n",
"dtype: object"
]
},
"execution_count": 69,
"metadata": {},
"output_type": "execute_result"
}
],
"source": [
"df.dtypes"
]
},
{
"cell_type": "markdown",
"metadata": {},
"source": [
"<b>Wonderful!</b>\n",
"\n",
"Now, we finally obtain the cleaned dataset with no missing values and all data in its proper format."
]
},
{
"cell_type": "markdown",
"metadata": {},
"source": [
"<h2 id=\"data_standardization\">Data Standardization</h2>\n",
"<p>\n",
"Data is usually collected from different agencies with different formats.\n",
"(Data Standardization is also a term for a particular type of data normalization, where we subtract the mean and divide by the standard deviation)\n",
"</p>\n",
" \n",
"<b>What is Standardization?</b>\n",
"<p>Standardization is the process of transforming data into a common format which allows the researcher to make the meaningful comparison.\n",
"</p>\n",
"\n",
"<b>Example</b>\n",
"<p>Transform mpg to L/100km:</p>\n",
"<p>In our dataset, the fuel consumption columns \"city-mpg\" and \"highway-mpg\" are represented by mpg (miles per gallon) unit. Assume we are developing an application in a country that accept the fuel consumption with L/100km standard</p>\n",
"<p>We will need to apply <b>data transformation</b> to transform mpg into L/100km?</p>\n"
]
},
{
"cell_type": "markdown",
"metadata": {},
"source": [
"<p>The formula for unit conversion is<p>\n",
"L/100km = 235 / mpg\n",
"<p>We can do many mathematical operations directly in Pandas.</p>"
]
},
{
"cell_type": "code",
"execution_count": 70,
"metadata": {
"collapsed": false,
"jupyter": {
"outputs_hidden": false
}
},
"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>symboling</th>\n",
" <th>normalized-losses</th>\n",
" <th>make</th>\n",
" <th>fuel-type</th>\n",
" <th>aspiration</th>\n",
" <th>num-of-doors</th>\n",
" <th>body-style</th>\n",
" <th>drive-wheels</th>\n",
" <th>engine-location</th>\n",
" <th>wheel-base</th>\n",
" <th>...</th>\n",
" <th>engine-size</th>\n",
" <th>fuel-system</th>\n",
" <th>bore</th>\n",
" <th>stroke</th>\n",
" <th>compression-ratio</th>\n",
" <th>horsepower</th>\n",
" <th>peak-rpm</th>\n",
" <th>city-mpg</th>\n",
" <th>highway-mpg</th>\n",
" <th>price</th>\n",
" </tr>\n",
" </thead>\n",
" <tbody>\n",
" <tr>\n",
" <th>0</th>\n",
" <td>3</td>\n",
" <td>0</td>\n",
" <td>alfa-romero</td>\n",
" <td>gas</td>\n",
" <td>std</td>\n",
" <td>two</td>\n",
" <td>convertible</td>\n",
" <td>rwd</td>\n",
" <td>front</td>\n",
" <td>88.6</td>\n",
" <td>...</td>\n",
" <td>130</td>\n",
" <td>mpfi</td>\n",
" <td>3.47</td>\n",
" <td>2.68</td>\n",
" <td>9.0</td>\n",
" <td>111</td>\n",
" <td>5000.0</td>\n",
" <td>21</td>\n",
" <td>27</td>\n",
" <td>13495.0</td>\n",
" </tr>\n",
" <tr>\n",
" <th>1</th>\n",
" <td>3</td>\n",
" <td>0</td>\n",
" <td>alfa-romero</td>\n",
" <td>gas</td>\n",
" <td>std</td>\n",
" <td>two</td>\n",
" <td>convertible</td>\n",
" <td>rwd</td>\n",
" <td>front</td>\n",
" <td>88.6</td>\n",
" <td>...</td>\n",
" <td>130</td>\n",
" <td>mpfi</td>\n",
" <td>3.47</td>\n",
" <td>2.68</td>\n",
" <td>9.0</td>\n",
" <td>111</td>\n",
" <td>5000.0</td>\n",
" <td>21</td>\n",
" <td>27</td>\n",
" <td>16500.0</td>\n",
" </tr>\n",
" <tr>\n",
" <th>2</th>\n",
" <td>1</td>\n",
" <td>0</td>\n",
" <td>alfa-romero</td>\n",
" <td>gas</td>\n",
" <td>std</td>\n",
" <td>two</td>\n",
" <td>hatchback</td>\n",
" <td>rwd</td>\n",
" <td>front</td>\n",
" <td>94.5</td>\n",
" <td>...</td>\n",
" <td>152</td>\n",
" <td>mpfi</td>\n",
" <td>2.68</td>\n",
" <td>3.47</td>\n",
" <td>9.0</td>\n",
" <td>154</td>\n",
" <td>5000.0</td>\n",
" <td>19</td>\n",
" <td>26</td>\n",
" <td>16500.0</td>\n",
" </tr>\n",
" <tr>\n",
" <th>3</th>\n",
" <td>2</td>\n",
" <td>164</td>\n",
" <td>audi</td>\n",
" <td>gas</td>\n",
" <td>std</td>\n",
" <td>four</td>\n",
" <td>sedan</td>\n",
" <td>fwd</td>\n",
" <td>front</td>\n",
" <td>99.8</td>\n",
" <td>...</td>\n",
" <td>109</td>\n",
" <td>mpfi</td>\n",
" <td>3.19</td>\n",
" <td>3.40</td>\n",
" <td>10.0</td>\n",
" <td>102</td>\n",
" <td>5500.0</td>\n",
" <td>24</td>\n",
" <td>30</td>\n",
" <td>13950.0</td>\n",
" </tr>\n",
" <tr>\n",
" <th>4</th>\n",
" <td>2</td>\n",
" <td>164</td>\n",
" <td>audi</td>\n",
" <td>gas</td>\n",
" <td>std</td>\n",
" <td>four</td>\n",
" <td>sedan</td>\n",
" <td>4wd</td>\n",
" <td>front</td>\n",
" <td>99.4</td>\n",
" <td>...</td>\n",
" <td>136</td>\n",
" <td>mpfi</td>\n",
" <td>3.19</td>\n",
" <td>3.40</td>\n",
" <td>8.0</td>\n",
" <td>115</td>\n",
" <td>5500.0</td>\n",
" <td>18</td>\n",
" <td>22</td>\n",
" <td>17450.0</td>\n",
" </tr>\n",
" </tbody>\n",
"</table>\n",
"<p>5 rows × 26 columns</p>\n",
"</div>"
],
"text/plain": [
" symboling normalized-losses make fuel-type aspiration \\\n",
"0 3 0 alfa-romero gas std \n",
"1 3 0 alfa-romero gas std \n",
"2 1 0 alfa-romero gas std \n",
"3 2 164 audi gas std \n",
"4 2 164 audi gas std \n",
"\n",
" num-of-doors body-style drive-wheels engine-location wheel-base ... \\\n",
"0 two convertible rwd front 88.6 ... \n",
"1 two convertible rwd front 88.6 ... \n",
"2 two hatchback rwd front 94.5 ... \n",
"3 four sedan fwd front 99.8 ... \n",
"4 four sedan 4wd front 99.4 ... \n",
"\n",
" engine-size fuel-system bore stroke compression-ratio horsepower \\\n",
"0 130 mpfi 3.47 2.68 9.0 111 \n",
"1 130 mpfi 3.47 2.68 9.0 111 \n",
"2 152 mpfi 2.68 3.47 9.0 154 \n",
"3 109 mpfi 3.19 3.40 10.0 102 \n",
"4 136 mpfi 3.19 3.40 8.0 115 \n",
"\n",
" peak-rpm city-mpg highway-mpg price \n",
"0 5000.0 21 27 13495.0 \n",
"1 5000.0 21 27 16500.0 \n",
"2 5000.0 19 26 16500.0 \n",
"3 5500.0 24 30 13950.0 \n",
"4 5500.0 18 22 17450.0 \n",
"\n",
"[5 rows x 26 columns]"
]
},
"execution_count": 70,
"metadata": {},
"output_type": "execute_result"
}
],
"source": [
"df.head()"
]
},
{
"cell_type": "code",
"execution_count": 71,
"metadata": {
"collapsed": false,
"jupyter": {
"outputs_hidden": false
}
},
"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>symboling</th>\n",
" <th>normalized-losses</th>\n",
" <th>make</th>\n",
" <th>fuel-type</th>\n",
" <th>aspiration</th>\n",
" <th>num-of-doors</th>\n",
" <th>body-style</th>\n",
" <th>drive-wheels</th>\n",
" <th>engine-location</th>\n",
" <th>wheel-base</th>\n",
" <th>...</th>\n",
" <th>fuel-system</th>\n",
" <th>bore</th>\n",
" <th>stroke</th>\n",
" <th>compression-ratio</th>\n",
" <th>horsepower</th>\n",
" <th>peak-rpm</th>\n",
" <th>city-mpg</th>\n",
" <th>highway-mpg</th>\n",
" <th>price</th>\n",
" <th>city-L/100km</th>\n",
" </tr>\n",
" </thead>\n",
" <tbody>\n",
" <tr>\n",
" <th>0</th>\n",
" <td>3</td>\n",
" <td>0</td>\n",
" <td>alfa-romero</td>\n",
" <td>gas</td>\n",
" <td>std</td>\n",
" <td>two</td>\n",
" <td>convertible</td>\n",
" <td>rwd</td>\n",
" <td>front</td>\n",
" <td>88.6</td>\n",
" <td>...</td>\n",
" <td>mpfi</td>\n",
" <td>3.47</td>\n",
" <td>2.68</td>\n",
" <td>9.0</td>\n",
" <td>111</td>\n",
" <td>5000.0</td>\n",
" <td>21</td>\n",
" <td>27</td>\n",
" <td>13495.0</td>\n",
" <td>11.190476</td>\n",
" </tr>\n",
" <tr>\n",
" <th>1</th>\n",
" <td>3</td>\n",
" <td>0</td>\n",
" <td>alfa-romero</td>\n",
" <td>gas</td>\n",
" <td>std</td>\n",
" <td>two</td>\n",
" <td>convertible</td>\n",
" <td>rwd</td>\n",
" <td>front</td>\n",
" <td>88.6</td>\n",
" <td>...</td>\n",
" <td>mpfi</td>\n",
" <td>3.47</td>\n",
" <td>2.68</td>\n",
" <td>9.0</td>\n",
" <td>111</td>\n",
" <td>5000.0</td>\n",
" <td>21</td>\n",
" <td>27</td>\n",
" <td>16500.0</td>\n",
" <td>11.190476</td>\n",
" </tr>\n",
" <tr>\n",
" <th>2</th>\n",
" <td>1</td>\n",
" <td>0</td>\n",
" <td>alfa-romero</td>\n",
" <td>gas</td>\n",
" <td>std</td>\n",
" <td>two</td>\n",
" <td>hatchback</td>\n",
" <td>rwd</td>\n",
" <td>front</td>\n",
" <td>94.5</td>\n",
" <td>...</td>\n",
" <td>mpfi</td>\n",
" <td>2.68</td>\n",
" <td>3.47</td>\n",
" <td>9.0</td>\n",
" <td>154</td>\n",
" <td>5000.0</td>\n",
" <td>19</td>\n",
" <td>26</td>\n",
" <td>16500.0</td>\n",
" <td>12.368421</td>\n",
" </tr>\n",
" <tr>\n",
" <th>3</th>\n",
" <td>2</td>\n",
" <td>164</td>\n",
" <td>audi</td>\n",
" <td>gas</td>\n",
" <td>std</td>\n",
" <td>four</td>\n",
" <td>sedan</td>\n",
" <td>fwd</td>\n",
" <td>front</td>\n",
" <td>99.8</td>\n",
" <td>...</td>\n",
" <td>mpfi</td>\n",
" <td>3.19</td>\n",
" <td>3.40</td>\n",
" <td>10.0</td>\n",
" <td>102</td>\n",
" <td>5500.0</td>\n",
" <td>24</td>\n",
" <td>30</td>\n",
" <td>13950.0</td>\n",
" <td>9.791667</td>\n",
" </tr>\n",
" <tr>\n",
" <th>4</th>\n",
" <td>2</td>\n",
" <td>164</td>\n",
" <td>audi</td>\n",
" <td>gas</td>\n",
" <td>std</td>\n",
" <td>four</td>\n",
" <td>sedan</td>\n",
" <td>4wd</td>\n",
" <td>front</td>\n",
" <td>99.4</td>\n",
" <td>...</td>\n",
" <td>mpfi</td>\n",
" <td>3.19</td>\n",
" <td>3.40</td>\n",
" <td>8.0</td>\n",
" <td>115</td>\n",
" <td>5500.0</td>\n",
" <td>18</td>\n",
" <td>22</td>\n",
" <td>17450.0</td>\n",
" <td>13.055556</td>\n",
" </tr>\n",
" </tbody>\n",
"</table>\n",
"<p>5 rows × 27 columns</p>\n",
"</div>"
],
"text/plain": [
" symboling normalized-losses make fuel-type aspiration \\\n",
"0 3 0 alfa-romero gas std \n",
"1 3 0 alfa-romero gas std \n",
"2 1 0 alfa-romero gas std \n",
"3 2 164 audi gas std \n",
"4 2 164 audi gas std \n",
"\n",
" num-of-doors body-style drive-wheels engine-location wheel-base ... \\\n",
"0 two convertible rwd front 88.6 ... \n",
"1 two convertible rwd front 88.6 ... \n",
"2 two hatchback rwd front 94.5 ... \n",
"3 four sedan fwd front 99.8 ... \n",
"4 four sedan 4wd front 99.4 ... \n",
"\n",
" fuel-system bore stroke compression-ratio horsepower peak-rpm city-mpg \\\n",
"0 mpfi 3.47 2.68 9.0 111 5000.0 21 \n",
"1 mpfi 3.47 2.68 9.0 111 5000.0 21 \n",
"2 mpfi 2.68 3.47 9.0 154 5000.0 19 \n",
"3 mpfi 3.19 3.40 10.0 102 5500.0 24 \n",
"4 mpfi 3.19 3.40 8.0 115 5500.0 18 \n",
"\n",
" highway-mpg price city-L/100km \n",
"0 27 13495.0 11.190476 \n",
"1 27 16500.0 11.190476 \n",
"2 26 16500.0 12.368421 \n",
"3 30 13950.0 9.791667 \n",
"4 22 17450.0 13.055556 \n",
"\n",
"[5 rows x 27 columns]"
]
},
"execution_count": 71,
"metadata": {},
"output_type": "execute_result"
}
],
"source": [
"# Convert mpg to L/100km by mathematical operation (235 divided by mpg)\n",
"df['city-L/100km'] = 235/df[\"city-mpg\"]\n",
"\n",
"# check your transformed data \n",
"df.head()"
]
},
{
"cell_type": "markdown",
"metadata": {},
"source": [
"<div class=\"alert alert-danger alertdanger\" style=\"margin-top: 20px\">\n",
"<h1> Question #2: </h1>\n",
"\n",
"<b>According to the example above, transform mpg to L/100km in the column of \"highway-mpg\", and change the name of column to \"highway-L/100km\".</b>\n",
"</div>"
]
},
{
"cell_type": "code",
"execution_count": 96,
"metadata": {
"collapsed": false,
"jupyter": {
"outputs_hidden": false
}
},
"outputs": [],
"source": [
"# Write your code below and press Shift+Enter to execute \n",
"df[\"highway-mpg\"] = 234 / df[\"highway-mpg\"]\n",
"df.rename(columns={\"highway-mpg\":\"highway-L/100km\"}, inplace=True)"
]
},
{
"cell_type": "markdown",
"metadata": {},
"source": [
"Double-click <b>here</b> for the solution.\n",
"\n",
"<!-- The answer is below:\n",
"\n",
"# transform mpg to L/100km by mathematical operation (235 divided by mpg)\n",
"df[\"highway-mpg\"] = 235/df[\"highway-mpg\"]\n",
"\n",
"# rename column name from \"highway-mpg\" to \"highway-L/100km\"\n",
"df.rename(columns={'\"highway-mpg\"':'highway-L/100km'}, inplace=True)\n",
"\n",
"# check your transformed data \n",
"df.head()\n",
"\n",
"-->\n"
]
},
{
"cell_type": "markdown",
"metadata": {},
"source": [
"<h2 id=\"data_normalization\">Data Normalization</h2>\n",
"\n",
"<b>Why normalization?</b>\n",
"<p>Normalization is the process of transforming values of several variables into a similar range. Typical normalizations include scaling the variable so the variable average is 0, scaling the variable so the variance is 1, or scaling variable so the variable values range from 0 to 1\n",
"</p>\n",
"\n",
"<b>Example</b>\n",
"<p>To demonstrate normalization, let's say we want to scale the columns \"length\", \"width\" and \"height\" </p>\n",
"<p><b>Target:</b>would like to Normalize those variables so their value ranges from 0 to 1.</p>\n",
"<p><b>Approach:</b> replace original value by (original value)/(maximum value)</p>"
]
},
{
"cell_type": "code",
"execution_count": 75,
"metadata": {
"collapsed": false,
"jupyter": {
"outputs_hidden": false
}
},
"outputs": [],
"source": [
"# replace (original value) by (original value)/(maximum value)\n",
"df['length'] = df['length']/df['length'].max()\n",
"df['width'] = df['width']/df['width'].max()"
]
},
{
"cell_type": "markdown",
"metadata": {},
"source": [
"<div class=\"alert alert-danger alertdanger\" style=\"margin-top: 20px\">\n",
"<h1> Questiont #3: </h1>\n",
"\n",
"<b>According to the example above, normalize the column \"height\".</b>\n",
"</div>"
]
},
{
"cell_type": "code",
"execution_count": 97,
"metadata": {
"collapsed": false,
"jupyter": {
"outputs_hidden": false
}
},
"outputs": [],
"source": [
"# Write your code below and press Shift+Enter to execute \n",
"df[\"height\"] = df['height']/df['height'].max()"
]
},
{
"cell_type": "markdown",
"metadata": {},
"source": [
"Double-click <b>here</b> for the solution.\n",
"\n",
"<!-- The answer is below:\n",
"\n",
"df['height'] = df['height']/df['height'].max() \n",
"# show the scaled columns\n",
"df[[\"length\",\"width\",\"height\"]].head()\n",
"\n",
"-->"
]
},
{
"cell_type": "markdown",
"metadata": {},
"source": [
"Here we can see, we've normalized \"length\", \"width\" and \"height\" in the range of [0,1]."
]
},
{
"cell_type": "markdown",
"metadata": {},
"source": [
"<h2 id=\"binning\">Binning</h2>\n",
"<b>Why binning?</b>\n",
"<p>\n",
" Binning is a process of transforming continuous numerical variables into discrete categorical 'bins', for grouped analysis.\n",
"</p>\n",
"\n",
"<b>Example: </b>\n",
"<p>In our dataset, \"horsepower\" is a real valued variable ranging from 48 to 288, it has 57 unique values. What if we only care about the price difference between cars with high horsepower, medium horsepower, and little horsepower (3 types)? Can we rearrange them into three ‘bins' to simplify analysis? </p>\n",
"\n",
"<p>We will use the Pandas method 'cut' to segment the 'horsepower' column into 3 bins </p>\n",
"\n"
]
},
{
"cell_type": "markdown",
"metadata": {},
"source": [
"<h3>Example of Binning Data In Pandas</h3>"
]
},
{
"cell_type": "markdown",
"metadata": {},
"source": [
" Convert data to correct format "
]
},
{
"cell_type": "code",
"execution_count": 100,
"metadata": {
"collapsed": false,
"jupyter": {
"outputs_hidden": false
}
},
"outputs": [],
"source": [
"df[\"horsepower\"]=df[\"horsepower\"].replace('?',0).astype(int, copy=True)"
]
},
{
"cell_type": "markdown",
"metadata": {},
"source": [
"Lets plot the histogram of horspower, to see what the distribution of horsepower looks like."
]
},
{
"cell_type": "code",
"execution_count": 101,
"metadata": {},
"outputs": [
{
"data": {
"text/plain": [
"Text(0.5, 1.0, 'horsepower bins')"
]
},
"execution_count": 101,
"metadata": {},
"output_type": "execute_result"
},
{
"data": {
"image/png": "iVBORw0KGgoAAAANSUhEUgAAAYUAAAEWCAYAAACJ0YulAAAABHNCSVQICAgIfAhkiAAAAAlwSFlzAAALEgAACxIB0t1+/AAAADh0RVh0U29mdHdhcmUAbWF0cGxvdGxpYiB2ZXJzaW9uMy4xLjEsIGh0dHA6Ly9tYXRwbG90bGliLm9yZy8QZhcZAAAXPklEQVR4nO3de7SddX3n8feHAKJylxOMeIky1Ck6GjVVGUek4gXBMYwWlFWcjKWNzlKrq9o2tlNHZ2onXmprrauuOF5iveIFyYyzVCaVoK2iQSOC0QYxIBCTA6KgTlHwO3/s33nYhpzknJM8Z59z8n6ttddz38/3d55kf/bz7P38dqoKSZIADhp1AZKkucNQkCR1DAVJUsdQkCR1DAVJUsdQkCR1DAWNVJJtSZ426jrmuiSnJblhD8vfmeTPZrMmLUwHj7oASfuuql4y6hq0MHimoAUhyYJ5g7OQ2qL5x1DQXLAsyZVJfpzko0kOm1iQ5PeSXJPkh0nWJ3nA0LJK8tIkW4GtGfirJDvbc12Z5JFt3XsleUuS65PsaJdb7t2WnZbkhiR/kuTmdknrt4f2c1SS9ycZT3Jdkv+S5KC27Lokj2vj57eaTm7Tv5vkU238oCSrk3w3yS1JLkxybFu2tG13QZLrgX+Y7A+1hxrfl+TPd2nPq9rfYnuSFw2te2aSbyW5PcmNSV69LwdPC4uhoLngXOAM4KHAo4D/BJDkqcD/aMuXANcBH9ll27OBJwAnA88ATgV+DTgaeD5wS1vvjW3+MuBfAScArx16nvsDx7X5K4G1SR7elr0dOAp4GPAU4D8CEy+yG4HT2vipwLVtnYnpjW3891utTwEeANwKvGOXtjwF+HXgmbv5G+2txt2te1Rb9wLgHUmOacveDby4qo4AHskeQkgHoKry4WNkD2AbcP7Q9JuAd7bxdwNvGlp2OPALYGmbLuCpQ8ufCvwz8ETgoKH5AX4KnDg07xTge238NOBO4L5Dyy8E/gxYBNwBnDy07MXApW38AmB9G98C/C7wkTZ9HfDYoWWnDz3HktaWg4GlrS0P28PfadIa2/j7gD8fWvf/AQcPrbsTeGIbv7614chRH38fc+/hmYLmgh8Mjf+MwYs/DN5RXzexoKp+wuCd/wlD639/aPk/AH/L4B34jiRrkxwJjAH3Aa5I8qMkPwI+0+ZPuLWqfjo0fV3b/3HAocN1tPGJGjYCT05yfwYB8lHgSUmWMninvrmt9xDgoqH9bwHuAo7fXVsmMVmNu3NLVd05ND38d30ecCZwXZKNSU7Zy351ADEUNJfdxODFFIAk9wXuB9w4tM6vdPNbVX9TVY8DHsHgctEfAjczeOf8iKo6uj2OqqrDhzY9pj3/hAe3/d/M4B39Q3ZZdmPb3zUMXnB/H7isqm5nEHKrgC9W1S/bNt8HnjW0/6Or6rCqmrQtuzFZjdNSVV+tqhXAYuBTDM44JMBQ0Nz2IeBFSZYluRfwF8DlVbVtdysn+Y0kT0hyCIPLRf8C3NVemN8F/FWSxW3dE5Lseu3+9UkOTfJk4NnAx6rqLgYvmm9IckSShwB/AHxgaLuNwMu4+/ODS3eZBnhne46HtP2PJVkxg7/JPWqczsZt299OclRV/QK4jcEZiwQYCprDqmoDg+v6nwC2AycCL9jDJkcyePG/lcGllVuAt7RlfwxcA3w5yW3A/wWGP6T9QdvuJuCDwEuq6ttt2csZhMy1wBcZhNV7hrbdCBwBXDbJNMDbgPXA55LcDnyZwQfk07GnGqfjhcC29nd4CXD+DJ5DC1Sq/JEdHdiSnAZ8oKoeOOpapFHzTEGS1DEUJEkdLx9JkjqeKUiSOvOi463jjjuuli5dOuoyJGleueKKK26uqrG9r3m3eREKS5cuZdOmTaMuQ5LmlSTX7X2tX+XlI0lSx1CQJHUMBUlSx1CQJHUMBUlSx1CQJHUMBUlSx1CQJHUMBUlSZ17c0azpWbr60yPZ77Y1Z41kv5L2H88UJEkdQ0GS1DEUJEkdQ0GS1DEUJEkdQ0GS1DEUJEkdQ0GS1DEUJEkdQ0GS1DEUJEkdQ0GS1OktFJI8PMnmocdtSV6Z5NgklyTZ2obH9FWDJGl6eguFqvpOVS2rqmXA44CfARcBq4ENVXUSsKFNS5LmgNm6fHQ68N2qug5YAaxr89cBZ89SDZKkvZitUHgB8OE2fnxVbQdow8W72yDJqiSbkmwaHx+fpTIl6cDWeygkORR4DvCx6WxXVWuranlVLR8bG+unOEnSr5iNM4VnAV+rqh1tekeSJQBtuHMWapAkTcFshMJ53H3pCGA9sLKNrwQunoUaJElT0GsoJLkP8HTgk0Oz1wBPT7K1LVvTZw2SpKk7uM8nr6qfAffbZd4tDL6NJEmaY7yjWZLUMRQkSR1DQZLUMRQkSR1DQZLUMRQkSR1DQZLUMRQkSR1DQZLUMRQkSR1DQZLUMRQkSR1DQZLUMRQkSR1DQZLUMRQkSZ1ef2RHB5alqz89sn1vW3PWyPYtLSSeKUiSOn3/RvPRST6e5NtJtiQ5JcmxSS5JsrUNj+mzBknS1PV9pvA24DNV9a+BRwNbgNXAhqo6CdjQpiVJc0BvoZDkSOBU4N0AVfXzqvoRsAJY11ZbB5zdVw2SpOnp80zhYcA48N4kX0/yP5PcFzi+qrYDtOHi3W2cZFWSTUk2jY+P91imJGlCn6FwMPBY4O+q6jHAT5nGpaKqWltVy6tq+djYWF81SpKG9BkKNwA3VNXlbfrjDEJiR5IlAG24s8caJEnT0FsoVNUPgO8neXibdTrwLWA9sLLNWwlc3FcNkqTp6fvmtZcDH0xyKHAt8CIGQXRhkguA64Fzeq5BkjRFvYZCVW0Glu9m0el97leSNDPe0SxJ6hgKkqSOoSBJ6hgKkqSOoSBJ6hgKkqSOoSBJ6hgKkqSOoSBJ6hgKkqSOoSBJ6hgKkqSOoSBJ6hgKkqSOoSBJ6hgKkqSOoSBJ6hgKkqROrz/HmWQbcDtwF3BnVS1PcizwUWApsA04t6pu7bMOSdLUzMaZwm9W1bKqmvit5tXAhqo6CdjQpiVJc8AoLh+tANa18XXA2SOoQZK0G32HQgGfS3JFklVt3vFVtR2gDRf3XIMkaYp6/UwBeFJV3ZRkMXBJkm9PdcMWIqsAHvzgB/dVnyRpSK9nClV1UxvuBC4CHg/sSLIEoA13TrLt2qpaXlXLx8bG+ixTktT0FgpJ7pvkiIlx4BnAVcB6YGVbbSVwcV81SJKmp8/LR8cDFyWZ2M+HquozSb4KXJjkAuB64Jwea5AkTUNvoVBV1wKP3s38W4DT+9qvJGnmvKNZktQxFCRJHUNBktQxFCRJHUNBktQxFCRJHUNBktQxFCRJHUNBktQxFCRJHUNBktSZUigk2TCVeZKk+W2PHeIlOQy4D3BckmOAtEVHAg/ouTZJ0izbWy+pLwZeySAAruDuULgNeEePdUmSRmCPoVBVbwPeluTlVfX2WapJkjQiU/o9hap6e5J/Cywd3qaq3t9TXZKkEZhSKCT5e+BEYDNwV5tdgKEgSQvIVH95bTlwclVVn8VIkkZrqvcpXAXcv89CJEmjN9UzheOAbyX5CnDHxMyqes7eNkyyCNgE3FhVz05yLPBRBp9PbAPOrapbp1m3JKkHUw2F1+3DPl4BbGFwbwPAamBDVa1JsrpN//E+PL8kaT+Z6rePNs7kyZM8EDgLeAPwB232CuC0Nr4OuBRDQZLmhKl2c3F7ktva41+S3JXktils+tfAHwG/HJp3fFVtB2jDxZPsc1WSTUk2jY+PT6VMSdI+mlIoVNURVXVkexwGPA/42z1tk+TZwM6qumImhVXV2qpaXlXLx8bGZvIUkqRpmupnCr+iqj7VPg/YkycBz0lyJnAYcGSSDwA7kiypqu1JlgA7Z1KDJGn/m+rNa88dmjyIwX0Le7xnoapeA7ymbX8a8OqqOj/Jm4GVwJo2vHj6ZUuS+jDVM4V/PzR+J4Ovkq6Y4T7XABcmuQC4Hjhnhs8jSdrPpvrtoxfty06q6lIG3zKiqm4BTt+X55Mk9WOq3z56YJKLkuxMsiPJJ9rXTSVJC8hUu7l4L7Cewe8qnAD8rzZPkrSATDUUxqrqvVV1Z3u8D/B7opK0wEw1FG5Ocn6SRe1xPnBLn4VJkmbfVEPhd4BzgR8A24HfAvbpw2dJ0twz1a+k/ndg5URvpq2n07cwCAtJ0gIx1TOFRw13b11VPwQe009JkqRRmWooHJTkmImJdqYwoy4yJElz11Rf2P8S+KckH2fQvcW5DLrDliQtIFO9o/n9STYBTwUCPLeqvtVrZZKkWTflS0AtBAwCSVrApvqZgiTpAGAoSJI6hoIkqWMoSJI6hoIkqWMoSJI6hoIkqdNbKCQ5LMlXknwjydVJXt/mH5vkkiRb2/CYvT2XJGl29HmmcAfw1Kp6NLAMOCPJE4HVwIaqOgnY0KYlSXNAb6FQAz9pk4e0RwErgHVt/jrg7L5qkCRNT6+fKbRfadsM7AQuqarLgeOrajtAGy6eZNtVSTYl2TQ+Pt5nmZKkptdQqKq7qmoZ8EDg8UkeOY1t11bV8qpaPjbmz0FL0myYlW8fVdWPgEuBM4AdSZYAtOHO2ahBkrR3fX77aCzJ0W383sDTgG8D64GVbbWVwMV91SBJmp4+fz1tCbAuySIG4XNhVf3vJF8CLkxyAXA9cE6PNUiSpqG3UKiqK9nN7zhX1S3A6X3tV5I0c97RLEnqGAqSpI6hIEnqGAqSpI6hIEnqGAqSpI6hIEnqGAqSpI6hIEnqGAqSpI6hIEnqGAqSpI6hIEnq9Nl1trTgLV396ZHte9uas0a2by1cnilIkjqGgiSp4+UjLQijvIwjLSSeKUiSOr2FQpIHJfl8ki1Jrk7yijb/2CSXJNnahsf0VYMkaXr6PFO4E3hVVf068ETgpUlOBlYDG6rqJGBDm5YkzQG9hUJVba+qr7Xx24EtwAnACmBdW20dcHZfNUiSpmdWPlNIshR4DHA5cHxVbYdBcACLJ9lmVZJNSTaNj4/PRpmSdMDrPRSSHA58AnhlVd021e2qam1VLa+q5WNjY/0VKEnq9BoKSQ5hEAgfrKpPttk7kixpy5cAO/usQZI0dX1++yjAu4EtVfXWoUXrgZVtfCVwcV81SJKmp8+b154EvBD4ZpLNbd6fAGuAC5NcAFwPnNNjDZKkaegtFKrqi0AmWXx6X/uVJM2cdzRLkjqGgiSpYyhIkjqGgiSpYyhIkjqGgiSpYyhIkjqGgiSpYyhIkjqGgiSpYyhIkjqGgiSpYyhIkjqGgiSpYyhIkjqGgiSpYyhIkjqGgiSp01soJHlPkp1Jrhqad2ySS5JsbcNj+tq/JGn6+jxTeB9wxi7zVgMbquokYEObliTNEb2FQlVdBvxwl9krgHVtfB1wdl/7lyRN32x/pnB8VW0HaMPFs7x/SdIezNkPmpOsSrIpyabx8fFRlyNJB4TZDoUdSZYAtOHOyVasqrVVtbyqlo+Njc1agZJ0IDt4lve3HlgJrGnDi2d5/9KCsXT1p0ey321rzhrJfjU7+vxK6oeBLwEPT3JDkgsYhMHTk2wFnt6mJUlzRG9nClV13iSLTu9rn5KkfTNnP2iWJM0+Q0GS1DEUJEkdQ0GS1DEUJEmd2b5P4YAyqu+RS9JMeaYgSeoYCpKkjqEgSeoYCpKkjqEgSeoYCpKkjqEgSeoYCpKkjjevSZqWUd6U6Q/89M8zBUlSZ8GfKdjVhKR9dSCdHXmmIEnqGAqSpM5IQiHJGUm+k+SaJKtHUYMk6Z5mPRSSLALeATwLOBk4L8nJs12HJOmeRnGm8Hjgmqq6tqp+DnwEWDGCOiRJuxjFt49OAL4/NH0D8IRdV0qyCljVJn+S5Dsz3N9xwM0z3Hausk3zg23az/LG/f6Uc/4YzaDNw216yHQ3HkUoZDfz6h4zqtYCa/d5Z8mmqlq+r88zl9im+cE2zX0LrT2w720axeWjG4AHDU0/ELhpBHVIknYxilD4KnBSkocmORR4AbB+BHVIknYx65ePqurOJC8DPgssAt5TVVf3uMt9vgQ1B9mm+cE2zX0LrT2wj21K1T0u50uSDlDe0SxJ6hgKkqTOgg6FhdCdRpJtSb6ZZHOSTW3esUkuSbK1DY8ZdZ17kuQ9SXYmuWpo3qRtSPKadsy+k+SZo6l6zyZp0+uS3NiO1eYkZw4tmw9telCSzyfZkuTqJK9o8+ftsdpDm+blsUpyWJKvJPlGa8/r2/z9d4yqakE+GHyI/V3gYcChwDeAk0dd1wzasQ04bpd5bwJWt/HVwBtHXede2nAq8Fjgqr21gUHXJ98A7gU8tB3DRaNuwxTb9Drg1btZd760aQnw2DZ+BPDPrfZ5e6z20KZ5eawY3Od1eBs/BLgceOL+PEYL+UxhIXensQJY18bXAWePsJa9qqrLgB/uMnuyNqwAPlJVd1TV94BrGBzLOWWSNk1mvrRpe1V9rY3fDmxh0APBvD1We2jTZOZ0m2rgJ23ykPYo9uMxWsihsLvuNPb0j2GuKuBzSa5oXX8AHF9V22Hwjx5YPLLqZm6yNsz34/ayJFe2y0sTp/Dzrk1JlgKPYfBOdEEcq13aBPP0WCVZlGQzsBO4pKr26zFayKEwpe405oEnVdVjGfQq+9Ikp466oJ7N5+P2d8CJwDJgO/CXbf68alOSw4FPAK+sqtv2tOpu5s3Jdu2mTfP2WFXVXVW1jEFvEI9P8sg9rD7t9izkUFgQ3WlU1U1tuBO4iMGp344kSwDacOfoKpyxydowb49bVe1o/2F/CbyLu0/T502bkhzC4MXzg1X1yTZ7Xh+r3bVpIRyrqvoRcClwBvvxGC3kUJj33WkkuW+SIybGgWcAVzFox8q22krg4tFUuE8ma8N64AVJ7pXkocBJwFdGUN+0TfynbP4Dg2MF86RNSQK8G9hSVW8dWjRvj9VkbZqvxyrJWJKj2/i9gacB32Z/HqNRf5re8yf1ZzL4tsF3gT8ddT0zqP9hDL458A3g6ok2APcDNgBb2/DYUde6l3Z8mMEp+i8YvHO5YE9tAP60HbPvAM8adf3TaNPfA98Ermz/GZfMszb9OwaXFq4ENrfHmfP5WO2hTfPyWAGPAr7e6r4KeG2bv9+Okd1cSJI6C/nykSRpmgwFSVLHUJAkdQwFSVLHUJAkdQwFLShJlg73XCppegwFqUky6z9POxPzpU7NT4aCFqJFSd7V+pv/XJJ7J1mW5MutA7SLJjpAS3Jpkr9IshF4RZJzklzV+qu/rK2zKMmbk3y1bf/iNv+0JJe15/tWkncmOagtOy+D38G4Kskb27xzk7y1jb8iybVt/MQkX2zjj0uysXWA+Nmhrgt+pc7Z/XPqQOI7Di1EJwHnVdXvJbkQeB7wR8DLq2pjkv8G/FfglW39o6vqKQBJvgk8s6punOhOgMHdyj+uqt9Ici/gH5N8ri17PIM+668DPgM8N8k/AW8EHgfcyqCX27OBy4A/bNs9GbglyQkM7rr9Quuj5+3AiqoaT/J84A3A7+xap9QXQ0EL0feqanMbv4JBb5hHV9XGNm8d8LGh9T86NP6PwPtamEx0CPcM4FFJfqtNH8UgeH4OfKWqJt7xf5jBC/wvgEurarzN/yBwalV9KsnhrT+rBwEfYvBjPU9u+3o48EjgkkGXPSxi0JXG7uqUemEoaCG6Y2j8LuDoyVZsfjoxUlUvSfIE4Cxgc5JlDLoffnlVfXZ4oySncc9uiIvdd1c84UvAixj0Q/MFBmcBpwCvAh4MXF1Vp+ytTqkvfqagA8GPgVuTPLlNvxDYuLsVk5xYVZdX1WuBmxm8o/8s8J/b5R2S/FrrtRYG/dk/tH2W8Hzgiwx+xOUpSY5Lsgg4b2h/lwGvbsOvA78J3FFVP2YQFGNJTmn7OSTJI/bfn0HaO88UdKBYCbwzyX2Aaxm8W9+dNyc5icG7/Q0Meqi9ElgKfK11xTzO3T93+CVgDfBvGLzQX1RVv0zyGuDz7Xn+T1VNdGX8BQZBc1lV3ZXk+wy6Pqaqft4uUf1NkqMY/P/8awY95Eqzwl5SpRlql49eXVXPHnUt0v7i5SNJUsczBUlSxzMFSVLHUJAkdQwFSVLHUJAkdQwFSVLn/wMTsupcD4pPjQAAAABJRU5ErkJggg==\n",
"text/plain": [
"<Figure size 432x288 with 1 Axes>"
]
},
"metadata": {
"needs_background": "light"
},
"output_type": "display_data"
}
],
"source": [
"%matplotlib inline\n",
"import matplotlib as plt\n",
"from matplotlib import pyplot\n",
"plt.pyplot.hist(df[\"horsepower\"])\n",
"\n",
"# set x/y labels and plot title\n",
"plt.pyplot.xlabel(\"horsepower\")\n",
"plt.pyplot.ylabel(\"count\")\n",
"plt.pyplot.title(\"horsepower bins\")"
]
},
{
"cell_type": "markdown",
"metadata": {},
"source": [
"<p>We would like 3 bins of equal size bandwidth so we use numpy's <code>linspace(start_value, end_value, numbers_generated</code> function.</p>\n",
"<p>Since we want to include the minimum value of horsepower we want to set start_value=min(df[\"horsepower\"]).</p>\n",
"<p>Since we want to include the maximum value of horsepower we want to set end_value=max(df[\"horsepower\"]).</p>\n",
"<p>Since we are building 3 bins of equal length, there should be 4 dividers, so numbers_generated=4.</p>"
]
},
{
"cell_type": "markdown",
"metadata": {},
"source": [
"We build a bin array, with a minimum value to a maximum value, with bandwidth calculated above. The bins will be values used to determine when one bin ends and another begins."
]
},
{
"cell_type": "code",
"execution_count": 102,
"metadata": {
"collapsed": false,
"jupyter": {
"outputs_hidden": false
}
},
"outputs": [
{
"data": {
"text/plain": [
"array([ 0., 96., 192., 288.])"
]
},
"execution_count": 102,
"metadata": {},
"output_type": "execute_result"
}
],
"source": [
"bins = np.linspace(min(df[\"horsepower\"]), max(df[\"horsepower\"]), 4)\n",
"bins"
]
},
{
"cell_type": "markdown",
"metadata": {},
"source": [
" We set group names:"
]
},
{
"cell_type": "code",
"execution_count": 107,
"metadata": {},
"outputs": [],
"source": [
"group_names = ['Low', 'Medium', 'High']"
]
},
{
"cell_type": "markdown",
"metadata": {},
"source": [
" We apply the function \"cut\" the determine what each value of \"df['horsepower']\" belongs to. "
]
},
{
"cell_type": "code",
"execution_count": 108,
"metadata": {
"collapsed": false,
"jupyter": {
"outputs_hidden": false
}
},
"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>horsepower</th>\n",
" <th>horsepower-binned</th>\n",
" </tr>\n",
" </thead>\n",
" <tbody>\n",
" <tr>\n",
" <th>0</th>\n",
" <td>111</td>\n",
" <td>Medium</td>\n",
" </tr>\n",
" <tr>\n",
" <th>1</th>\n",
" <td>111</td>\n",
" <td>Medium</td>\n",
" </tr>\n",
" <tr>\n",
" <th>2</th>\n",
" <td>154</td>\n",
" <td>Medium</td>\n",
" </tr>\n",
" <tr>\n",
" <th>3</th>\n",
" <td>102</td>\n",
" <td>Medium</td>\n",
" </tr>\n",
" <tr>\n",
" <th>4</th>\n",
" <td>115</td>\n",
" <td>Medium</td>\n",
" </tr>\n",
" <tr>\n",
" <th>5</th>\n",
" <td>110</td>\n",
" <td>Medium</td>\n",
" </tr>\n",
" <tr>\n",
" <th>6</th>\n",
" <td>110</td>\n",
" <td>Medium</td>\n",
" </tr>\n",
" <tr>\n",
" <th>7</th>\n",
" <td>110</td>\n",
" <td>Medium</td>\n",
" </tr>\n",
" <tr>\n",
" <th>8</th>\n",
" <td>140</td>\n",
" <td>Medium</td>\n",
" </tr>\n",
" <tr>\n",
" <th>9</th>\n",
" <td>160</td>\n",
" <td>Medium</td>\n",
" </tr>\n",
" <tr>\n",
" <th>10</th>\n",
" <td>101</td>\n",
" <td>Medium</td>\n",
" </tr>\n",
" <tr>\n",
" <th>11</th>\n",
" <td>101</td>\n",
" <td>Medium</td>\n",
" </tr>\n",
" <tr>\n",
" <th>12</th>\n",
" <td>121</td>\n",
" <td>Medium</td>\n",
" </tr>\n",
" <tr>\n",
" <th>13</th>\n",
" <td>121</td>\n",
" <td>Medium</td>\n",
" </tr>\n",
" <tr>\n",
" <th>14</th>\n",
" <td>121</td>\n",
" <td>Medium</td>\n",
" </tr>\n",
" <tr>\n",
" <th>15</th>\n",
" <td>182</td>\n",
" <td>Medium</td>\n",
" </tr>\n",
" <tr>\n",
" <th>16</th>\n",
" <td>182</td>\n",
" <td>Medium</td>\n",
" </tr>\n",
" <tr>\n",
" <th>17</th>\n",
" <td>182</td>\n",
" <td>Medium</td>\n",
" </tr>\n",
" <tr>\n",
" <th>18</th>\n",
" <td>48</td>\n",
" <td>Low</td>\n",
" </tr>\n",
" <tr>\n",
" <th>19</th>\n",
" <td>70</td>\n",
" <td>Low</td>\n",
" </tr>\n",
" </tbody>\n",
"</table>\n",
"</div>"
],
"text/plain": [
" horsepower horsepower-binned\n",
"0 111 Medium\n",
"1 111 Medium\n",
"2 154 Medium\n",
"3 102 Medium\n",
"4 115 Medium\n",
"5 110 Medium\n",
"6 110 Medium\n",
"7 110 Medium\n",
"8 140 Medium\n",
"9 160 Medium\n",
"10 101 Medium\n",
"11 101 Medium\n",
"12 121 Medium\n",
"13 121 Medium\n",
"14 121 Medium\n",
"15 182 Medium\n",
"16 182 Medium\n",
"17 182 Medium\n",
"18 48 Low\n",
"19 70 Low"
]
},
"execution_count": 108,
"metadata": {},
"output_type": "execute_result"
}
],
"source": [
"df['horsepower-binned'] = pd.cut(df['horsepower'], bins, labels=group_names, include_lowest=True )\n",
"df[['horsepower','horsepower-binned']].head(20)"
]
},
{
"cell_type": "markdown",
"metadata": {},
"source": [
"Lets see the number of vehicles in each bin."
]
},
{
"cell_type": "code",
"execution_count": 109,
"metadata": {},
"outputs": [
{
"data": {
"text/plain": [
"Low 106\n",
"Medium 93\n",
"High 6\n",
"Name: horsepower-binned, dtype: int64"
]
},
"execution_count": 109,
"metadata": {},
"output_type": "execute_result"
}
],
"source": [
"df[\"horsepower-binned\"].value_counts()"
]
},
{
"cell_type": "markdown",
"metadata": {},
"source": [
"Lets plot the distribution of each bin."
]
},
{
"cell_type": "code",
"execution_count": 110,
"metadata": {},
"outputs": [
{
"data": {
"text/plain": [
"Text(0.5, 1.0, 'horsepower bins')"
]
},
"execution_count": 110,
"metadata": {},
"output_type": "execute_result"
},
{
"data": {
"image/png": "iVBORw0KGgoAAAANSUhEUgAAAYUAAAEWCAYAAACJ0YulAAAABHNCSVQICAgIfAhkiAAAAAlwSFlzAAALEgAACxIB0t1+/AAAADh0RVh0U29mdHdhcmUAbWF0cGxvdGxpYiB2ZXJzaW9uMy4xLjEsIGh0dHA6Ly9tYXRwbG90bGliLm9yZy8QZhcZAAAV6ElEQVR4nO3de7RkZX3m8e9DNwrIXRpsQWkkxAkyirFR0QGJxEvESbM0Ki4xLZIgWUZxIknazEQdLwleJsbbhNUZEbwi8Qajjsq0ocGoaKMEhdbAIJfGFg5IEIkBgd/8sd/eFsfTzelDn6pzTn0/a9Wqvd93X35Ve516au869VaqCkmSALYbdQGSpLnDUJAk9QwFSVLPUJAk9QwFSVLPUJAk9QwFzUlJrkny26OuY65LclSSDVvoPz3JXw6zJs1vi0ddgKTZU1Unj7oGzS+eKWhBS7Jg3vgspMeiuctQ0Fx2aJLLktyW5BNJdtjUkeQPk1yV5CdJzkvy8IG+SvLKJFcCV6bzriQ3tW1dluSQtuyDk7wzyXVJbmyXW3ZsfUcl2ZDkL5Lc3C5pvWRgP7sl+VCSiSTXJvlvSbZrfdcmeUKbPr7VdHCb/4Mkn23T2yVZleT/JbklyTlJ9mx9y9p6Jya5DvjK5p6oLdR4ZpK3THo8r23PxcYkJwws+5wkVyS5PckNSU59IAdP85OhoLnshcCzgQOAxwIvA0jydOCvW/9S4Frg7EnrHgs8CTgYeCZwJPDrwO7Ai4Bb2nJva+2HAr8G7Au8fmA7DwP2au0rgdVJHt363gvsBjwKeBrw+8CmF9m1wFFt+kjg6rbMpvm1bfrVrdanAQ8HbgXeP+mxPA34DeBZUzxH91fjVMvu1pY9EXh/kj1a3weAV1TVLsAhbCGEtIBVlTdvc+4GXAMcPzD/duD0Nv0B4O0DfTsDvwCWtfkCnj7Q/3TgX4AnA9sNtAe4AzhwoO1w4Idt+ijgbuAhA/3nAH8JLALuBA4e6HsFcEGbPhE4r02vB/4AOLvNXwv85kDf0QPbWNoey2JgWXssj9rC87TZGtv0mcBbBpb9ObB4YNmbgCe36evaY9h11Mff2+huniloLvvxwPS/0b34Q/eO+tpNHVX1M7p3/vsOLH/9QP9XgPfRvQO/McnqJLsCS4CdgEuS/GuSfwW+2No3ubWq7hiYv7btfy/gQYN1tOlNNawFjkjyMLoA+QTw1CTL6N6pX9qW2x/4zMD+1wP3APtM9Vg2Y3M1TuWWqrp7YH7weX0+8Bzg2iRrkxx+P/vVAmQoaD76Ed2LKQBJHgI8FLhhYJn7DP9bVe+pqicAj6G7XPSnwM1075wfU1W7t9tuVbXzwKp7tO1v8si2/5vp3tHvP6nvhra/q+hecF8NXFhVt9OF3EnAV6vq3rbO9cDvDOx/96raoao2+1imsLkat0pVfauqVgB7A5+lO+PQmDEUNB99DDghyaFJHgz8FXBxVV0z1cJJDkvypCTb010u+nfgnvbC/PfAu5Ls3ZbdN8nka/f/PcmDkhwBPBf4h6q6h+5F861JdkmyP/AnwEcG1lsL/DG//PzggknzAKe3bezf9r8kyYoZPCe/UuPWrNzWfUmS3arqF8BP6c5YNGYMBc07VbWG7rr+p4CNwIHAcVtYZVe6F/9b6S6t3AK8s/X9OXAV8I0kPwX+LzD4Ie2P23o/Aj4KnFxV3299r6ILmauBr9KF1RkD664FdgEu3Mw8wLuB84AvJ7kd+AbdB+RbY0s1bo2XAte05+Fk4PgZbEPzXKr8kR1pKkmOAj5SVfuNuhZpWDxTkCT1DAVJUs/LR5KknmcKkqTevB5ga6+99qply5aNugxJmlcuueSSm6tqyVR98zoUli1bxrp160ZdhiTNK0mu3Vyfl48kST1DQZLUMxQkST1DQZLUMxQkST1DQZLUMxQkST1DQZLUMxQkSb15/Y3mB2rZqs+PuoQF65rTjhl1CZJmwDMFSVLPUJAk9QwFSVLPUJAk9QwFSVLPUJAk9QwFSVJv1kIhyRlJbkryvYG2PZOcn+TKdr/HQN/rklyV5AdJnjVbdUmSNm82zxTOBJ49qW0VsKaqDgLWtHmSHAwcBzymrfM/kyyaxdokSVOYtVCoqguBn0xqXgGc1abPAo4daD+7qu6sqh8CVwFPnK3aJElTG/ZnCvtU1UaAdr93a98XuH5guQ2t7VckOSnJuiTrJiYmZrVYSRo3c+WD5kzRVlMtWFWrq2p5VS1fsmTJLJclSeNl2KFwY5KlAO3+pta+AXjEwHL7AT8acm2SNPaGHQrnASvb9Erg3IH245I8OMkBwEHAN4dcmySNvVkbOjvJx4GjgL2SbADeAJwGnJPkROA64AUAVXV5knOAK4C7gVdW1T2zVZskaWqzFgpV9eLNdB29meXfCrx1tuqRJN2/ufJBsyRpDjAUJEk9Q0GS1DMUJEk9Q0GS1DMUJEk9Q0GS1DMUJEm9WfvymrStLVv1+VGXsGBdc9oxoy5Bc4RnCpKknqEgSeoZCpKknqEgSeoZCpKknqEgSeoZCpKknqEgSeoZCpKknqEgSeoZCpKknqEgSeoZCpKknqEgSeoZCpKknqEgSeoZCpKknqEgSeoZCpKknqEgSeoZCpKk3khCIcl/SXJ5ku8l+XiSHZLsmeT8JFe2+z1GUZskjbOhh0KSfYFXA8ur6hBgEXAcsApYU1UHAWvavCRpiEZ1+WgxsGOSxcBOwI+AFcBZrf8s4NgR1SZJY2vooVBVNwDvBK4DNgK3VdWXgX2qamNbZiOw97Brk6RxN4rLR3vQnRUcADwceEiS47di/ZOSrEuybmJiYrbKlKSxNIrLR78N/LCqJqrqF8CngacANyZZCtDub5pq5apaXVXLq2r5kiVLhla0JI2DUYTCdcCTk+yUJMDRwHrgPGBlW2YlcO4IapOksbZ42DusqouTfBL4NnA38B1gNbAzcE6SE+mC4wXDrk2Sxt3QQwGgqt4AvGFS8510Zw2SpBHxG82SpJ6hIEnqGQqSpJ6hIEnqGQqSpJ6hIEnqGQqSpJ6hIEnqGQqSpJ6hIEnqGQqSpJ6hIEnqGQqSpJ6hIEnqGQqSpJ6hIEnqGQqSpJ6hIEnqGQqSpJ6hIEnqGQqSpJ6hIEnqGQqSpJ6hIEnqGQqSpJ6hIEnqGQqSpJ6hIEnqGQqSpJ6hIEnqjSQUkuye5JNJvp9kfZLDk+yZ5PwkV7b7PUZRmySNs2mFQpI102nbCu8GvlhV/wF4HLAeWAWsqaqDgDVtXpI0RIu31JlkB2AnYK/2zj2ta1fg4TPZYZJdgSOBlwFU1V3AXUlWAEe1xc4CLgD+fCb7kCTNzBZDAXgF8Bq6ALiEX4bCT4H3z3CfjwImgA8meVzb7inAPlW1EaCqNibZe4bblyTN0BYvH1XVu6vqAODUqnpUVR3Qbo+rqvfNcJ+Lgd8E/q6qHg/cwVZcKkpyUpJ1SdZNTEzMsARJ0lTu70wBgKp6b5KnAMsG16mqD81gnxuADVV1cZv/JF0o3JhkaTtLWArctJlaVgOrAZYvX14z2L8kaTOmFQpJPgwcCFwK3NOaC9jqUKiqHye5Psmjq+oHwNHAFe22Ejit3Z+7tduWJD0w0woFYDlwcFVtq3fmrwI+muRBwNXACXSXss5JciJwHfCCbbQvSdI0TTcUvgc8DNi4LXZaVZfSBc1kR2+L7UuSZma6obAXcEWSbwJ3bmqsqt+dlaokSSMx3VB442wWIUmaG6b730drZ7sQSdLoTfe/j26n+28jgAcB2wN3VNWus1WYJGn4pnumsMvgfJJjgSfOSkWSpJGZ0SipVfVZ4OnbuBZJ0ohN9/LR8wZmt6P7d1K/TSxJC8x0//voPw9M3w1cA6zY5tVIkkZqup8pnDDbhUiSRm+6P7KzX5LPJLkpyY1JPpVkv9kuTpI0XNP9oPmDwHl0v6uwL/C/W5skaQGZbigsqaoPVtXd7XYmsGQW65IkjcB0Q+HmJMcnWdRuxwO3zGZhkqThm24ovBx4IfBjupFSf49uuGtJ0gIy3X9JfTOwsqpuBUiyJ/BOurCQJC0Q0z1TeOymQACoqp8Aj5+dkiRJozLdUNguyR6bZtqZwnTPMiRJ88R0X9j/B/C1JJ+kG97ihcBbZ60qSdJITPcbzR9Kso5uELwAz6uqK2a1MknS0E37ElALAYNAkhawGQ2dLUlamAwFSVLPUJAk9QwFSVLPUJAk9QwFSVLPUJAk9QwFSVLPUJAk9QwFSVJvZKHQfsHtO0k+1+b3THJ+kivb/R73tw1J0rY1yjOFU4D1A/OrgDVVdRCwps1LkoZoJKGQZD/gGOB/DTSvAM5q02cBxw67Lkkad6M6U/hb4M+Aewfa9qmqjQDtfu9RFCZJ42zooZDkucBNVXXJDNc/Kcm6JOsmJia2cXWSNN5GcabwVOB3k1wDnA08PclHgBuTLAVo9zdNtXJVra6q5VW1fMmSJcOqWZLGwtBDoapeV1X7VdUy4DjgK1V1PHAesLItthI4d9i1SdK4m0vfUzgNeEaSK4FntHlJ0hBN++c4Z0NVXQBc0KZvAY4eZT2SNO7m0pmCJGnEDAVJUs9QkCT1DAVJUs9QkCT1DAVJUs9QkCT1DAVJUs9QkCT1DAVJUs9QkCT1DAVJUs9QkCT1DAVJUs9QkCT1DAVJUs9QkCT1DAVJUs9QkCT1DAVJUs9QkCT1DAVJUs9QkCT1DAVJUs9QkCT1DAVJUs9QkCT1DAVJUs9QkCT1DAVJUs9QkCT1hh4KSR6R5B+TrE9yeZJTWvueSc5PcmW732PYtUnSuBvFmcLdwGur6jeAJwOvTHIwsApYU1UHAWvavCRpiIYeClW1saq+3aZvB9YD+wIrgLPaYmcBxw67NkkadyP9TCHJMuDxwMXAPlW1EbrgAPbezDonJVmXZN3ExMSwSpWksTCyUEiyM/Ap4DVV9dPprldVq6tqeVUtX7JkyewVKEljaCShkGR7ukD4aFV9ujXfmGRp618K3DSK2iRpnI3iv48CfABYX1V/M9B1HrCyTa8Ezh12bZI07haPYJ9PBV4KfDfJpa3tL4DTgHOSnAhcB7xgBLVJ0lgbeihU1VeBbKb76GHWIkm6L7/RLEnqGQqSpJ6hIEnqGQqSpJ6hIEnqGQqSpJ6hIEnqGQqSpJ6hIEnqGQqSpJ6hIEnqGQqSpJ6hIEnqGQqSpJ6hIEnqGQqSpJ6hIEnqGQqSpJ6hIEnqGQqSpJ6hIEnqLR51AZIWrmWrPj/qEhasa047Zla265mCJKlnKEiSeoaCJKlnKEiSeoaCJKlnKEiSeoaCJKlnKEiSenMuFJI8O8kPklyVZNWo65GkcTKnQiHJIuD9wO8ABwMvTnLwaKuSpPExp0IBeCJwVVVdXVV3AWcDK0ZckySNjbk29tG+wPUD8xuAJw0ukOQk4KQ2+7MkPxhSbaO2F3DzqIuYrrxt1BXMCfPmmHm8gHl0vOABH7P9N9cx10IhU7TVfWaqVgOrh1PO3JFkXVUtH3Udmj6P2fzi8erMtctHG4BHDMzvB/xoRLVI0tiZa6HwLeCgJAckeRBwHHDeiGuSpLExpy4fVdXdSf4Y+BKwCDijqi4fcVlzxdhdMlsAPGbzi8cLSFXd/1KSpLEw1y4fSZJGyFCQJPUMhTkgyc9GXYM6SSrJhwfmFyeZSPK5rdzOBUmWt+kvJNl9W9eq6Zn895XkZUne16ZPTvL797N+v/w4mFMfNEtzwB3AIUl2rKqfA88AbnggG6yq52yTyrTNVdXpo65hrvFMYY5Ksn+SNUkua/ePTLIoydXp7J7k3iRHtuUvSvJro657gfg/wDFt+sXAxzd1JHlIkjOSfCvJd5KsaO07Jjm7Ha9PADsOrHNNkr2SLEvyvYH2U5O8sU1fkORdSS5Msj7JYUk+neTKJG8ZwmMeS0nemOTUNn1YO35fT/KOwWMFPDzJF9vxePuIyh0KQ2Hueh/woap6LPBR4D1VdQ/wL3SDBf4n4BLgiCQPBvarqqtGVu3CcjZwXJIdgMcCFw/0/VfgK1V1GPBbwDuSPAT4I+Df2vF6K/CEGez3rqo6EjgdOBd4JXAI8LIkD53xo9GOSS7ddAPetJnlPgicXFWHA/dM6jsUeBHwH4EXJXnE5JUXCkNh7joc+Fib/jBdCABcBBzZbn/d2g+j++KftoGqugxYRneW8IVJ3c8EVrUXlwuAHYBH0h2Pjwysf9kMdr3pi5rfBS6vqo1VdSdwNff9pr+2zs+r6tBNN+D1kxdon/nsUlVfa00fm7TImqq6rar+HbiCLYwdNN8ZCvPHpi+UXAQcQTei7BeA3YGjgAtHU9aCdR7wTgYuHTUBnj/wIvPIqlrf+u7vSz93c9+/uR0m9d/Z7u8dmN407+d/s2uqcdcGDR6Pe1jAx8NQmLu+RjfMB8BLgK+26YuBpwD3tnctlwKvoAsLbTtnAG+qqu9Oav8S8KokAUjy+NZ+Id1xIskhdJedJrsR2DvJQ9slv+fOSuXaalV1K3B7kie3puO2tPxCZijMDTsl2TBw+xPg1cAJSS4DXgqcAtAuJ1wPfKOtexGwC90lB20jVbWhqt49Rdebge2By9oHkW9u7X8H7NyO158B35xim7+gu559MfA54PuzUbtm7ERgdZKv05053DbiekbCYS4kCUiyc1X9rE2vApZW1SkjLmvoFux1MUnaSsckeR3d6+K1wMtGW85oeKYgSer5mYIkqWcoSJJ6hoIkqWcoaCxMHndI0tQMBel+JJkX/6U3X+rU3GYoaJwsSvL3SS5P8uU2sumhSb7RRsf8TJI9oB+19K+SrAVOSfKCJN9L8s9JLmzLLGqjaX6rrf+K1n5UG+30M0muSHJ6ku1a34uTfLdt622t7YVJ/qZNn5Lk6jZ9YJKvtuknJFmb5JIkX0qydKo6h/t0aiHynYXGyUHAi6vqD5OcAzyf7tvHr6qqtUneBLwBeE1bfveqehpAku8Cz6qqG/LLH8w5Ebitqg5rw1b8U5Ivt74n0o1mey3wReB5Sb4GvI1uBNVbgS8nOZZuiIw/besdAdySZF+6wQ4vSrI98F5gRVVNJHkR3UisL59cp/RAGQoaJz+sqkvb9CXAgXQvqGtb21nAPwws/4mB6X8Czmxh8unW9kzgsUl+r83vRhc8dwHfrKpN7/g/TvcC/wvggqqaaO0fBY6sqs8m2TnJLnSjoX6MbtTVI9q+Hk03hPb5bcilRcDGzdQpPSCGgsbJ5JEu7+8nMu/YNFFVJyd5Et2P71ya5FC68XFeVVVfGlwpyVH86oipxZZH4vw6cALwA7rxrF5ON3z6a+mG5r68jfO/xTqlB8rPFDTObgNuTXJEm38psHaqBZMcWFUXV9XrgZvp3tF/CfijdnmHJL/efnAH4IlJDmifJbyIbpTbi4GnpfsVtkV0v9ewaX8XAqe2++/Q/YDPnVV1G11QLElyeNvP9kkes+2eBumXPFPQuFsJnJ5kJ7ofszlhM8u9I8lBdO/21wD/TPdDOsuAb7ehtCeAY9vyXwdOo/ulrguBz1TVvW1snX9s2/lCVZ3blr+ILmgurKp7klxPG0W1qu5ql6jek2Q3ur/bvwUu30bPgdRz7CNpG2uXj06tKn8vQfOOl48kST3PFCRJPc8UJEk9Q0GS1DMUJEk9Q0GS1DMUJEm9/w93/l1NzJpNXQAAAABJRU5ErkJggg==\n",
"text/plain": [
"<Figure size 432x288 with 1 Axes>"
]
},
"metadata": {
"needs_background": "light"
},
"output_type": "display_data"
}
],
"source": [
"%matplotlib inline\n",
"import matplotlib as plt\n",
"from matplotlib import pyplot\n",
"pyplot.bar(group_names, df[\"horsepower-binned\"].value_counts())\n",
"\n",
"# set x/y labels and plot title\n",
"plt.pyplot.xlabel(\"horsepower\")\n",
"plt.pyplot.ylabel(\"count\")\n",
"plt.pyplot.title(\"horsepower bins\")"
]
},
{
"cell_type": "markdown",
"metadata": {},
"source": [
"<p>\n",
" Check the dataframe above carefully, you will find the last column provides the bins for \"horsepower\" with 3 categories (\"Low\",\"Medium\" and \"High\"). \n",
"</p>\n",
"<p>\n",
" We successfully narrow the intervals from 57 to 3!\n",
"</p>"
]
},
{
"cell_type": "markdown",
"metadata": {},
"source": [
"<h3>Bins visualization</h3>\n",
"Normally, a histogram is used to visualize the distribution of bins we created above. "
]
},
{
"cell_type": "code",
"execution_count": 111,
"metadata": {
"collapsed": false,
"jupyter": {
"outputs_hidden": false
}
},
"outputs": [
{
"data": {
"text/plain": [
"Text(0.5, 1.0, 'horsepower bins')"
]
},
"execution_count": 111,
"metadata": {},
"output_type": "execute_result"
},
{
"data": {
"image/png": "iVBORw0KGgoAAAANSUhEUgAAAYwAAAEWCAYAAAB1xKBvAAAABHNCSVQICAgIfAhkiAAAAAlwSFlzAAALEgAACxIB0t1+/AAAADh0RVh0U29mdHdhcmUAbWF0cGxvdGxpYiB2ZXJzaW9uMy4xLjEsIGh0dHA6Ly9tYXRwbG90bGliLm9yZy8QZhcZAAAWGUlEQVR4nO3de7hddX3n8feHcPHCvQkYwRpgqFN0FDVVqSMyYr1Pw9iq+BQntbToPF5wRtvGdqq2Uztora21Tn3ieInVitQbzDiPykQJ2lY0KCIQKYjhGklAVHRmUPA7f6zfqdt4TvLL4Zyz98l5v55nP3ut31prr+9vr+R89lp7799OVSFJ0u7sM+4CJEmLg4EhSepiYEiSuhgYkqQuBoYkqYuBIUnqYmBoIiXZmuTJ465j0iU5JclNu1j+9iR/sJA1ae+177gLkDR/qurF465Bew/PMLRXS7LXvCjam/qixcnA0CQ7McnlSb6T5INJ7jO1IMlvJbk2ybeSXJDkgSPLKslLklwDXJPBnyfZ3h7r8iQPa+sekORNSW5Icmu7hHPftuyUJDcl+b0kt7XLZL82sp9Dkrw3yY4k1yf5z0n2acuuT/LoNn1Gq+mENv+bST7WpvdJsi7J15PcnuS8JIe3ZavadmcmuQH49ExP1C5qfE+SP96pP69sz8W2JC8cWfcZSa5KcmeSm5O86t4cPO19DAxNsucCTwOOAR4O/DpAkicB/7UtXwlcD5y707anAY8FTgCeApwM/BxwKPA84Pa23hta+4nAvwCOAl4z8jgPAJa39rXA+iQPacveChwCHAs8Efj3wNQf4E3AKW36ZOC6ts7U/KY2/fJW6xOBBwJ3AG/bqS9PBH4eeOo0z9Huapxu3UPaumcCb0tyWFv2TuBFVXUQ8DB2EVBaoqrKm7eJuwFbgTNG5t8IvL1NvxN448iyA4EfAqvafAFPGln+JOCfgMcB+4y0B/g+cNxI20nAN9r0KcDdwP1Hlp8H/AGwDLgLOGFk2YuAi9r0mcAFbXoL8JvAuW3+euBRI8tOHXmMla0v+wKrWl+O3cXzNGONbfo9wB+PrPt/gX1H1t0OPK5N39D6cPC4j7+3ybx5hqFJ9s2R6f/DEAwwvBK/fmpBVX2P4YzhqJH1bxxZ/mngrxheud+aZH2Sg4EVwP2AS5N8O8m3gU+09il3VNX3R+avb/tfDuw/WkebnqphE/CEJA9gCJcPAo9PsorhFf5lbb0HAx8d2f8W4B7gyOn6MoOZapzO7VV198j86PP6K8AzgOuTbEpy0m72qyXGwNBidAvDH1oAktwf+Bng5pF1fmIY5qr6y6p6NPBQhktQvw3cxvCK+6FVdWi7HVJVB45selh7/Ck/2/Z/G8OZwIN3WnZz29+1DH+MXw5cXFV3MgTgWcDnqupHbZsbgaeP7P/QqrpPVc3Yl2nMVOMeqaovVtUa4AjgYwxnKtI/MzC0GP0t8MIkJyY5APgT4JKq2jrdykl+Icljk+zHcAnq/wH3tD/a7wD+PMkRbd2jkuz8XsEfJtk/yROAZwF/V1X3MPxBfX2Sg5I8GPhPwPtGttsEvJQfv19x0U7zAG9vj/Hgtv8VSdbM4jn5qRr3ZOO27a8lOaSqfgh8l+FMR/pnBoYWnarayPA+woeBbcBxwOm72ORghmC4g+Fyze3Am9qy3wWuBT6f5LvA/wZG3zD+ZtvuFuD9wIur6mtt2csYAug64HMMQfaukW03AQcBF88wD/AW4ALgU0nuBD7P8Gb9nthVjXviBcDW9jy8GDhjFo+hvViq/AElaTpJTgHeV1VHj7sWaRJ4hiFJ6mJgSJK6eElKktTFMwxJUpdFPZjZ8uXLa9WqVeMuQ5IWlUsvvfS2qlqx+zV/0qIOjFWrVrF58+ZxlyFJi0qS63e/1k/zkpQkqYuBIUnqYmBIkroYGJKkLgaGJKmLgSFJ6mJgSJK6GBiSpC4GhiSpy6L+pve9tWrdx8ddgnay9ZxnjrsESTPwDEOS1MXAkCR1MTAkSV0MDElSFwNDktTFwJAkdTEwJEld5i0wkrwryfYkV4y0HZ7kwiTXtPvDRpa9Osm1Sa5O8tT5qkuSNDvzeYbxHuBpO7WtAzZW1fHAxjZPkhOA04GHtm3+W5Jl81ibJGkPzVtgVNXFwLd2al4DbGjTG4DTRtrPraq7quobwLXAY+arNknSnlvo9zCOrKptAO3+iNZ+FHDjyHo3tbafkuSsJJuTbN6xY8e8FitJ+rFJedM707TVdCtW1fqqWl1Vq1esWDHPZUmSpix0YNyaZCVAu9/e2m8CHjSy3tHALQtcmyRpFxY6MC4A1rbptcD5I+2nJzkgyTHA8cAXFrg2SdIuzNvw5kk+AJwCLE9yE/Ba4BzgvCRnAjcAzwGoqiuTnAdcBdwNvKSq7pmv2iRJe27eAqOqnj/DolNnWP/1wOvnqx5J0r0zKW96S5ImnIEhSepiYEiSuhgYkqQuBoYkqYuBIUnqYmBIkroYGJKkLvP2xT1pNlat+/i4S9BOtp7zzHGXoAnhGYYkqYuBIUnqYmBIkroYGJKkLgaGJKmLgSFJ6mJgSJK6GBiSpC4GhiSpi4EhSepiYEiSuhgYkqQuBoYkqYuBIUnqYmBIkroYGJKkLgaGJKmLgSFJ6mJgSJK6GBiSpC4GhiSpy1gCI8l/THJlkiuSfCDJfZIcnuTCJNe0+8PGUZskaXoLHhhJjgJeDqyuqocBy4DTgXXAxqo6HtjY5iVJE2Jcl6T2Be6bZF/gfsAtwBpgQ1u+AThtTLVJkqax4IFRVTcDbwJuALYB36mqTwFHVtW2ts424IiFrk2SNLNxXJI6jOFs4hjggcD9k5yxB9uflWRzks07duyYrzIlSTsZxyWpJwPfqKodVfVD4CPALwK3JlkJ0O63T7dxVa2vqtVVtXrFihULVrQkLXXjCIwbgMcluV+SAKcCW4ALgLVtnbXA+WOoTZI0g30XeodVdUmSDwFfAu4GvgysBw4EzktyJkOoPGeha5MkzWzBAwOgql4LvHan5rsYzjYkSRPIb3pLkroYGJKkLgaGJKmLgSFJ6mJgSJK6GBiSpC4GhiSpi4EhSepiYEiSuhgYkqQuBoYkqYuBIUnqYmBIkroYGJKkLgaGJKmLgSFJ6mJgSJK6GBiSpC4GhiSpi4EhSepiYEiSuhgYkqQuBoYkqYuBIUnqYmBIkroYGJKkLgaGJKmLgSFJ6mJgSJK6GBiSpC5jCYwkhyb5UJKvJdmS5KQkhye5MMk17f6wcdQmSZpeV2Ak2djTtgfeAnyiqv4l8AhgC7AO2FhVxwMb27wkaULsu6uFSe4D3A9Y3l7xpy06GHjgbHaY5GDgZODXAarqB8APkqwBTmmrbQAuAn53NvuQJM29XQYG8CLgFQzhcCk/DozvAm+b5T6PBXYA707yiPa4ZwNHVtU2gKraluSIWT6+JGke7PKSVFW9paqOAV5VVcdW1THt9oiq+qtZ7nNf4FHAX1fVI4HvsweXn5KclWRzks07duyYZQmSpD21uzMMAKrqrUl+EVg1uk1VvXcW+7wJuKmqLmnzH2IIjFuTrGxnFyuB7TPUsh5YD7B69eqaxf4lSbPQFRhJ/gY4DrgMuKc1F7DHgVFV30xyY5KHVNXVwKnAVe22Fjin3Z+/p48tSZo/XYEBrAZOqKq5ekX/MuD9SfYHrgNeyHB57LwkZwI3AM+Zo31JkuZAb2BcATwA2DYXO62qyxhCaGenzsXjS5LmXm9gLAeuSvIF4K6pxqr65XmpSpI0cXoD43XzWYQkafL1fkpq03wXIkmabL2fkrqT4VNRAPsD+wHfr6qD56swSdJk6T3DOGh0PslpwGPmpSJJ0kSa1Wi1VfUx4ElzXIskaYL1XpJ69sjsPgwfifVb1pK0hPR+SurfjkzfDWwF1sx5NZKkidX7HsYL57sQSdJk6/0BpaOTfDTJ9iS3JvlwkqPnuzhJ0uTofdP73cAFDL+LcRTwP1qbJGmJ6A2MFVX17qq6u93eA6yYx7okSROmNzBuS3JGkmXtdgZw+3wWJkmaLL2B8RvAc4FvMoxY+6sMQ5JLkpaI3o/V/hdgbVXdAZDkcOBNDEEiSVoCes8wHj4VFgBV9S3gkfNTkiRpEvUGxj5JDpuaaWcYvWcnkqS9QO8f/T8D/iHJhxiGBHku8Pp5q0qSNHF6v+n93iSbGQYcDPDsqrpqXiuTJE2U7stKLSAMCUlaomY1vLkkaekxMCRJXQwMSVIXA0OS1MXAkCR1MTAkSV0MDElSFwNDktTFwJAkdTEwJEldxhYY7Zf7vpzkf7b5w5NcmOSadn/Y7h5DkrRwxnmGcTawZWR+HbCxqo4HNrZ5SdKEGEtgJDkaeCbw30ea1wAb2vQG4LSFrkuSNLNxnWH8BfA7wI9G2o6sqm0A7f6IcRQmSZreggdGkmcB26vq0lluf1aSzUk279ixY46rkyTNZBxnGI8HfjnJVuBc4ElJ3gfcmmQlQLvfPt3GVbW+qlZX1eoVK1YsVM2StOQteGBU1aur6uiqWgWcDny6qs4ALgDWttXWAucvdG2SpJlN0vcwzgF+Kck1wC+1eUnShOj+idb5UFUXARe16duBU8dZjyRpZpN0hiFJmmAGhiSpi4EhSepiYEiSuhgYkqQuBoYkqYuBIUnqYmBIkroYGJKkLgaGJKmLgSFJ6mJgSJK6GBiSpC4GhiSpi4EhSepiYEiSuhgYkqQuBoYkqYuBIUnqYmBIkroYGJKkLgaGJKmLgSFJ6mJgSJK6GBiSpC4GhiSpi4EhSepiYEiSuhgYkqQuBoYkqYuBIUnqsuCBkeRBST6TZEuSK5Oc3doPT3Jhkmva/WELXZskaWbjOMO4G3hlVf088DjgJUlOANYBG6vqeGBjm5ckTYgFD4yq2lZVX2rTdwJbgKOANcCGttoG4LSFrk2SNLOxvoeRZBXwSOAS4Miq2gZDqABHzLDNWUk2J9m8Y8eOhSpVkpa8sQVGkgOBDwOvqKrv9m5XVeuranVVrV6xYsX8FShJ+gljCYwk+zGExfur6iOt+dYkK9vylcD2cdQmSZreOD4lFeCdwJaqevPIoguAtW16LXD+QtcmSZrZvmPY5+OBFwBfTXJZa/s94BzgvCRnAjcAzxlDbZKkGSx4YFTV54DMsPjUhaxFktTPb3pLkroYGJKkLgaGJKmLgSFJ6mJgSJK6GBiSpC4GhiSpi4EhSepiYEiSuhgYkqQuBoYkqYuBIUnqYmBIkroYGJKkLgaGJKmLgSFJ6mJgSJK6GBiSpC4GhiSpi4EhSepiYEiSuuw77gIkTbZV6z4+7hI0ja3nPHPB9+kZhiSpi4EhSepiYEiSuhgYkqQuBoYkqYuBIUnqYmBIkroYGJKkLhMXGEmeluTqJNcmWTfueiRJg4kKjCTLgLcBTwdOAJ6f5ITxViVJggkLDOAxwLVVdV1V/QA4F1gz5pokSUzeWFJHATeOzN8EPHZ0hSRnAWe12e8lufpe7G85cNu92H7S7G39Afu0WNinBZY3zGqzqT49eDYbT1pgZJq2+omZqvXA+jnZWbK5qlbPxWNNgr2tP2CfFgv7tDjc2z5N2iWpm4AHjcwfDdwyplokSSMmLTC+CByf5Jgk+wOnAxeMuSZJEhN2Saqq7k7yUuCTwDLgXVV15Tzuck4ubU2Qva0/YJ8WC/u0ONyrPqWqdr+WJGnJm7RLUpKkCWVgSJK6LMnA2FuGH0myNclXk1yWZHNrOzzJhUmuafeHjbvOXUnyriTbk1wx0jZjH5K8uh23q5M8dTxV79oMfXpdkpvbsbosyTNGlk10n5I8KMlnkmxJcmWSs1v7oj1Ou+jTYj5O90nyhSRfaX36w9Y+d8epqpbUjeHN9K8DxwL7A18BThh3XbPsy1Zg+U5tbwTWtel1wBvGXedu+nAy8Cjgit31gWG4mK8ABwDHtOO4bNx96OzT64BXTbPuxPcJWAk8qk0fBPxTq3vRHqdd9GkxH6cAB7bp/YBLgMfN5XFaimcYe/vwI2uADW16A3DaGGvZraq6GPjWTs0z9WENcG5V3VVV3wCuZTieE2WGPs1k4vtUVduq6ktt+k5gC8OoDIv2OO2iTzNZDH2qqvpem92v3Yo5PE5LMTCmG35kV/9QJlkBn0pyaRsyBeDIqtoGw38K4IixVTd7M/VhsR+7lya5vF2ymrossKj6lGQV8EiGV697xXHaqU+wiI9TkmVJLgO2AxdW1Zwep6UYGLsdfmQReXxVPYphdN+XJDl53AXNs8V87P4aOA44EdgG/FlrXzR9SnIg8GHgFVX13V2tOk3bYunToj5OVXVPVZ3IMErGY5I8bBer73GflmJg7DXDj1TVLe1+O/BRhtPJW5OsBGj328dX4azN1IdFe+yq6tb2n/lHwDv48an/ouhTkv0Y/rC+v6o+0poX9XGark+L/ThNqapvAxcBT2MOj9NSDIy9YviRJPdPctDUNPAU4AqGvqxtq60Fzh9PhffKTH24ADg9yQFJjgGOB74whvr22NR/2ObfMRwrWAR9ShLgncCWqnrzyKJFe5xm6tMiP04rkhzapu8LPBn4GnN5nMb9zv6YPk3wDIZPRXwd+P1x1zPLPhzL8AmHrwBXTvUD+BlgI3BNuz983LXuph8fYDj1/yHDK54zd9UH4PfbcbsaePq469+DPv0N8FXg8vYfdeVi6RPwrxkuVVwOXNZuz1jMx2kXfVrMx+nhwJdb7VcAr2ntc3acHBpEktRlKV6SkiTNgoEhSepiYEiSuhgYkqQuBoYkqYuBoSUhyarR0WMl7TkDQ9qNJBP1U8YzWSx1avEyMLSULEvyjvZbAZ9Kct8kJyb5fBts7qNTg80luSjJnyTZBJyd5DlJrmi/NXBxW2dZkj9N8sW2/Yta+ylJLm6Pd1WStyfZpy17fobfMLkiyRta23OTvLlNn53kujZ9XJLPtelHJ9nUBpr85MhQDz9R58I+nVpqfEWipeR44PlV9VtJzgN+Bfgd4GVVtSnJHwGvBV7R1j+0qp4IkOSrwFOr6uap4RcYvsH9nar6hSQHAH+f5FNt2WMYfm/geuATwLOT/APwBuDRwB0MIw2fBlwM/Hbb7gnA7UmOYvg28mfbmEdvBdZU1Y4kzwNeD/zGznVK88nA0FLyjaq6rE1fyjAq6aFVtam1bQD+bmT9D45M/z3wnhY0U4PvPQV4eJJfbfOHMITSD4AvVNXUmcIHGP74/xC4qKp2tPb3AydX1ceSHNjGBnsQ8LcMP8L0hLavhwAPAy4chkBiGcPQI9PVKc0bA0NLyV0j0/cAh860YvP9qYmqenGSxwLPBC5LciLD8NAvq6pPjm6U5BR+epjoYvrhpKf8I/BChjF9Pstw9nAS8ErgZ4Erq+qk3dUpzSffw9BS9h3gjiRPaPMvADZNt2KS46rqkqp6DXAbw5nAJ4H/0C4ZkeTn2sjBMPwWwTHtvYvnAZ9j+IGeJyZZnmQZ8PyR/V0MvKrdfxn4N8BdVfUdhhBZkeSktp/9kjx07p4GqY9nGFrq1gJvT3I/4DqGV/nT+dMkxzOcJWxkGCX4cmAV8KU2XPYOfvzzl/8InAP8K4YQ+GhV/SjJq4HPtMf5X1U1NdT0ZxlC6OKquifJjQxDU1NVP2iXvf4yySEM/2//gmGUYmnBOFqtNMfaJalXVdWzxl2LNJe8JCVJ6uIZhiSpi2cYkqQuBoYkqYuBIUnqYmBIkroYGJKkLv8fU8mXe4yddCIAAAAASUVORK5CYII=\n",
"text/plain": [
"<Figure size 432x288 with 1 Axes>"
]
},
"metadata": {
"needs_background": "light"
},
"output_type": "display_data"
}
],
"source": [
"%matplotlib inline\n",
"import matplotlib as plt\n",
"from matplotlib import pyplot\n",
"\n",
"a = (0,1,2)\n",
"\n",
"# draw historgram of attribute \"horsepower\" with bins = 3\n",
"plt.pyplot.hist(df[\"horsepower\"], bins = 3)\n",
"\n",
"# set x/y labels and plot title\n",
"plt.pyplot.xlabel(\"horsepower\")\n",
"plt.pyplot.ylabel(\"count\")\n",
"plt.pyplot.title(\"horsepower bins\")"
]
},
{
"cell_type": "markdown",
"metadata": {},
"source": [
"The plot above shows the binning result for attribute \"horsepower\". "
]
},
{
"cell_type": "markdown",
"metadata": {},
"source": [
"<h2 id=\"indicator\">Indicator variable (or dummy variable)</h2>\n",
"<b>What is an indicator variable?</b>\n",
"<p>\n",
" An indicator variable (or dummy variable) is a numerical variable used to label categories. They are called 'dummies' because the numbers themselves don't have inherent meaning. \n",
"</p>\n",
"\n",
"<b>Why we use indicator variables?</b>\n",
"<p>\n",
" So we can use categorical variables for regression analysis in the later modules.\n",
"</p>\n",
"<b>Example</b>\n",
"<p>\n",
" We see the column \"fuel-type\" has two unique values, \"gas\" or \"diesel\". Regression doesn't understand words, only numbers. To use this attribute in regression analysis, we convert \"fuel-type\" into indicator variables.\n",
"</p>\n",
"\n",
"<p>\n",
" We will use the panda's method 'get_dummies' to assign numerical values to different categories of fuel type. \n",
"</p>"
]
},
{
"cell_type": "code",
"execution_count": 112,
"metadata": {
"collapsed": false,
"jupyter": {
"outputs_hidden": false
}
},
"outputs": [
{
"data": {
"text/plain": [
"Index(['symboling', 'normalized-losses', 'make', 'fuel-type', 'aspiration',\n",
" 'num-of-doors', 'body-style', 'drive-wheels', 'engine-location',\n",
" 'wheel-base', 'length', 'width', 'height', 'curb-weight', 'engine-type',\n",
" 'num-of-cylinders', 'engine-size', 'fuel-system', 'bore', 'stroke',\n",
" 'compression-ratio', 'horsepower', 'peak-rpm', 'city-mpg',\n",
" 'highway-L/100km', 'price', 'horsepower-binned'],\n",
" dtype='object')"
]
},
"execution_count": 112,
"metadata": {},
"output_type": "execute_result"
}
],
"source": [
"df.columns"
]
},
{
"cell_type": "markdown",
"metadata": {},
"source": [
"get indicator variables and assign it to data frame \"dummy_variable_1\" "
]
},
{
"cell_type": "code",
"execution_count": 113,
"metadata": {
"collapsed": false,
"jupyter": {
"outputs_hidden": false
}
},
"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>diesel</th>\n",
" <th>gas</th>\n",
" </tr>\n",
" </thead>\n",
" <tbody>\n",
" <tr>\n",
" <th>0</th>\n",
" <td>0</td>\n",
" <td>1</td>\n",
" </tr>\n",
" <tr>\n",
" <th>1</th>\n",
" <td>0</td>\n",
" <td>1</td>\n",
" </tr>\n",
" <tr>\n",
" <th>2</th>\n",
" <td>0</td>\n",
" <td>1</td>\n",
" </tr>\n",
" <tr>\n",
" <th>3</th>\n",
" <td>0</td>\n",
" <td>1</td>\n",
" </tr>\n",
" <tr>\n",
" <th>4</th>\n",
" <td>0</td>\n",
" <td>1</td>\n",
" </tr>\n",
" </tbody>\n",
"</table>\n",
"</div>"
],
"text/plain": [
" diesel gas\n",
"0 0 1\n",
"1 0 1\n",
"2 0 1\n",
"3 0 1\n",
"4 0 1"
]
},
"execution_count": 113,
"metadata": {},
"output_type": "execute_result"
}
],
"source": [
"dummy_variable_1 = pd.get_dummies(df[\"fuel-type\"])\n",
"dummy_variable_1.head()"
]
},
{
"cell_type": "markdown",
"metadata": {},
"source": [
"change column names for clarity "
]
},
{
"cell_type": "code",
"execution_count": 114,
"metadata": {
"collapsed": false,
"jupyter": {
"outputs_hidden": false
}
},
"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>diesel</th>\n",
" <th>gas</th>\n",
" </tr>\n",
" </thead>\n",
" <tbody>\n",
" <tr>\n",
" <th>0</th>\n",
" <td>0</td>\n",
" <td>1</td>\n",
" </tr>\n",
" <tr>\n",
" <th>1</th>\n",
" <td>0</td>\n",
" <td>1</td>\n",
" </tr>\n",
" <tr>\n",
" <th>2</th>\n",
" <td>0</td>\n",
" <td>1</td>\n",
" </tr>\n",
" <tr>\n",
" <th>3</th>\n",
" <td>0</td>\n",
" <td>1</td>\n",
" </tr>\n",
" <tr>\n",
" <th>4</th>\n",
" <td>0</td>\n",
" <td>1</td>\n",
" </tr>\n",
" </tbody>\n",
"</table>\n",
"</div>"
],
"text/plain": [
" diesel gas\n",
"0 0 1\n",
"1 0 1\n",
"2 0 1\n",
"3 0 1\n",
"4 0 1"
]
},
"execution_count": 114,
"metadata": {},
"output_type": "execute_result"
}
],
"source": [
"dummy_variable_1.rename(columns={'fuel-type-gas':'gas', 'fuel-type-diesel':'diesel'}, inplace=True)\n",
"dummy_variable_1.head()"
]
},
{
"cell_type": "markdown",
"metadata": {},
"source": [
"We now have the value 0 to represent \"gas\" and 1 to represent \"diesel\" in the column \"fuel-type\". We will now insert this column back into our original dataset. "
]
},
{
"cell_type": "code",
"execution_count": 122,
"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>symboling</th>\n",
" <th>normalized-losses</th>\n",
" <th>make</th>\n",
" <th>fuel-type</th>\n",
" <th>aspiration</th>\n",
" <th>num-of-doors</th>\n",
" <th>body-style</th>\n",
" <th>drive-wheels</th>\n",
" <th>engine-location</th>\n",
" <th>wheel-base</th>\n",
" <th>...</th>\n",
" <th>city-mpg</th>\n",
" <th>highway-L/100km</th>\n",
" <th>price</th>\n",
" <th>horsepower-binned</th>\n",
" <th>diesel</th>\n",
" <th>gas</th>\n",
" <th>diesel</th>\n",
" <th>gas</th>\n",
" <th>diesel</th>\n",
" <th>gas</th>\n",
" </tr>\n",
" </thead>\n",
" <tbody>\n",
" <tr>\n",
" <th>0</th>\n",
" <td>3</td>\n",
" <td>?</td>\n",
" <td>alfa-romero</td>\n",
" <td>gas</td>\n",
" <td>std</td>\n",
" <td>two</td>\n",
" <td>convertible</td>\n",
" <td>rwd</td>\n",
" <td>front</td>\n",
" <td>88.6</td>\n",
" <td>...</td>\n",
" <td>21</td>\n",
" <td>8.666667</td>\n",
" <td>13495</td>\n",
" <td>Medium</td>\n",
" <td>0</td>\n",
" <td>1</td>\n",
" <td>0</td>\n",
" <td>1</td>\n",
" <td>0</td>\n",
" <td>1</td>\n",
" </tr>\n",
" <tr>\n",
" <th>1</th>\n",
" <td>3</td>\n",
" <td>?</td>\n",
" <td>alfa-romero</td>\n",
" <td>gas</td>\n",
" <td>std</td>\n",
" <td>two</td>\n",
" <td>convertible</td>\n",
" <td>rwd</td>\n",
" <td>front</td>\n",
" <td>88.6</td>\n",
" <td>...</td>\n",
" <td>21</td>\n",
" <td>8.666667</td>\n",
" <td>16500</td>\n",
" <td>Medium</td>\n",
" <td>0</td>\n",
" <td>1</td>\n",
" <td>0</td>\n",
" <td>1</td>\n",
" <td>0</td>\n",
" <td>1</td>\n",
" </tr>\n",
" <tr>\n",
" <th>2</th>\n",
" <td>1</td>\n",
" <td>?</td>\n",
" <td>alfa-romero</td>\n",
" <td>gas</td>\n",
" <td>std</td>\n",
" <td>two</td>\n",
" <td>hatchback</td>\n",
" <td>rwd</td>\n",
" <td>front</td>\n",
" <td>94.5</td>\n",
" <td>...</td>\n",
" <td>19</td>\n",
" <td>9.000000</td>\n",
" <td>16500</td>\n",
" <td>Medium</td>\n",
" <td>0</td>\n",
" <td>1</td>\n",
" <td>0</td>\n",
" <td>1</td>\n",
" <td>0</td>\n",
" <td>1</td>\n",
" </tr>\n",
" <tr>\n",
" <th>3</th>\n",
" <td>2</td>\n",
" <td>164</td>\n",
" <td>audi</td>\n",
" <td>gas</td>\n",
" <td>std</td>\n",
" <td>four</td>\n",
" <td>sedan</td>\n",
" <td>fwd</td>\n",
" <td>front</td>\n",
" <td>99.8</td>\n",
" <td>...</td>\n",
" <td>24</td>\n",
" <td>7.800000</td>\n",
" <td>13950</td>\n",
" <td>Medium</td>\n",
" <td>0</td>\n",
" <td>1</td>\n",
" <td>0</td>\n",
" <td>1</td>\n",
" <td>0</td>\n",
" <td>1</td>\n",
" </tr>\n",
" <tr>\n",
" <th>4</th>\n",
" <td>2</td>\n",
" <td>164</td>\n",
" <td>audi</td>\n",
" <td>gas</td>\n",
" <td>std</td>\n",
" <td>four</td>\n",
" <td>sedan</td>\n",
" <td>4wd</td>\n",
" <td>front</td>\n",
" <td>99.4</td>\n",
" <td>...</td>\n",
" <td>18</td>\n",
" <td>10.636364</td>\n",
" <td>17450</td>\n",
" <td>Medium</td>\n",
" <td>0</td>\n",
" <td>1</td>\n",
" <td>0</td>\n",
" <td>1</td>\n",
" <td>0</td>\n",
" <td>1</td>\n",
" </tr>\n",
" <tr>\n",
" <th>...</th>\n",
" <td>...</td>\n",
" <td>...</td>\n",
" <td>...</td>\n",
" <td>...</td>\n",
" <td>...</td>\n",
" <td>...</td>\n",
" <td>...</td>\n",
" <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",
" <td>...</td>\n",
" <td>...</td>\n",
" <td>...</td>\n",
" <td>...</td>\n",
" <td>...</td>\n",
" </tr>\n",
" <tr>\n",
" <th>200</th>\n",
" <td>-1</td>\n",
" <td>95</td>\n",
" <td>volvo</td>\n",
" <td>gas</td>\n",
" <td>std</td>\n",
" <td>four</td>\n",
" <td>sedan</td>\n",
" <td>rwd</td>\n",
" <td>front</td>\n",
" <td>109.1</td>\n",
" <td>...</td>\n",
" <td>23</td>\n",
" <td>8.357143</td>\n",
" <td>16845</td>\n",
" <td>Medium</td>\n",
" <td>0</td>\n",
" <td>1</td>\n",
" <td>0</td>\n",
" <td>1</td>\n",
" <td>0</td>\n",
" <td>1</td>\n",
" </tr>\n",
" <tr>\n",
" <th>201</th>\n",
" <td>-1</td>\n",
" <td>95</td>\n",
" <td>volvo</td>\n",
" <td>gas</td>\n",
" <td>turbo</td>\n",
" <td>four</td>\n",
" <td>sedan</td>\n",
" <td>rwd</td>\n",
" <td>front</td>\n",
" <td>109.1</td>\n",
" <td>...</td>\n",
" <td>19</td>\n",
" <td>9.360000</td>\n",
" <td>19045</td>\n",
" <td>Medium</td>\n",
" <td>0</td>\n",
" <td>1</td>\n",
" <td>0</td>\n",
" <td>1</td>\n",
" <td>0</td>\n",
" <td>1</td>\n",
" </tr>\n",
" <tr>\n",
" <th>202</th>\n",
" <td>-1</td>\n",
" <td>95</td>\n",
" <td>volvo</td>\n",
" <td>gas</td>\n",
" <td>std</td>\n",
" <td>four</td>\n",
" <td>sedan</td>\n",
" <td>rwd</td>\n",
" <td>front</td>\n",
" <td>109.1</td>\n",
" <td>...</td>\n",
" <td>18</td>\n",
" <td>10.173913</td>\n",
" <td>21485</td>\n",
" <td>Medium</td>\n",
" <td>0</td>\n",
" <td>1</td>\n",
" <td>0</td>\n",
" <td>1</td>\n",
" <td>0</td>\n",
" <td>1</td>\n",
" </tr>\n",
" <tr>\n",
" <th>203</th>\n",
" <td>-1</td>\n",
" <td>95</td>\n",
" <td>volvo</td>\n",
" <td>diesel</td>\n",
" <td>turbo</td>\n",
" <td>four</td>\n",
" <td>sedan</td>\n",
" <td>rwd</td>\n",
" <td>front</td>\n",
" <td>109.1</td>\n",
" <td>...</td>\n",
" <td>26</td>\n",
" <td>8.666667</td>\n",
" <td>22470</td>\n",
" <td>Medium</td>\n",
" <td>1</td>\n",
" <td>0</td>\n",
" <td>1</td>\n",
" <td>0</td>\n",
" <td>1</td>\n",
" <td>0</td>\n",
" </tr>\n",
" <tr>\n",
" <th>204</th>\n",
" <td>-1</td>\n",
" <td>95</td>\n",
" <td>volvo</td>\n",
" <td>gas</td>\n",
" <td>turbo</td>\n",
" <td>four</td>\n",
" <td>sedan</td>\n",
" <td>rwd</td>\n",
" <td>front</td>\n",
" <td>109.1</td>\n",
" <td>...</td>\n",
" <td>19</td>\n",
" <td>9.360000</td>\n",
" <td>22625</td>\n",
" <td>Medium</td>\n",
" <td>0</td>\n",
" <td>1</td>\n",
" <td>0</td>\n",
" <td>1</td>\n",
" <td>0</td>\n",
" <td>1</td>\n",
" </tr>\n",
" </tbody>\n",
"</table>\n",
"<p>205 rows × 33 columns</p>\n",
"</div>"
],
"text/plain": [
" symboling normalized-losses make fuel-type aspiration \\\n",
"0 3 ? alfa-romero gas std \n",
"1 3 ? alfa-romero gas std \n",
"2 1 ? alfa-romero gas std \n",
"3 2 164 audi gas std \n",
"4 2 164 audi gas std \n",
".. ... ... ... ... ... \n",
"200 -1 95 volvo gas std \n",
"201 -1 95 volvo gas turbo \n",
"202 -1 95 volvo gas std \n",
"203 -1 95 volvo diesel turbo \n",
"204 -1 95 volvo gas turbo \n",
"\n",
" num-of-doors body-style drive-wheels engine-location wheel-base ... \\\n",
"0 two convertible rwd front 88.6 ... \n",
"1 two convertible rwd front 88.6 ... \n",
"2 two hatchback rwd front 94.5 ... \n",
"3 four sedan fwd front 99.8 ... \n",
"4 four sedan 4wd front 99.4 ... \n",
".. ... ... ... ... ... ... \n",
"200 four sedan rwd front 109.1 ... \n",
"201 four sedan rwd front 109.1 ... \n",
"202 four sedan rwd front 109.1 ... \n",
"203 four sedan rwd front 109.1 ... \n",
"204 four sedan rwd front 109.1 ... \n",
"\n",
" city-mpg highway-L/100km price horsepower-binned diesel gas diesel \\\n",
"0 21 8.666667 13495 Medium 0 1 0 \n",
"1 21 8.666667 16500 Medium 0 1 0 \n",
"2 19 9.000000 16500 Medium 0 1 0 \n",
"3 24 7.800000 13950 Medium 0 1 0 \n",
"4 18 10.636364 17450 Medium 0 1 0 \n",
".. ... ... ... ... ... .. ... \n",
"200 23 8.357143 16845 Medium 0 1 0 \n",
"201 19 9.360000 19045 Medium 0 1 0 \n",
"202 18 10.173913 21485 Medium 0 1 0 \n",
"203 26 8.666667 22470 Medium 1 0 1 \n",
"204 19 9.360000 22625 Medium 0 1 0 \n",
"\n",
" gas diesel gas \n",
"0 1 0 1 \n",
"1 1 0 1 \n",
"2 1 0 1 \n",
"3 1 0 1 \n",
"4 1 0 1 \n",
".. .. ... .. \n",
"200 1 0 1 \n",
"201 1 0 1 \n",
"202 1 0 1 \n",
"203 0 1 0 \n",
"204 1 0 1 \n",
"\n",
"[205 rows x 33 columns]"
]
},
"execution_count": 122,
"metadata": {},
"output_type": "execute_result"
}
],
"source": [
"# merge data frame \"df\" and \"dummy_variable_1\" \n",
"df = pd.concat([df, dummy_variable_1], axis=1)\n",
"df\n",
"# drop original column \"fuel-type\" from \"df\"\n",
"#df.drop(\"fuel-type-diesel\", axis = 1, inplace=True)"
]
},
{
"cell_type": "code",
"execution_count": 116,
"metadata": {
"collapsed": false,
"jupyter": {
"outputs_hidden": false
}
},
"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>symboling</th>\n",
" <th>normalized-losses</th>\n",
" <th>make</th>\n",
" <th>fuel-type</th>\n",
" <th>aspiration</th>\n",
" <th>num-of-doors</th>\n",
" <th>body-style</th>\n",
" <th>drive-wheels</th>\n",
" <th>engine-location</th>\n",
" <th>wheel-base</th>\n",
" <th>...</th>\n",
" <th>stroke</th>\n",
" <th>compression-ratio</th>\n",
" <th>horsepower</th>\n",
" <th>peak-rpm</th>\n",
" <th>city-mpg</th>\n",
" <th>highway-L/100km</th>\n",
" <th>price</th>\n",
" <th>horsepower-binned</th>\n",
" <th>diesel</th>\n",
" <th>gas</th>\n",
" </tr>\n",
" </thead>\n",
" <tbody>\n",
" <tr>\n",
" <th>0</th>\n",
" <td>3</td>\n",
" <td>?</td>\n",
" <td>alfa-romero</td>\n",
" <td>gas</td>\n",
" <td>std</td>\n",
" <td>two</td>\n",
" <td>convertible</td>\n",
" <td>rwd</td>\n",
" <td>front</td>\n",
" <td>88.6</td>\n",
" <td>...</td>\n",
" <td>2.68</td>\n",
" <td>9.0</td>\n",
" <td>111</td>\n",
" <td>5000</td>\n",
" <td>21</td>\n",
" <td>8.666667</td>\n",
" <td>13495</td>\n",
" <td>Medium</td>\n",
" <td>0</td>\n",
" <td>1</td>\n",
" </tr>\n",
" <tr>\n",
" <th>1</th>\n",
" <td>3</td>\n",
" <td>?</td>\n",
" <td>alfa-romero</td>\n",
" <td>gas</td>\n",
" <td>std</td>\n",
" <td>two</td>\n",
" <td>convertible</td>\n",
" <td>rwd</td>\n",
" <td>front</td>\n",
" <td>88.6</td>\n",
" <td>...</td>\n",
" <td>2.68</td>\n",
" <td>9.0</td>\n",
" <td>111</td>\n",
" <td>5000</td>\n",
" <td>21</td>\n",
" <td>8.666667</td>\n",
" <td>16500</td>\n",
" <td>Medium</td>\n",
" <td>0</td>\n",
" <td>1</td>\n",
" </tr>\n",
" <tr>\n",
" <th>2</th>\n",
" <td>1</td>\n",
" <td>?</td>\n",
" <td>alfa-romero</td>\n",
" <td>gas</td>\n",
" <td>std</td>\n",
" <td>two</td>\n",
" <td>hatchback</td>\n",
" <td>rwd</td>\n",
" <td>front</td>\n",
" <td>94.5</td>\n",
" <td>...</td>\n",
" <td>3.47</td>\n",
" <td>9.0</td>\n",
" <td>154</td>\n",
" <td>5000</td>\n",
" <td>19</td>\n",
" <td>9.000000</td>\n",
" <td>16500</td>\n",
" <td>Medium</td>\n",
" <td>0</td>\n",
" <td>1</td>\n",
" </tr>\n",
" <tr>\n",
" <th>3</th>\n",
" <td>2</td>\n",
" <td>164</td>\n",
" <td>audi</td>\n",
" <td>gas</td>\n",
" <td>std</td>\n",
" <td>four</td>\n",
" <td>sedan</td>\n",
" <td>fwd</td>\n",
" <td>front</td>\n",
" <td>99.8</td>\n",
" <td>...</td>\n",
" <td>3.40</td>\n",
" <td>10.0</td>\n",
" <td>102</td>\n",
" <td>5500</td>\n",
" <td>24</td>\n",
" <td>7.800000</td>\n",
" <td>13950</td>\n",
" <td>Medium</td>\n",
" <td>0</td>\n",
" <td>1</td>\n",
" </tr>\n",
" <tr>\n",
" <th>4</th>\n",
" <td>2</td>\n",
" <td>164</td>\n",
" <td>audi</td>\n",
" <td>gas</td>\n",
" <td>std</td>\n",
" <td>four</td>\n",
" <td>sedan</td>\n",
" <td>4wd</td>\n",
" <td>front</td>\n",
" <td>99.4</td>\n",
" <td>...</td>\n",
" <td>3.40</td>\n",
" <td>8.0</td>\n",
" <td>115</td>\n",
" <td>5500</td>\n",
" <td>18</td>\n",
" <td>10.636364</td>\n",
" <td>17450</td>\n",
" <td>Medium</td>\n",
" <td>0</td>\n",
" <td>1</td>\n",
" </tr>\n",
" </tbody>\n",
"</table>\n",
"<p>5 rows × 29 columns</p>\n",
"</div>"
],
"text/plain": [
" symboling normalized-losses make fuel-type aspiration num-of-doors \\\n",
"0 3 ? alfa-romero gas std two \n",
"1 3 ? alfa-romero gas std two \n",
"2 1 ? alfa-romero gas std two \n",
"3 2 164 audi gas std four \n",
"4 2 164 audi gas std four \n",
"\n",
" body-style drive-wheels engine-location wheel-base ... stroke \\\n",
"0 convertible rwd front 88.6 ... 2.68 \n",
"1 convertible rwd front 88.6 ... 2.68 \n",
"2 hatchback rwd front 94.5 ... 3.47 \n",
"3 sedan fwd front 99.8 ... 3.40 \n",
"4 sedan 4wd front 99.4 ... 3.40 \n",
"\n",
" compression-ratio horsepower peak-rpm city-mpg highway-L/100km price \\\n",
"0 9.0 111 5000 21 8.666667 13495 \n",
"1 9.0 111 5000 21 8.666667 16500 \n",
"2 9.0 154 5000 19 9.000000 16500 \n",
"3 10.0 102 5500 24 7.800000 13950 \n",
"4 8.0 115 5500 18 10.636364 17450 \n",
"\n",
" horsepower-binned diesel gas \n",
"0 Medium 0 1 \n",
"1 Medium 0 1 \n",
"2 Medium 0 1 \n",
"3 Medium 0 1 \n",
"4 Medium 0 1 \n",
"\n",
"[5 rows x 29 columns]"
]
},
"execution_count": 116,
"metadata": {},
"output_type": "execute_result"
}
],
"source": [
"df.head()"
]
},
{
"cell_type": "markdown",
"metadata": {},
"source": [
"The last two columns are now the indicator variable representation of the fuel-type variable. It's all 0s and 1s now."
]
},
{
"cell_type": "markdown",
"metadata": {},
"source": [
"<div class=\"alert alert-danger alertdanger\" style=\"margin-top: 20px\">\n",
"<h1> Question #4: </h1>\n",
"\n",
"<b>As above, create indicator variable to the column of \"aspiration\": \"std\" to 0, while \"turbo\" to 1.</b>\n",
"</div>"
]
},
{
"cell_type": "code",
"execution_count": 123,
"metadata": {
"collapsed": false,
"jupyter": {
"outputs_hidden": false
}
},
"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>std</th>\n",
" <th>turbo</th>\n",
" </tr>\n",
" </thead>\n",
" <tbody>\n",
" <tr>\n",
" <th>0</th>\n",
" <td>1</td>\n",
" <td>0</td>\n",
" </tr>\n",
" <tr>\n",
" <th>1</th>\n",
" <td>1</td>\n",
" <td>0</td>\n",
" </tr>\n",
" <tr>\n",
" <th>2</th>\n",
" <td>1</td>\n",
" <td>0</td>\n",
" </tr>\n",
" <tr>\n",
" <th>3</th>\n",
" <td>1</td>\n",
" <td>0</td>\n",
" </tr>\n",
" <tr>\n",
" <th>4</th>\n",
" <td>1</td>\n",
" <td>0</td>\n",
" </tr>\n",
" <tr>\n",
" <th>...</th>\n",
" <td>...</td>\n",
" <td>...</td>\n",
" </tr>\n",
" <tr>\n",
" <th>200</th>\n",
" <td>1</td>\n",
" <td>0</td>\n",
" </tr>\n",
" <tr>\n",
" <th>201</th>\n",
" <td>0</td>\n",
" <td>1</td>\n",
" </tr>\n",
" <tr>\n",
" <th>202</th>\n",
" <td>1</td>\n",
" <td>0</td>\n",
" </tr>\n",
" <tr>\n",
" <th>203</th>\n",
" <td>0</td>\n",
" <td>1</td>\n",
" </tr>\n",
" <tr>\n",
" <th>204</th>\n",
" <td>0</td>\n",
" <td>1</td>\n",
" </tr>\n",
" </tbody>\n",
"</table>\n",
"<p>205 rows × 2 columns</p>\n",
"</div>"
],
"text/plain": [
" std turbo\n",
"0 1 0\n",
"1 1 0\n",
"2 1 0\n",
"3 1 0\n",
"4 1 0\n",
".. ... ...\n",
"200 1 0\n",
"201 0 1\n",
"202 1 0\n",
"203 0 1\n",
"204 0 1\n",
"\n",
"[205 rows x 2 columns]"
]
},
"execution_count": 123,
"metadata": {},
"output_type": "execute_result"
}
],
"source": [
"# Write your code below and press Shift+Enter to execute \n",
"dummy_variable_2 = pd.get_dummies(df['aspiration'])\n",
"dummy_variable_2"
]
},
{
"cell_type": "code",
"execution_count": 124,
"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>aspiration-std</th>\n",
" <th>aspiration-turbo</th>\n",
" </tr>\n",
" </thead>\n",
" <tbody>\n",
" <tr>\n",
" <th>0</th>\n",
" <td>1</td>\n",
" <td>0</td>\n",
" </tr>\n",
" <tr>\n",
" <th>1</th>\n",
" <td>1</td>\n",
" <td>0</td>\n",
" </tr>\n",
" <tr>\n",
" <th>2</th>\n",
" <td>1</td>\n",
" <td>0</td>\n",
" </tr>\n",
" <tr>\n",
" <th>3</th>\n",
" <td>1</td>\n",
" <td>0</td>\n",
" </tr>\n",
" <tr>\n",
" <th>4</th>\n",
" <td>1</td>\n",
" <td>0</td>\n",
" </tr>\n",
" </tbody>\n",
"</table>\n",
"</div>"
],
"text/plain": [
" aspiration-std aspiration-turbo\n",
"0 1 0\n",
"1 1 0\n",
"2 1 0\n",
"3 1 0\n",
"4 1 0"
]
},
"execution_count": 124,
"metadata": {},
"output_type": "execute_result"
}
],
"source": [
"dummy_variable_2.rename(columns={'std':'aspiration-std', 'turbo': 'aspiration-turbo'}, inplace=True)\n",
"dummy_variable_2.head()"
]
},
{
"cell_type": "markdown",
"metadata": {},
"source": [
"Double-click <b>here</b> for the solution.\n",
"\n",
"<!-- The answer is below:\n",
"\n",
"# get indicator variables of aspiration and assign it to data frame \"dummy_variable_2\"\n",
"dummy_variable_2 = pd.get_dummies(df['aspiration'])\n",
"\n",
"# change column names for clarity\n",
"dummy_variable_2.rename(columns={'std':'aspiration-std', 'turbo': 'aspiration-turbo'}, inplace=True)\n",
"\n",
"# show first 5 instances of data frame \"dummy_variable_1\"\n",
"dummy_variable_2.head()\n",
"\n",
"-->"
]
},
{
"cell_type": "markdown",
"metadata": {},
"source": [
" <div class=\"alert alert-danger alertdanger\" style=\"margin-top: 20px\">\n",
"<h1> Question #5: </h1>\n",
"\n",
"<b>Merge the new dataframe to the original dataframe then drop the column 'aspiration'</b>\n",
"</div>"
]
},
{
"cell_type": "code",
"execution_count": 125,
"metadata": {
"collapsed": false,
"jupyter": {
"outputs_hidden": false
}
},
"outputs": [],
"source": [
"# Write your code below and press Shift+Enter to execute \n",
"df = pd.concat([df,dummy_variable_2] , axis=1)\n"
]
},
{
"cell_type": "code",
"execution_count": 126,
"metadata": {},
"outputs": [],
"source": [
"df.drop(\"aspiration\", axis=1, inplace=True)"
]
},
{
"cell_type": "code",
"execution_count": 127,
"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>symboling</th>\n",
" <th>normalized-losses</th>\n",
" <th>make</th>\n",
" <th>fuel-type</th>\n",
" <th>num-of-doors</th>\n",
" <th>body-style</th>\n",
" <th>drive-wheels</th>\n",
" <th>engine-location</th>\n",
" <th>wheel-base</th>\n",
" <th>length</th>\n",
" <th>...</th>\n",
" <th>price</th>\n",
" <th>horsepower-binned</th>\n",
" <th>diesel</th>\n",
" <th>gas</th>\n",
" <th>diesel</th>\n",
" <th>gas</th>\n",
" <th>diesel</th>\n",
" <th>gas</th>\n",
" <th>aspiration-std</th>\n",
" <th>aspiration-turbo</th>\n",
" </tr>\n",
" </thead>\n",
" <tbody>\n",
" <tr>\n",
" <th>0</th>\n",
" <td>3</td>\n",
" <td>?</td>\n",
" <td>alfa-romero</td>\n",
" <td>gas</td>\n",
" <td>two</td>\n",
" <td>convertible</td>\n",
" <td>rwd</td>\n",
" <td>front</td>\n",
" <td>88.6</td>\n",
" <td>168.8</td>\n",
" <td>...</td>\n",
" <td>13495</td>\n",
" <td>Medium</td>\n",
" <td>0</td>\n",
" <td>1</td>\n",
" <td>0</td>\n",
" <td>1</td>\n",
" <td>0</td>\n",
" <td>1</td>\n",
" <td>1</td>\n",
" <td>0</td>\n",
" </tr>\n",
" <tr>\n",
" <th>1</th>\n",
" <td>3</td>\n",
" <td>?</td>\n",
" <td>alfa-romero</td>\n",
" <td>gas</td>\n",
" <td>two</td>\n",
" <td>convertible</td>\n",
" <td>rwd</td>\n",
" <td>front</td>\n",
" <td>88.6</td>\n",
" <td>168.8</td>\n",
" <td>...</td>\n",
" <td>16500</td>\n",
" <td>Medium</td>\n",
" <td>0</td>\n",
" <td>1</td>\n",
" <td>0</td>\n",
" <td>1</td>\n",
" <td>0</td>\n",
" <td>1</td>\n",
" <td>1</td>\n",
" <td>0</td>\n",
" </tr>\n",
" <tr>\n",
" <th>2</th>\n",
" <td>1</td>\n",
" <td>?</td>\n",
" <td>alfa-romero</td>\n",
" <td>gas</td>\n",
" <td>two</td>\n",
" <td>hatchback</td>\n",
" <td>rwd</td>\n",
" <td>front</td>\n",
" <td>94.5</td>\n",
" <td>171.2</td>\n",
" <td>...</td>\n",
" <td>16500</td>\n",
" <td>Medium</td>\n",
" <td>0</td>\n",
" <td>1</td>\n",
" <td>0</td>\n",
" <td>1</td>\n",
" <td>0</td>\n",
" <td>1</td>\n",
" <td>1</td>\n",
" <td>0</td>\n",
" </tr>\n",
" <tr>\n",
" <th>3</th>\n",
" <td>2</td>\n",
" <td>164</td>\n",
" <td>audi</td>\n",
" <td>gas</td>\n",
" <td>four</td>\n",
" <td>sedan</td>\n",
" <td>fwd</td>\n",
" <td>front</td>\n",
" <td>99.8</td>\n",
" <td>176.6</td>\n",
" <td>...</td>\n",
" <td>13950</td>\n",
" <td>Medium</td>\n",
" <td>0</td>\n",
" <td>1</td>\n",
" <td>0</td>\n",
" <td>1</td>\n",
" <td>0</td>\n",
" <td>1</td>\n",
" <td>1</td>\n",
" <td>0</td>\n",
" </tr>\n",
" <tr>\n",
" <th>4</th>\n",
" <td>2</td>\n",
" <td>164</td>\n",
" <td>audi</td>\n",
" <td>gas</td>\n",
" <td>four</td>\n",
" <td>sedan</td>\n",
" <td>4wd</td>\n",
" <td>front</td>\n",
" <td>99.4</td>\n",
" <td>176.6</td>\n",
" <td>...</td>\n",
" <td>17450</td>\n",
" <td>Medium</td>\n",
" <td>0</td>\n",
" <td>1</td>\n",
" <td>0</td>\n",
" <td>1</td>\n",
" <td>0</td>\n",
" <td>1</td>\n",
" <td>1</td>\n",
" <td>0</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",
" <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",
" <td>...</td>\n",
" <td>...</td>\n",
" <td>...</td>\n",
" </tr>\n",
" <tr>\n",
" <th>200</th>\n",
" <td>-1</td>\n",
" <td>95</td>\n",
" <td>volvo</td>\n",
" <td>gas</td>\n",
" <td>four</td>\n",
" <td>sedan</td>\n",
" <td>rwd</td>\n",
" <td>front</td>\n",
" <td>109.1</td>\n",
" <td>188.8</td>\n",
" <td>...</td>\n",
" <td>16845</td>\n",
" <td>Medium</td>\n",
" <td>0</td>\n",
" <td>1</td>\n",
" <td>0</td>\n",
" <td>1</td>\n",
" <td>0</td>\n",
" <td>1</td>\n",
" <td>1</td>\n",
" <td>0</td>\n",
" </tr>\n",
" <tr>\n",
" <th>201</th>\n",
" <td>-1</td>\n",
" <td>95</td>\n",
" <td>volvo</td>\n",
" <td>gas</td>\n",
" <td>four</td>\n",
" <td>sedan</td>\n",
" <td>rwd</td>\n",
" <td>front</td>\n",
" <td>109.1</td>\n",
" <td>188.8</td>\n",
" <td>...</td>\n",
" <td>19045</td>\n",
" <td>Medium</td>\n",
" <td>0</td>\n",
" <td>1</td>\n",
" <td>0</td>\n",
" <td>1</td>\n",
" <td>0</td>\n",
" <td>1</td>\n",
" <td>0</td>\n",
" <td>1</td>\n",
" </tr>\n",
" <tr>\n",
" <th>202</th>\n",
" <td>-1</td>\n",
" <td>95</td>\n",
" <td>volvo</td>\n",
" <td>gas</td>\n",
" <td>four</td>\n",
" <td>sedan</td>\n",
" <td>rwd</td>\n",
" <td>front</td>\n",
" <td>109.1</td>\n",
" <td>188.8</td>\n",
" <td>...</td>\n",
" <td>21485</td>\n",
" <td>Medium</td>\n",
" <td>0</td>\n",
" <td>1</td>\n",
" <td>0</td>\n",
" <td>1</td>\n",
" <td>0</td>\n",
" <td>1</td>\n",
" <td>1</td>\n",
" <td>0</td>\n",
" </tr>\n",
" <tr>\n",
" <th>203</th>\n",
" <td>-1</td>\n",
" <td>95</td>\n",
" <td>volvo</td>\n",
" <td>diesel</td>\n",
" <td>four</td>\n",
" <td>sedan</td>\n",
" <td>rwd</td>\n",
" <td>front</td>\n",
" <td>109.1</td>\n",
" <td>188.8</td>\n",
" <td>...</td>\n",
" <td>22470</td>\n",
" <td>Medium</td>\n",
" <td>1</td>\n",
" <td>0</td>\n",
" <td>1</td>\n",
" <td>0</td>\n",
" <td>1</td>\n",
" <td>0</td>\n",
" <td>0</td>\n",
" <td>1</td>\n",
" </tr>\n",
" <tr>\n",
" <th>204</th>\n",
" <td>-1</td>\n",
" <td>95</td>\n",
" <td>volvo</td>\n",
" <td>gas</td>\n",
" <td>four</td>\n",
" <td>sedan</td>\n",
" <td>rwd</td>\n",
" <td>front</td>\n",
" <td>109.1</td>\n",
" <td>188.8</td>\n",
" <td>...</td>\n",
" <td>22625</td>\n",
" <td>Medium</td>\n",
" <td>0</td>\n",
" <td>1</td>\n",
" <td>0</td>\n",
" <td>1</td>\n",
" <td>0</td>\n",
" <td>1</td>\n",
" <td>0</td>\n",
" <td>1</td>\n",
" </tr>\n",
" </tbody>\n",
"</table>\n",
"<p>205 rows × 34 columns</p>\n",
"</div>"
],
"text/plain": [
" symboling normalized-losses make fuel-type num-of-doors \\\n",
"0 3 ? alfa-romero gas two \n",
"1 3 ? alfa-romero gas two \n",
"2 1 ? alfa-romero gas two \n",
"3 2 164 audi gas four \n",
"4 2 164 audi gas four \n",
".. ... ... ... ... ... \n",
"200 -1 95 volvo gas four \n",
"201 -1 95 volvo gas four \n",
"202 -1 95 volvo gas four \n",
"203 -1 95 volvo diesel four \n",
"204 -1 95 volvo gas four \n",
"\n",
" body-style drive-wheels engine-location wheel-base length ... price \\\n",
"0 convertible rwd front 88.6 168.8 ... 13495 \n",
"1 convertible rwd front 88.6 168.8 ... 16500 \n",
"2 hatchback rwd front 94.5 171.2 ... 16500 \n",
"3 sedan fwd front 99.8 176.6 ... 13950 \n",
"4 sedan 4wd front 99.4 176.6 ... 17450 \n",
".. ... ... ... ... ... ... ... \n",
"200 sedan rwd front 109.1 188.8 ... 16845 \n",
"201 sedan rwd front 109.1 188.8 ... 19045 \n",
"202 sedan rwd front 109.1 188.8 ... 21485 \n",
"203 sedan rwd front 109.1 188.8 ... 22470 \n",
"204 sedan rwd front 109.1 188.8 ... 22625 \n",
"\n",
" horsepower-binned diesel gas diesel gas diesel gas aspiration-std \\\n",
"0 Medium 0 1 0 1 0 1 1 \n",
"1 Medium 0 1 0 1 0 1 1 \n",
"2 Medium 0 1 0 1 0 1 1 \n",
"3 Medium 0 1 0 1 0 1 1 \n",
"4 Medium 0 1 0 1 0 1 1 \n",
".. ... ... .. ... .. ... .. ... \n",
"200 Medium 0 1 0 1 0 1 1 \n",
"201 Medium 0 1 0 1 0 1 0 \n",
"202 Medium 0 1 0 1 0 1 1 \n",
"203 Medium 1 0 1 0 1 0 0 \n",
"204 Medium 0 1 0 1 0 1 0 \n",
"\n",
" aspiration-turbo \n",
"0 0 \n",
"1 0 \n",
"2 0 \n",
"3 0 \n",
"4 0 \n",
".. ... \n",
"200 0 \n",
"201 1 \n",
"202 0 \n",
"203 1 \n",
"204 1 \n",
"\n",
"[205 rows x 34 columns]"
]
},
"execution_count": 127,
"metadata": {},
"output_type": "execute_result"
}
],
"source": [
"df"
]
},
{
"cell_type": "markdown",
"metadata": {},
"source": [
"Double-click <b>here</b> for the solution.\n",
"\n",
"<!-- The answer is below:\n",
"\n",
"#merge the new dataframe to the original datafram\n",
"df = pd.concat([df, dummy_variable_2], axis=1)\n",
"\n",
"# drop original column \"aspiration\" from \"df\"\n",
"df.drop('aspiration', axis = 1, inplace=True)\n",
"\n",
"-->"
]
},
{
"cell_type": "markdown",
"metadata": {},
"source": [
"save the new csv "
]
},
{
"cell_type": "code",
"execution_count": 128,
"metadata": {},
"outputs": [],
"source": [
"df.to_csv('clean_df.csv')"
]
},
{
"cell_type": "markdown",
"metadata": {},
"source": [
"<h1>Thank you for completing this notebook</h1>"
]
},
{
"cell_type": "markdown",
"metadata": {},
"source": [
"<div class=\"alert alert-block alert-info\" style=\"margin-top: 20px\">\n",
"\n",
" <p><a href=\"https://cocl.us/DA0101EN_NotbookLink_Top_bottom\"><img src=\"https://s3-api.us-geo.objectstorage.softlayer.net/cf-courses-data/CognitiveClass/DA0101EN/Images/BottomAd.png\" width=\"750\" align=\"center\"></a></p>\n",
"</div>"
]
},
{
"cell_type": "markdown",
"metadata": {},
"source": [
"<h3>About the Authors:</h3>\n",
"\n",
"This notebook was written by <a href=\"https://www.linkedin.com/in/mahdi-noorian-58219234/\" target=\"_blank\">Mahdi Noorian PhD</a>, <a href=\"https://www.linkedin.com/in/joseph-s-50398b136/\" target=\"_blank\">Joseph Santarcangelo</a>, Bahare Talayian, Eric Xiao, Steven Dong, Parizad, Hima Vsudevan and <a href=\"https://www.linkedin.com/in/fiorellawever/\" target=\"_blank\">Fiorella Wenver</a> and <a href=\" https://www.linkedin.com/in/yi-leng-yao-84451275/ \" target=\"_blank\" >Yi Yao</a>.\n",
"\n",
"<p><a href=\"https://www.linkedin.com/in/joseph-s-50398b136/\" target=\"_blank\">Joseph Santarcangelo</a> is a Data Scientist at IBM, and holds a PhD in Electrical Engineering. His research focused on using Machine Learning, Signal Processing, and Computer Vision to determine how videos impact human cognition. Joseph has been working for IBM since he completed his PhD.</p>"
]
},
{
"cell_type": "markdown",
"metadata": {},
"source": [
"<hr>\n",
"<p>Copyright &copy; 2018 IBM Developer Skills Network. This notebook and its source code are released under the terms of the <a href=\"https://cognitiveclass.ai/mit-license/\">MIT License</a>.</p>"
]
}
],
"metadata": {
"anaconda-cloud": {},
"kernelspec": {
"display_name": "Python",
"language": "python",
"name": "conda-env-python-py"
},
"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.7"
}
},
"nbformat": 4,
"nbformat_minor": 4
}
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment