1/4/2020 data-wrangling - Jupyter Notebook
(https://cocl.us/corsera_da0101en_notebook_top)
(https://www.bigdatauniversity.com)
Data Analysis with Python
Data Wrangling
Welcome!
By the end of this notebook, you will have learned the basics of Data Wrangling!
Table of content
Identify and handle missing values
Identify missing values
Deal with missing values
Correct data format
localhost:8888/notebooks/Downloads/data-wrangling.ipynb 1/14
1/4/2020 data-wrangling - Jupyter Notebook
Data standardization
Data Normalization (centering/scaling)
Binning
Indicator variable
Estimated Time Needed: 30 min
What is the purpose of Data Wrangling?
Data Wrangling is the process of converting data from the initial format to a format that may be better for
analysis.
What is the fuel consumption (L/100k) rate for the diesel car?
Import data
You can find the "Automobile Data Set" from the following link: 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). We will be using this data set throughout this course.
Import pandas
In [ ]:
import pandas as pd
import matplotlib.pylab as plt
Reading the data set from the URL and adding the related
headers.
URL of the dataset
This dataset was hosted on IBM Cloud object click HERE (https://cocl.us/corsera_da0101en_notebook_bottom)
for free storage
Python list headers containing name of headers
In [ ]:
headers = ["symboling","normalized-losses","make","fuel-type","aspiration", "num-of-doors",
"drive-wheels","engine-location","wheel-base", "length","width","height","curb-wei
"num-of-cylinders", "engine-size","fuel-system","bore","stroke","compression-ratio
"peak-rpm","city-mpg","highway-mpg","price"]
Use the Pandas method read_csv() to load the data from the web address. Set the parameter "names" equal
localhost:8888/notebooks/Downloads/data-wrangling.ipynb 2/14
1/4/2020 data-wrangling - Jupyter Notebook
to the Python list "headers".
In [ ]:
df = pd.read_csv(filename, names = headers)
Use the method head() to display the first five rows of the dataframe.
In [ ]:
# To see what the data set looks like, we'll use the head() method.
df.head()
As we can see, several question marks appeared in the dataframe; those are missing values which may hinder
our further analysis.
So, how do we identify all those missing values and deal with them?
How to work with missing data?
Steps for working with missing data:
1. dentify missing data
2. deal with missing data
3. correct data format
Identify and handle missing values
Identify missing values
Convert "?" to NaN
In the car dataset, missing data comes with the question mark "?". 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:
.replace(A, B, inplace = True)
to replace A by B
In [ ]:
import numpy as np
# replace "?" to NaN
df.replace("?", np.nan, inplace = True)
df.head(5)
dentify_missing_values
Evaluating for Missing Data
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:
localhost:8888/notebooks/Downloads/data-wrangling.ipynb 3/14
1/4/2020 data-wrangling - Jupyter Notebook
1. .isnull()
2. .notnull()
The output is a boolean value indicating whether the value that is passed into the argument is in fact missing
data.
In [ ]:
missing_data = df.isnull()
missing_data.head(5)
"True" stands for missing value, while "False" stands for not missing value.
Count missing values in each column
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.
In [ ]:
for column in missing_data.columns.values.tolist():
print(column)
print (missing_data[column].value_counts())
print("")
Based on the summary above, each column has 205 rows of data, seven columns containing missing data:
1. "normalized-losses": 41 missing data
2. "num-of-doors": 2 missing data
3. "bore": 4 missing data
4. "stroke" : 4 missing data
5. "horsepower": 2 missing data
6. "peak-rpm": 2 missing data
7. "price": 4 missing data
Deal with missing data
How to deal with missing data?
1. drop data
a. drop the whole row
b. drop the whole column
2. replace data
a. replace it by mean
b. replace it by frequency
c. replace it based on other functions
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. 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:
localhost:8888/notebooks/Downloads/data-wrangling.ipynb 4/14
1/4/2020 data-wrangling - Jupyter Notebook
Replace by mean:
"normalized-losses": 41 missing data, replace them with mean
"stroke": 4 missing data, replace them with mean
"bore": 4 missing data, replace them with mean
"horsepower": 2 missing data, replace them with mean
"peak-rpm": 2 missing data, replace them with mean
Replace by frequency:
"num-of-doors": 2 missing data, replace them with "four".
Reason: 84% sedans is four doors. Since four doors is most frequent, it is most likely to occur
Drop the whole row:
"price": 4 missing data, simply delete the whole row
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
Calculate the average of the column
In [ ]:
avg_norm_loss = df["normalized-losses"].astype("float").mean(axis=0)
print("Average of normalized-losses:", avg_norm_loss)
Replace "NaN" by mean value in "normalized-losses" column
In [ ]:
df["normalized-losses"].replace(np.nan, avg_norm_loss, inplace=True)
Calculate the mean value for 'bore' column
In [ ]:
avg_bore=df['bore'].astype('float').mean(axis=0)
print("Average of bore:", avg_bore)
Replace NaN by mean value
In [ ]:
df["bore"].replace(np.nan, avg_bore, inplace=True)
Question #1:
According to the example above, replace NaN in "stroke" column by mean.
localhost:8888/notebooks/Downloads/data-wrangling.ipynb 5/14
1/4/2020 data-wrangling - Jupyter Notebook
In [ ]:
# Write your code below and press Shift+Enter to execute
Double-click here for the solution.
Calculate the mean value for the 'horsepower' column:
In [ ]:
avg_horsepower = df['horsepower'].astype('float').mean(axis=0)
print("Average horsepower:", avg_horsepower)
Replace "NaN" by mean value:
In [ ]:
df['horsepower'].replace(np.nan, avg_horsepower, inplace=True)
Calculate the mean value for 'peak-rpm' column:
In [ ]:
avg_peakrpm=df['peak-rpm'].astype('float').mean(axis=0)
print("Average peak rpm:", avg_peakrpm)
Replace NaN by mean value:
In [ ]:
df['peak-rpm'].replace(np.nan, avg_peakrpm, inplace=True)
To see which values are present in a particular column, we can use the ".value_counts()" method:
In [ ]:
df['num-of-doors'].value_counts()
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:
In [ ]:
df['num-of-doors'].value_counts().idxmax()
The replacement procedure is very similar to what we have seen previously
localhost:8888/notebooks/Downloads/data-wrangling.ipynb 6/14
1/4/2020 data-wrangling - Jupyter Notebook
In [ ]:
#replace the missing 'num-of-doors' values by the most frequent
df["num-of-doors"].replace(np.nan, "four", inplace=True)
Finally, let's drop all rows that do not have price data:
In [ ]:
# simply drop whole row with NaN in "price" column
df.dropna(subset=["price"], axis=0, inplace=True)
# reset index, because we droped two rows
df.reset_index(drop=True, inplace=True)
In [ ]:
df.head()
Good! Now, we obtain the dataset with no missing values.
Correct data format
We are almost there!
The last step in data cleaning is checking and making sure that all data is in the correct format (int, float, text or
other).
In Pandas, we use
.dtype() to check the data type
.astype() to change the data type
Lets list the data types for each column
In [ ]:
df.dtypes
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.
Convert data types to proper format
localhost:8888/notebooks/Downloads/data-wrangling.ipynb 7/14
1/4/2020 data-wrangling - Jupyter Notebook
In [ ]:
df[["bore", "stroke"]] = df[["bore", "stroke"]].astype("float")
df[["normalized-losses"]] = df[["normalized-losses"]].astype("int")
df[["price"]] = df[["price"]].astype("float")
df[["peak-rpm"]] = df[["peak-rpm"]].astype("float")
Let us list the columns after the conversion
In [ ]:
df.dtypes
Wonderful!
Now, we finally obtain the cleaned dataset with no missing values and all data in its proper format.
Data Standardization
Data is usually collected from different agencies with different formats. (Data Standardization is also a term for a
particular type of data normalization, where we subtract the mean and divide by the standard deviation)
What is Standardization?
Standardization is the process of transforming data into a common format which allows the researcher to make
the meaningful comparison.
Example
Transform mpg to L/100km:
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
We will need to apply data transformation to transform mpg into L/100km?
The formula for unit conversion is
L/100km = 235 / mpg
We can do many mathematical operations directly in Pandas.
In [ ]:
df.head()
In [ ]:
# Convert mpg to L/100km by mathematical operation (235 divided by mpg)
df['city-L/100km'] = 235/df["city-mpg"]
# check your transformed data
df.head()
localhost:8888/notebooks/Downloads/data-wrangling.ipynb 8/14
1/4/2020 data-wrangling - Jupyter Notebook
Question #2:
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".
In [ ]:
# Write your code below and press Shift+Enter to execute
Double-click here for the solution.
Data Normalization
Why normalization?
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
Example
To demonstrate normalization, let's say we want to scale the columns "length", "width" and "height"
Target:would like to Normalize those variables so their value ranges from 0 to 1.
Approach: replace original value by (original value)/(maximum value)
In [ ]:
# replace (original value) by (original value)/(maximum value)
df['length'] = df['length']/df['length'].max()
df['width'] = df['width']/df['width'].max()
Questiont #3:
According to the example above, normalize the column "height".
In [ ]:
# Write your code below and press Shift+Enter to execute
Double-click here for the solution.
Here we can see, we've normalized "length", "width" and "height" in the range of [0,1].
localhost:8888/notebooks/Downloads/data-wrangling.ipynb 9/14
1/4/2020 data-wrangling - Jupyter Notebook
Binning
Why binning?
Binning is a process of transforming continuous numerical variables into discrete categorical 'bins', for grouped
analysis.
Example:
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?
We will use the Pandas method 'cut' to segment the 'horsepower' column into 3 bins
Example of Binning Data In Pandas
Convert data to correct format
In [ ]:
df["horsepower"]=df["horsepower"].astype(int, copy=True)
Lets plot the histogram of horspower, to see what the distribution of horsepower looks like.
In [ ]:
%matplotlib inline
import matplotlib as plt
from matplotlib import pyplot
plt.pyplot.hist(df["horsepower"])
# set x/y labels and plot title
plt.pyplot.xlabel("horsepower")
plt.pyplot.ylabel("count")
plt.pyplot.title("horsepower bins")
We would like 3 bins of equal size bandwidth so we use numpy's linspace(start_value, end_value,
numbers_generated function.
Since we want to include the minimum value of horsepower we want to set start_value=min(df["horsepower"]).
Since we want to include the maximum value of horsepower we want to set end_value=max(df["horsepower"]).
Since we are building 3 bins of equal length, there should be 4 dividers, so numbers_generated=4.
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.
In [ ]:
bins = np.linspace(min(df["horsepower"]), max(df["horsepower"]), 4)
bins
localhost:8888/notebooks/Downloads/data-wrangling.ipynb 10/14
1/4/2020 data-wrangling - Jupyter Notebook
We set group names:
In [ ]:
group_names = ['Low', 'Medium', 'High']
We apply the function "cut" the determine what each value of "df['horsepower']" belongs to.
In [ ]:
df['horsepower-binned'] = pd.cut(df['horsepower'], bins, labels=group_names, include_lowest
df[['horsepower','horsepower-binned']].head(20)
Lets see the number of vehicles in each bin.
In [ ]:
df["horsepower-binned"].value_counts()
Lets plot the distribution of each bin.
In [ ]:
%matplotlib inline
import matplotlib as plt
from matplotlib import pyplot
pyplot.bar(group_names, df["horsepower-binned"].value_counts())
# set x/y labels and plot title
plt.pyplot.xlabel("horsepower")
plt.pyplot.ylabel("count")
plt.pyplot.title("horsepower bins")
Check the dataframe above carefully, you will find the last column provides the bins for "horsepower" with 3
categories ("Low","Medium" and "High").
We successfully narrow the intervals from 57 to 3!
Bins visualization
Normally, a histogram is used to visualize the distribution of bins we created above.
localhost:8888/notebooks/Downloads/data-wrangling.ipynb 11/14
1/4/2020 data-wrangling - Jupyter Notebook
In [ ]:
%matplotlib inline
import matplotlib as plt
from matplotlib import pyplot
a = (0,1,2)
# draw historgram of attribute "horsepower" with bins = 3
plt.pyplot.hist(df["horsepower"], bins = 3)
# set x/y labels and plot title
plt.pyplot.xlabel("horsepower")
plt.pyplot.ylabel("count")
plt.pyplot.title("horsepower bins")
The plot above shows the binning result for attribute "horsepower".
Indicator variable (or dummy variable)
What is an indicator variable?
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.
Why we use indicator variables?
So we can use categorical variables for regression analysis in the later modules.
Example
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.
We will use the panda's method 'get_dummies' to assign numerical values to different categories of fuel type.
In [ ]:
df.columns
get indicator variables and assign it to data frame "dummy_variable_1"
In [ ]:
dummy_variable_1 = pd.get_dummies(df["fuel-type"])
dummy_variable_1.head()
change column names for clarity
In [ ]:
dummy_variable_1.rename(columns={'fuel-type-diesel':'gas', 'fuel-type-diesel':'diesel'}, in
dummy_variable_1.head()
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.
localhost:8888/notebooks/Downloads/data-wrangling.ipynb 12/14
1/4/2020 data-wrangling - Jupyter Notebook
In [ ]:
# merge data frame "df" and "dummy_variable_1"
df = pd.concat([df, dummy_variable_1], axis=1)
# drop original column "fuel-type" from "df"
df.drop("fuel-type", axis = 1, inplace=True)
In [ ]:
df.head()
The last two columns are now the indicator variable representation of the fuel-type variable. It's all 0s and 1s
now.
Question #4:
As above, create indicator variable to the column of "aspiration": "std" to 0, while "turbo" to 1.
In [ ]:
# Write your code below and press Shift+Enter to execute
Double-click here for the solution.
Question #5:
Merge the new dataframe to the original dataframe then drop the column 'aspiration'
In [ ]:
# Write your code below and press Shift+Enter to execute
Double-click here for the solution.
save the new csv
In [ ]:
df.to_csv('clean_df.csv')
Thank you for completing this notebook
localhost:8888/notebooks/Downloads/data-wrangling.ipynb 13/14
1/4/2020 data-wrangling - Jupyter Notebook
<p><a href="https://cocl.us/corsera_da0101en_notebook_bottom"><img src="http
s://s3-api.us-geo.objectstorage.softlayer.net/cf-courses-data/CognitiveClass/D
A0101EN/Images/BottomAd.png" width="750" align="center"></a></p>
About the Authors:
This notebook was written by Mahdi Noorian PhD (https://www.linkedin.com/in/mahdi-noorian-58219234/),
Joseph Santarcangelo (https://www.linkedin.com/in/joseph-s-50398b136/), Bahare Talayian, Eric Xiao, Steven
Dong, Parizad, Hima Vsudevan and Fiorella Wenver (https://www.linkedin.com/in/fiorellawever/) and Yi Yao.
Joseph Santarcangelo (https://www.linkedin.com/in/joseph-s-50398b136/) 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.
Copyright © 2018 IBM Developer Skills Network. This notebook and its source code are released under the
terms of the MIT License (https://cognitiveclass.ai/mit-license/).
localhost:8888/notebooks/Downloads/data-wrangling.ipynb 14/14
