Scribd
Upload
128 views

Data Visualization Using Plotly, Matplotlib, Seaborn and Squarify - Data Science

This document discusses data visualization techniques using Python libraries like Plotly, Matplotlib, Seaborn and Squarify. It provides an overview of various chart types like line charts, bar charts, histograms, boxplots, pie charts and scatter plots that can be created using these libraries. It also includes a case study on analyzing employee attrition rate from an HR dataset using these visualization techniques.

Uploaded by

RR
Copyright
© © All Rights Reserved
Available Formats
Download as PDF, TXT or read online on Scribd
128 views

Data Visualization Using Plotly, Matplotlib, Seaborn and Squarify - Data Science

This document discusses data visualization techniques using Python libraries like Plotly, Matplotlib, Seaborn and Squarify. It provides an overview of various chart types like line charts, bar charts, histograms, boxplots, pie charts and scatter plots that can be created using these libraries. It also includes a case study on analyzing employee attrition rate from an HR dataset using these visualization techniques.

Uploaded by

RR
Copyright
© © All Rights Reserved
Available Formats
Download as PDF, TXT or read online on Scribd
You are on page 1/ 61

Data Visualization using plotly, matplotlib,

seaborn and squarify | Data Science


Data Visualization is one of the important activities we perform when doing Exploratory
Data Analysis. It helps in preparing business reports, visual dashboards, storytelling etc
important tasks. In this post I have explained how to ask questions from the data and in
return get the self-explanatory graphs. In this You will learn the use of various python
libraries like plotly, matplotlib, seaborn, squarify etc to plot those graphs.

Key takeaways from this post are:


• Asking questions from data set
• Univariate Analysis
• Bivariate Analysis
• Analysis of more than 3 variables
• 3D Visualization
• Case Study on employee Attrition Rate using HR Data Set
plotly
• Visualization library for the data Era
Line Chart in plotly
• 2 numeric variables with 1-1 mapping, i.e in situations where we have 1 y value
corresponding to 1 x value

You can export images to html file only with offline mode

• https://plot.ly/python/static-image-export/

• https://plot.ly/python/privacy/

Note that this is a bare chart with no information, later in the activity we will add title, x
labels and y labels.
Basic Bar chart in plotly
• 1 Categorical variable

Histogram in plotly
• 1 numeric variable
Boxplot in plotly
• 1 Numeric variable
Pie chart in plotly

• 1 Categorical variable
Note: We do not suggest you use pie chart, one reason being the total is not always
obvious and second, having many levels will make the chart cluttered.
Scatter plot in plotly

• 2 numeric variables
• One x might have multiple corresponding y values
Tree map
https://plot.ly/python/treemaps/
Case Study
Now let us use our new found skill to extract insights from a dataset
hr_data Description
Education 1 ‘Below College’ 2 ‘College’ 3 ‘Bachelor’ 4 ‘Master’ 5 ‘Doctor’
EnvironmentSatisfaction 1 ‘Low’ 2 ‘Medium’ 3 ‘High’ 4 ‘Very High’

JobInvolvement 1 ‘Low’ 2 ‘Medium’ 3 ‘High’ 4 ‘Very High’


JobSatisfaction 1 ‘Low’ 2 ‘Medium’ 3 ‘High’ 4 ‘Very High’
PerformanceRating 1 ‘Low’ 2 ‘Good’ 3 ‘Excellent’ 4 ‘Outstanding’
RelationshipSatisfaction 1 ‘Low’ 2 ‘Medium’ 3 ‘High’ 4 ‘Very High’

WorkLifeBalance 1 ‘Bad’ 2 ‘Good’ 3 ‘Better’ 4 ‘Best’


Checking the datatypes
Checking the number of unique values in each column
Observations:

Most columns have fewer than 4 unique levels

NumCompaniesWorked and PercentSalaryHike have less than 15 values and we can convert
these into categorical values for analysis purposes,
this is fairly subjective. You can also continue with these as integer values.

Replacing the integers with above values with the values in the description
• hr_data.Education = hr_data.Education.replace(to_replace=[1,2,3,4,5],value=[‘Below
College’, ‘College’, ‘Bachelor’, ‘Master’, ‘Doctor’])
• hr_data.EnvironmentSatisfaction =
hr_data.EnvironmentSatisfaction.replace(to_replace=[1,2,3,4],value=[‘Low’,
‘Medium’, ‘High’, ‘Very High’])
• hr_data.JobInvolvement =
hr_data.JobInvolvement.replace(to_replace=[1,2,3,4],value=[‘Low’, ‘Medium’, ‘High’,
‘Very High’])

• hr_data.JobSatisfaction =
hr_data.JobSatisfaction.replace(to_replace=[1,2,3,4],value=[‘Low’, ‘Medium’, ‘High’,
‘Very High’])
• hr_data.PerformanceRating =
hr_data.PerformanceRating.replace(to_replace=[1,2,3,4],value=[‘Low’, ‘Good’,
‘Excellent’, ‘Outstanding’])
• hr_data.RelationshipSatisfaction =
hr_data.RelationshipSatisfaction.replace(to_replace=[1,2,3,4],value=[‘Low’,
‘Medium’, ‘High’, ‘Very High’])
• hr_data.WorkLifeBalance =
hr_data.WorkLifeBalance.replace(to_replace=[1,2,3,4],value=[‘Bad’, ‘Good’, ‘Better’,
‘Best’])
Extract categorical columns
Columns with 15 or less levels are considered as categorical columns for the purpose of this
analysis

We have decided to treat all the columns with 15 or less levels as categorical columns, the
following few lines of code extract all the columns which satisfy the condition.
Print the categorical column names

Check if the above columns are categorical in the data set

Type Conversion
• n dimensional type conversion to ‘category’ is not implemented yet
Categorical attributes summary

Extracting Numeric Columns


Exploratory Data Analysis
Univariate Analysis
1. What is the attrition rate in the company?

Attrition in numbers (pandas)

This is one way to tell matplotlib to plot the graphs in the notebook
Attrition rate in percentage (pandas)
plotly In percentages
2. What is the Gender Distribution in the company?
Steps to create a bar chart with counts for a categorical variable in plotly

• Steps to create a bar chart with counts for a categorical variable


o create an object and store the counts (optional)
o create a bar object
▪ pass the x values
▪ pass the y values
▪ optional :
▪ text to be displayed
▪ text position
▪ color of the bar
▪ name of the bar (trace in plotly terminology)
o create a layout object
▪ title – font and size of title
▪ x axis – font and size of xaxis text
▪ y axis – font and size of yaxis text
o create a figure object:
▪ add data
▪ add layout
o plot the figure object
Observations:
Irrespective of the distance bin, there is a global pattern i.e every bin
has more male employees
One of the metric to find out if you have chosen the correct number of clusters
is to see if you can give a name to all your clusters in terms of business.

This is all for now. I have also created a report on Employee Attrition Rate
Analysis. you may like to check it as well. Please read it using the below link.

Report on Employee Attrition Rate Analysis

Thank you for reading. Your comments, thoughts on this post are most
welcome.

You might also like