Data Science and Artificial Intelligence

Data Visualization using Matplotlib and Seaborn

In this blog, we will learn how data can be visualized with the help of two of the Python most important libraries Matplotlib and Seaborn.

Also, we will read about plotting 3D graphs using Matplotlib and an Introduction to Seaborn, a compliment for Matplotlib, later in this blog. Also, the above has been explained with the help of a Use Case, visualizing data for different scenarios.

Free Step-by-step Guide To Become A Data Scientist

Subscribe and get this detailed guide absolutely FREE

The concept of using pictures and graphs to understand data has been around for many years. As day by day, the data is getting increased it is a challenge to visualize these data and provide productive results within the lesser amount of time. Thus, Data visualization comes to the rescue to convey concepts in a universal manner and to experiment in different scenarios by making slight adjustments.

Data visualization is a process of describing information in a graphical or pictorial format which helps the decision makers to analyze the data in an easier way.

  • Data visualization just not makes data more beautiful but also provides insight into complex data sets
  • Helps in identifying areas that need attention or improvement.
  • Helps to understand which fields to place where
  • Helps to predict scenarios and more.

Now, as we have understood a glimpse of Data visualization. Now, let us see how data can be visualized using Matplotlib.

INTRODUCTION TO MATPLOTLIB

Matplotlib is a Python 2D plotting library used to create 2D graphs and plots by using python scripts. It has a module named pyplot which makes things easy for plotting by providing the feature to control line styles, font properties, formatting axes, etc. Matplotlib consists of several plots like line, bar, scatter, histogram, etc.

IMPORTING MATPLOTLIB

from matplotlib import pyplot as plt
#or
import matplotlib.pyplot as plt 
%matplotlib inline

The plt is used as an alias name for Matplotlib and will be used in the rest of the coding example in this blog. Pyplot is the core object that contains the methods to create all sorts of charts and features in a plot.

The %matplotlib inline is a jupyter notebook specific command that lets you see the plots in the notebook itself.

There are the following key plots that you need to know well for basic data visualization. They are:

  • Line Plot
  • Bar Chart
  • Histogram Plot
  • Scatter Plot
  • Stack Plot
  • Pie Chart
  • Box Plot

We will see the respective plotting in detail as follows.

LINE PLOT

This is the simplest of all plotting type, as it is the visualization of a single function.

Let us see the below program to understand how line plotting is done.

import matplotlib.pyplot as plt
plt.plot([1,2,3], [4,5,6])
plt.show()

In the above program plot()  is the function used to plot the line chart, that takes two variables to plot the line.

When we plot the line using the function plot() the graph gets plotted internally but to visualize externally we use the function show().

Let us see more example to understand the line chart in detail.

import matplotlib.pyplot as plt
x = [1,2,3]
y = [4,5,6]

x2 = [2,3,4]
y2 = [6,7,8]

plt.plot(x, y, label = 'First Line', color='red', linestyle='solid')
plt.plot(x2, y2, label = 'Second Line', color='g', linestyle='dashed')
plt.xlabel('x-axis')
plt.ylabel('y-axis')
plt.title('Line Graph')

plt.legend()
plt.show()

In the above program, two lines have been created using variable x & y and x2 & y2. We can also make use of NumPy library to create the arrays X and Y.

The plt.plot() function takes additional arguments that can be used to specify different specifications.

Like in the above program we used argument such as:

  • label: to give a label to each line we used in the program.
  • color: to assign different colors to the lines. We can specify these colors in any way such as by name, color code, hex code, etc.
  • linestyle: to adjust the line style in any form such as dashed, dotted, solid, dashdot. We can also use codes to specify these linestyle such as, ‘–’, ‘:’, ‘-’, ‘-.’, respectively.

If you want to be extremely concise we can combine the colors and linestyle together into a single non-keyword argument as, ‘-g’, ‘-.r’, etc.

The plt.xlabel() and plt.ylabel() function is used to give names to the x-axis and y-axis of the graph plotted, respectively.

The plt.title() method is used to give a title to the graph and it usually appears at the topside of the graph.

The plt.legend() method is used when multiple lines are being shown within a single axis, it can be useful to create a plot legend that labels each line type. Matplotlib has a built-in way of quickly creating such a legend and it is done using this method.

The plt.legend() function keeps track of the line style and color and matches these with the correct label.

There are many more similar methods, which you may check on the official website of Matplotlib.

Another type of plotting technique is the Barchart and Histogram. Let us see its working in detail as follows.

BAR CHART

A Bar Graph is used to compare data among different categories. Bar Chart can be represented Horizontally and Vertically.

Let us see this with an example.

x = [2,4,6, 8, 10]
y = [4,2,5,6,7]

x2 = [1, 3, 5, 7, 9]
y2 = [5,3,6,4,7]

plt.bar(x,y, label = 'Bar1', color = 'r')
plt.bar(x2,y2, label = 'Bar2', color= 'c')

plt.xlabel('x-axis')
plt.ylabel('y-axis')
plt.title('Bar Graph')
plt.legend()
plt.show()

The bar graph is plotted using the bar() method.

In the above program, two bars are mentioned as Bar1 and Bar2. Bar1 is plotted using the data of x & y and Bar2 is plotted using the data of x2 & y2.

Bar1 is shown with color code ‘r’ i.e, with red color and Bar2 is shown with color code ‘c’ i.e., with cyan color.

We can also use different parameters such as height, width, align, ticklabels, etc.

We can also generate a horizontal bar graph. For this we use the method plt.barh() in place of plt.bar() method. We urge you to practice it by yourself for a better understanding.

HISTOGRAMS

Histograms are similar to Bar Chart, however, Histograms are used to show distribution. These are helpful when we have data as arrays.

Let us see this with an example where the age of the population is plotted with respect to bin.

Bin refers to the range of values that are divided into a series of usually the same size of intervals.

popul_age = [22,4,32,54,21,37,76,36,86,46.35,99,36,76,7,15,38,100,105,43]

bins = [0,10,20,30,40,50,60,70,80,90,100,110]

plt.hist(popul_age, bins, histtype = 'bar', rwidth= 0.5)
plt.xlabel('Age Group')
plt.ylabel('No. of People')
plt.title('Histogram Graph')
plt.show()

In the above program, the popul_age shows the age of various people. The variable Bin shows the number of people within a particular age group.

Therefore, in the output we can see that people of the age group 30-40 are more in numbers.

The method hist() is used to plot histograms.

The keyword histtype shows the various types of histograms that can be bar, barstacked, step, stepfilled. rwidth tells the relative width of the bars.

Similarly, we can use other parameters also as and when required.

Let us now understand about Scatter Plots and Stack Plots.

SCATTER PLOT

Scatter Plot is much alike to line graph in which instead of points being joined by line segments, the points are shown individually with a dot, circle or any other shape.

We can plot Scatter Plot graphs by using both plt.plot() and plt.scatter() methods.

Let’s first see an example to create a scatter plot using plt.plot() method:

import matplotlib.pyplot as plt
import numpy as np

x = np.linspace(0, 10, 30)
y = np.sin(x)

plt.plot(x, y, 'o', color='black')

In the above program we have created 2 arrays using NumPy library.

These 2 arrays are plotted using the method plt.plot() method. The attribute ‘o’ is used to display the shape of the scatter.

Now we will see an example to create a scatter plot using plt.scatter() method.

x = [1,2,3,4,5,6,7,8]
y = [5,2,4,2,1,4,5,2]
plt.scatter(x, y, label='skitScat', marker='*', color = 'k', s = 30)

plt.xlabel('x-axis')
plt.ylabel('y-axis')
plt.title('Scatter Graph')
plt.legend()
plt.show()

In the above program, the two arrays are plotted using the plt.scatter() method.

The keyword marker is used to display the shape in which the scatter plot will be plotted and s refers to the size of the scatter.

We can also use these character codes with line and color codes to plot points along with a line connecting them. Let us see the code below:

x = np.linspace(0, 10, 30)
y = np.sin(x)

plt.plot(x, y, '-ok')

In the above program we can see that x and y are passed as the array variable, ‘-’ is linetype, ‘o’ is the scatter plot point style and ‘k’ for the color.

The plt.plot() method is different from plt.scatter() as it does not provide the option to change the color and size of point dynamically. Whereas the latter allows us to do that.

Let’s see this by creating a random scatter plot with points of many colors and sizes.  

rng = np.random.RandomState(0)
x = rng.randn(100)
y = rng.randn(100)
colors = rng.rand(100)
sizes = 1000 * rng.rand(100)

plt.scatter(x, y, c=colors, s=sizes, alpha=0.2, cmap='viridis')
plt.colorbar();  # shows color scale

In the above program, two arrays have been created using numpy library, color is mapped to be within the range of 100. Size is given in pixel. cmap stands for colormap and is the instance or registered colormap name.

STACK PLOT

A stack plot is a plot which shows the whole data set with easy visualization of how each part makes up the whole.

Each constituent of the stack plot is stacked on top of each other.

It is more like a pie chart which shows all the various constituents of a data set. However, it is still different as stack plots have axes, unlike pie charts. Pie charts have basically one numerical data set with labels.

Let us understand this with the below code

import matplotlib.pyplot as plt

days = [1,2,3,4,5]

sleeping = [7,8,6,11,7]
eating =   [2,3,4,3,2]
working =  [7,8,7,2,2]
playing =  [8,5,7,8,13]

plt.plot([],[],color='m', label='Sleeping', linewidth=5)
plt.plot([],[],color='c', label='Eating', linewidth=5)
plt.plot([],[],color='r', label='Working', linewidth=5)
plt.plot([],[],color='k', label='Playing', linewidth=5)

plt.stackplot(days, sleeping,eating,working,playing, colors=['m','c','r','k'])

plt.xlabel('x')
plt.ylabel('y')
plt.title('Stack Plot')
plt.legend()
plt.show()

In the above code, we have considered a situation where we have taken data of 5 days since each day consists of 24 hours, it is divided into activities that we carry out on a daily basis i.e, sleeping, eating, working and playing.

We have plotted these activities with different labels, giving linewidth of 5 to each.

We have plotted it into stackplot using the plt.stackplot() method. Therefore the output will look something like this:

PIE CHART

A pie chart is a circular statistical diagram. The area of the whole chart represents the whole of the data. The areas of the pie chart represents the percentage of parts of data and it is called wedges.

Pie charts can be drawn using the function pie() in the pyplot module.

By default, the pyplot arranges the pies or wedges in counter-clockwise direction.

Let us now look into the code:

import matplotlib.pyplot as plt

day = [1,2,3,4,5]

sleeping = [7, 8, 6, 11, 7]
eating = [2, 3, 4, 3, 2]
working = [7, 8, 7, 2, 2]
playing = [8, 5, 7, 8, 13]

slices = [7, 2, 2, 13]
activities = ['sleeping', 'eating', 'working', 'playing']
cols = ['c', 'b', 'r', 'k']

plt.pie(slices, labels = activities, colors = cols, startangle = 90, shadow = True, explode = (0, 0.1, 0, 0))
plt.show()

In the above program, we have taken the same previous example on stackplot, where we have taken the data of five days and divided each day which is of 24 hours into slices of different activities and plotted these data into a pie chart using the plt.pie() method.

Within this method, we specify the “slices,” which are the relevant sizes for each part. Then, we specify the color list for the corresponding slices. Next, we can optionally specify the “Start angle” for the graph. This lets you start the line where you want. In our case, we chose a 90-degree angle for the pie chart.

We can optionally add a shadow to the plot for a bit of character and then we used “explode” to pull out a slice a bit.

So the output will be:

BOX PLOT

A box plot (or box-and-whisker plot) shows the distribution of quantitative data in a way that facilitates comparisons between variables or across levels of a categorical variable. The box shows the quartiles of the dataset while the whiskers extend to show the rest of the distribution, except for points that are determined to be “outliers” using a method that is a function of the inter-quartile range.

It can also tell you if your data is symmetrical, how tightly your data is grouped, and if and how your data is skewed.

The code for Box plot is as follows:

import numpy as np
import matplotlib.pyplot as plt
%matplotlib inline

fig, ax = plt.subplots()

# generate some random data
data1 = np.random.normal(0, 5, 100)
data2 = np.random.normal(0, 6, 100)
data3 = np.random.normal(0, 7, 100)
data4 = np.random.normal(0, 8, 100)
data = list([data1, data2, data3, data4])

# build a box plot
ax.boxplot(data)
ax.set_title('box plot')

xticklabels=['category 1', 'category 2', 'category 3', 'category 4']
ax.set_xticklabels(xticklabels)

# show the plot
plt.show()

In the above code we have created box plot with four elements. To create boxplot graph we use plt.boxplot() method. The data passed to the ax.boxplot() method can be a list or NumPy array.

The xticklabels( labels ) sets the x-axis tick labels for the current axes.

So the output would be:

INTRODUCTION TO 3D MATPLOTLIB

Matplotlib was initially designed with only two-dimensional plotting in mind

The mpl_toolkits.mplot3d import axes3d submodule included with Matplotlib provides the methods necessary to create 3D surface plots with Python.

We will now create 3D plots for Bar Charts and Scatter Plots.

3D BAR CHARTS

The difference between 2D and 3D bar chart is that, with a 3D bar other than having a starting point, height and width of the bar we also get the depth of the bar.

Let us understand this with the help of a basic example:

from mpl_toolkits.mplot3d import axes3d
import matplotlib.pyplot as plt
import numpy as np

fig = plt.figure()
ax1 = plt.axes(projection='3d')

a = [4,2,5,7,8,2,9,3,7,8]
b = [5,6,7,8,2,5,6,3,7,2]
c = np.zeros(10)

x = np.ones(10)
y = np.ones(10)
z = [5,3,7,4,8,2,4,8,9,1]

ax1.bar3d(a, b, c, x, y, z, color = 'cyan')


ax1.set_xlabel('x axis')
ax1.set_ylabel('y axis')
ax1.set_zlabel('z axis')

plt.show()

In the above program, three-dimensional plots are enabled by importing the mplot3d toolkit. plt.figure() method is used to create the 3D figure.

Once this submodule is imported, three-dimensional axes can be created by passing the keyword projection=’3d’ to any of the normal axes.

Then we have declared different variables with list and numpy and then plotted these variables using bar3d() method giving it a ‘Cyan’ color. Therefore the output will look something like this:

3D SCATTER PLOT

from mpl_toolkits.mplot3d import axes3d
import matplotlib.pyplot as plt

fig = plt.figure()
ax1 = plt.axes(projection='3d')

x = [4,2,5,7,8,2,9,3,7,8]
y = [5,6,7,8,2,5,6,3,7,2]
z = [1,2,6,3,2,7,3,3,7,2]

x2 = [-1,-2,-3,-4,-5,-6,-7,-8,-9,-10]
y2 = [-5,-6,-7,-8,-2,-5,-6,-3,-7,-2]
z2 = [1,2,6,3,2,7,3,3,7,2]

ax1.scatter(x, y, z, c='g', marker='o')
ax1.scatter(x2, y2, z2, c ='r', marker='o')

ax1.set_xlabel('x axis')
ax1.set_ylabel('y axis')
ax1.set_zlabel('z axis')

plt.show()

In the above program, we have taken two sets of variables and plotted each with a different color using the plt.scatter() method.

Therefore the output will look something like this:

INTRODUCTION TO SEABORN

We have already read about  Matplotlib, a 2D plotting library that allows us to create 2D and 3D graphs. Another complimentary package that is based on this data visualization library is Seaborn, which provides a high-level interface to draw statistical graphics.

Seaborn aims to make visualization a central part of exploring and understanding data. Its dataset-oriented plotting functions operate on dataframes and arrays containing whole datasets and internally perform the necessary mapping and statistical aggregation to produce informative plots.

We import seaborn, which is the only library necessary, as follows

import seaborn as sns

sns is the alias for Seaborn. Internally seaborn use Matplotlib to draw plots.

HISTOGRAM

We are already familiar with Histogram and a ‘hist’ function already exists in Matplotlib. A histogram represents the distribution of data by forming bins along with the range of the data and then drawing bars to show the number of observations that fall in each bin.

To illustrate this let us see the code below:

import seaborn as sns
import numpy as np

x = np.random.normal(size=100)
sns.distplot(x, kde=False, rug=True);

In the above program, we have created an array using the numpy library and plotted the histogram using the displot() method.

The keyword ‘kde’ passed stands for Kernel Density Estimate and is  a non-parametric way to estimate the probability density function of a random variable. By default kde is True.

The keyword rug adds a rug plot which is used to draw a small vertical tick at each observation.

Therefore the output will look something like this:

Let us now make kde as true, which by default it is and also remove rug and see what happens:

sns.distplot(x);

The kernel density estimate may be less familiar, but it can be a useful tool for plotting the shape of a distribution. Like the histogram, the KDE plots encode the density of observations on one axis with height along the other axis.

If we use the kdeplot() function in seaborn, we get the same curve. Let’s look at an example.

sns.kdeplot(x, shade=True);

Until now we have seen plotting univariate distributions. Where univariate refers to an expression, equation, function or polynomial of only one variable. Now we will see an example on plotting bivariate distributions.

SCATTER PLOT

The most familiar way to visualize a bivariate distribution is a scatter plot, where each observation is shown with a point at the x and y values.

We can draw a scatterplot with the matplotlib plt.scatter function, and it is also the default kind of plot shown by the jointplot() function in seaborn.

mean, cov = [0, 1], [(1, .5), (.5, 1)]
data = np.random.multivariate_normal(mean, cov, 200)
df = pd.DataFrame(data, columns=["x", "y"])

sns.jointplot(x="x", y="y", data=df);

BOX PLOT

We have already read about the Box Plot using Matplotlib library. Let us now see how plotting of Box Plot is done using Seaborn library.

import seaborn as sns

sns.set_style("whitegrid")
data = np.random.normal(size=(20, 6)) + np.arange(6) / 2
sns.boxplot(data=data)

In the above example, the method set_style is used to set the theme as a background with white grids.

Then we have created 2 numpy arrays (the first one having 20 arrays with 6 elements and the other one with 6 elements from to 5 diving each by 2) and summed up both.

And then the Boxplot has been plotted using the boxplot() method passing data as the argument.

So the output would be:

Let us understand this with the help of a use case that would help understand the above concepts better.

We have taken a dataset which consists of the marks secured by the students in various subjects named as StudentsPerformance.csv.

You can download the dataset from the below link:

studentperformance.csv

This data set consists of the marks secured by the students in various subjects.

Column List

  • gender
  • race/ethnicity
  • parental level of education
  • test preparation course
  • lunch
  • math score
  • reading score
  • writing score

1.  First, we will import all important libraries and then import the csv file.

import numpy as np
import pandas as pd
import seaborn as sns
import matplotlib.pyplot as plt

data=pd.read_csv(r'C:\Users\Aeon\StudentsPerformance.csv')
#note: give the path where your file has been located.

Now we will recognize and analyze our data using a wide variety of functions in the pandas library.

#we'll see the first five lines.
data.head()
#we'll see the last five lines.
data.tail()
#it is a process that shows the property value in the data set and shows the numbers in the register values.
data.info()
#It is a function that shows the analysis of numerical values.
data.describe()
#It shows the data types in the data set.
data.dtypes
#all data control for null values
data.isnull().sum()
#show count Gender
data['gender'].value_counts()

DATA VISUALIZATION

1. We will see visualization based on gender with the help of bar graphs.

sns.barplot(x=data['gender'].value_counts().index,y=data['gender'].value_counts().values)
plt.title('Genders other rate')
plt.ylabel('Rates')
#plt.legend(loc=0)
plt.show()

2. Count of the column race/ethnicity is shown with the below graph

plt.figure(figsize=(7,7))
sns.barplot(x=data['race/ethnicity'].value_counts().index,
              y=data['race/ethnicity'].value_counts().values)
plt.xlabel('Race/Ethnicity')
plt.ylabel('Frequency')
plt.title('Show of Race/Ethnicity Bar Plot')
plt.show()

3. Plotting graphs showing comparison for ‘writing score’, ‘reading score’ and ‘math score’ for both the ‘genders’ based on ‘parental level of education’ respectively.

plt.figure(figsize=(10,7))
sns.barplot(x = "parental level of education", y = "writing score", hue = "gender", data = data)
plt.xticks(rotation=45)
plt.show()
plt.figure(figsize=(10,7))
sns.barplot(x = "parental level of education", y = "reading score", hue = "gender", data = data)
plt.xticks(rotation=45)
plt.show()
plt.figure(figsize=(10,7))
sns.barplot(x = "parental level of education", y = "math score", hue = "gender", data = data)
plt.xticks(rotation=45)
plt.show()

4. Plotting a graph for both the genders comparing math score for different levels of education.

plt.figure(figsize=(12,7))
sns.catplot(y="gender", x="math score",
                 hue="parental level of education",
                 data=data, kind="bar")
plt.title('for Parental Level Of Education Gender & Math_Score')
plt.show()

5. Visualizing different groups based on percentage with the help of a piechart.

labels=data['race/ethnicity'].value_counts().index
colors=['cyan','pink','orange','lightgreen','yellow']
explode=[0,0,0.1,0,0]
values=data['race/ethnicity'].value_counts().values

#visualization
plt.figure(figsize=(7,7))
plt.pie(values,explode=explode,labels=labels,colors=colors,autopct='%1.1f%%')
plt.title('Race/Ethnicity According Analysis',color='black',fontsize=10)
plt.show()

6. Plotting a graph for math score vs writing score for both the genders using scatter plot.

sns.lmplot(x='math score',y='writing score',hue='gender',data=data)
plt.xlabel('Math Score')
plt.ylabel('Writing Score')
plt.title('Math Score vs Writing Score')
plt.show()

7.  Visualizing frequency of math score vs writing score vs reading score using kde plot.

sns.kdeplot(data['math score'], shade=True,color='c')
sns.kdeplot(data['reading score'],shade=True,color='r')
sns.kdeplot(data['writing score'],shade=True,color='b')
plt.xlabel('Values')
plt.ylabel('Frequency')
plt.title('Math Score vs Reading Score vs Writing Score Kde Plot System Analysis')
plt.show()

8. Visualization for math score for both the genders using Box Plot

sns.boxplot(x=data['gender'],y=data['math score'])
plt.show()

9. Data visualization using Pairplot.

Pairplot plots pairwise relationships in a dataset.

sns.pairplot(data)
plt.show()

Like above, we can perform numerous operations on various data and create Data Visualization using several plotting techniques.

This brings us to the end of our blog on Data Visualization. Hope this blog helped you to understand and use plotting techniques to create various Data Visualization.

Keep visiting our website for more Data Science and Big Data related blogs.

You can refer to our previous blogs on Python important libraries Numpy and Pandas, whose links have been given below:

One Comment

  1. Hello everybody! I’m relatively new to 3d printing and I have a lot of questions on the topic, so I hope you will not get mad at me for asking here at least a few of them. I think before I’ll get seriously into sculpting I should focus on the software itself, and that’s what I would like to ask you about. Mainly, should I begin with the most crudest program there is or would it be better to start on something more complicated? I’m worried that I’ll get some unwanted quirks while working with less complex software. Second question is about the software as well: should I look for CAD software that will allow me design and slice it in it, or should I use a separate software for each of them? Will it even make a difference? Weirdly, I couldn’t find the answer to that, as it seems like most blogs and sites want to focus on the very basics (like what is 3d printing and so on), and while the answers to those questions are fine, it seems like no one wants to go into the details (it looks like some of them even steal from each other! I swear I’ve found the same answers to the same questions on at least 3 different sites) but I’m getting off-topic… The last question is about 3d pens. Would it be possible to somehow convert whatever I draw with a 3d pen to a 3d model in a CAD software? For example, if I’ll draw a horse with 3d pen, would it be possible to get its outline in a program? I’m not sure how that would even work, but the very idea sounds appealing to me. Anyway, I think I’ll stop here just in case no one will answer me and all of this writing will go to waste. I apologise that I’m using your content to ask questions, but I hope you’ll understand and advice a rookie like me. Anyway, thank you for posting. I did learn something from this and that’s always appreciated. Thank you, and I hope to hear back from you very soon 🙂

Leave a Reply

Your email address will not be published. Required fields are marked *

This site uses Akismet to reduce spam. Learn how your comment data is processed.

Related Articles

Close
Close