In this lab, you'll identify multicollinearity in the Ames Housing dataset.
You will be able to:
- Create a scatter matrix and correlation matrix
- Assess and interpret the output of a correlation matrix
- Identify if variables are exhibiting collinearity
- Decide how to address the collinearity in the data set
Let's reimport the Ames Housing data assign the numeric variables we want to keep to numeric_vars
.
import pandas as pd
import numpy as np
import matplotlib.pyplot as plt
%matplotlib inline
pd.options.display.max_columns = 999
pd.options.display.max_rows = 999
ames = pd.read_csv('ames.csv')
numeric_vars = ['LotFrontage', 'LotArea', 'BsmtFinSF1', 'BsmtFinSF2', 'BsmtUnfSF',
'TotalBsmtSF', '1stFlrSF', '2ndFlrSF','LowQualFinSF', 'GrLivArea',
'BsmtFullBath', 'BsmtHalfBath', 'FullBath', 'HalfBath', 'BedroomAbvGr',
'KitchenAbvGr', 'TotRmsAbvGrd', 'Fireplaces', 'GarageYrBlt', 'GarageCars',
'GarageArea', 'WoodDeckSF', 'OpenPorchSF', 'EnclosedPorch', '3SsnPorch',
'ScreenPorch', 'PoolArea']
Create a new dataframe named ames_preprocessed
that contains only the features in numeric_vars
.
# create single dataframe called ames_preprocessed
Create the scatter matrix for the Ames Housing data. This takes a few minutes to load!
# use pd.plotting.scatter_matrix
The scatter matrix took a while to load and is hard to read. Run the code below to see if adjusting some of the visualization settings helps.
sm = pd.plotting.scatter_matrix(ames_preprocessed, figsize=[20, 20]);
# Rotates the text
[s.xaxis.label.set_rotation(90) for s in sm.reshape(-1)]
[s.yaxis.label.set_rotation(0) for s in sm.reshape(-1)]
#May need to offset label when rotating to prevent overlap of figure
[s.get_yaxis().set_label_coords(-1,0.5) for s in sm.reshape(-1)]
#Hide all ticks
[s.set_xticks(()) for s in sm.reshape(-1)]
[s.set_yticks(()) for s in sm.reshape(-1)]
plt.show()
#__Solution__
sm = pd.plotting.scatter_matrix(ames_preprocessed, figsize=[20, 20]);
# Rotates the text
[s.xaxis.label.set_rotation(90) for s in sm.reshape(-1)]
[s.yaxis.label.set_rotation(0) for s in sm.reshape(-1)]
#May need to offset label when rotating to prevent overlap of figure
[s.get_yaxis().set_label_coords(-1,0.5) for s in sm.reshape(-1)]
#Hide all ticks
[s.set_xticks(()) for s in sm.reshape(-1)]
[s.set_yticks(()) for s in sm.reshape(-1)]
plt.show()
The enhanced plot demonstrates that with larger datasets, scatter matricies become less useful. Through careful examination of the matrix it's clear that TotRmsAbvGrd
seems correlated with GrLivArea
, but how easy to use would this matrix if a dataset has hundreds or thousands of variables? Also visual approach to finding correlation cannot be automated, so a numeric approach is a good next step.
Next, create and look at the correlation matrix:
Return True
for positive or negative correlations that are bigger than 0.75 in the correlation matrix:
Now, include stack
and zip
to create a more robust solution that will return the variable pairs from the correlation matrix that have correlations over .75, but less than 1.
Which varibles are highly correlated in the Ames Housing data set?
# write answer here
Now that you know which variables are correlated with eachother, which would you drop from the dataset?
# write answer here
#__SOLUTIOM__
"""
Since three different pairs of variables are highly correlated, the correct approach would be to drop one variable from each pair.
One approach would be to drop Garage Area, Total Rooms, and Total Basement Square Feet.
Garage Area: The size of the garage is dependent on how many cars are in it. If you wanted to still keep the information captured by Garage Area, you could create a new variable "Average space per car" before dropping Garage Area.
Total Rooms: There are other variables that count the number of kitchens, bathrooms, bedrooms, etc.
Total Basement Square Feet: The first floor of a building is usually built upon the foundation, which contains the basement. To keep the information that there *is* a basement, you could create a variable "HasBasement", when TotalBsmtSF >0, before deleting the original variable.
"""
Remove the chosen variables from ames_preprocessed
.
# write code here
Good job! You got some hands-on practice creating and interpreting a scatter matrix and correlation matrix to identify if variables are collinear in the Ames Housing data set. You also edited the Ames Housing data set so highly correlated variables are removed.