Data visualization provides insight into the distribution and relationships between variables in a dataset.<\/p>\n
This insight can be helpful in selecting data preparation techniques to apply prior to modeling and the types of algorithms that may be most suited to the data.<\/p>\n
Seaborn is a data visualization library for Python that runs on top of the popular Matplotlib data visualization library, although it provides a simple interface and aesthetically better-looking plots.<\/p>\n
In this tutorial, you will discover a gentle introduction to Seaborn data visualization for machine learning.<\/p>\n
After completing this tutorial, you will know:<\/p>\n
- \n
- How to summarize the distribution of variables using bar charts, histograms, and box and whisker plots.<\/li>\n
- How to summarize relationships using line plots and scatter plots.<\/li>\n
- How to compare the distribution and relationships of variables for different class values on the same plot.<\/li>\n<\/ul>\n
Let’s get started.<\/p>\n
![\"How](\"https:\/\/machinelearningmastery.com\/wp-content\/uploads\/2020\/06\/How-to-use-Seaborn-Data-Visualization-for-Machine-Learning.jpg\")
<\/p>\nHow to use Seaborn Data Visualization for Machine Learning
Photo by Martin Pettitt<\/a>, some rights reserved.<\/p>\n<\/div>\nTutorial Overview<\/h2>\n
This tutorial is divided into six parts; they are:<\/p>\n
- \n
- Seaborn Data Visualization Library<\/li>\n
- Line Plots<\/li>\n
- Bar Chart Plots<\/li>\n
- Histogram Plots<\/li>\n
- Box and Whisker Plots<\/li>\n
- Scatter Plots<\/li>\n<\/ul>\n
Seaborn Data Visualization Library<\/h2>\n
The primary plotting library for Python is called Matplotlib<\/a>.<\/p>\n
Seaborn<\/a> is a plotting library that offers a simpler interface, sensible defaults for plots needed for machine learning, and most importantly, the plots are aesthetically better looking than those in Matplotlib.<\/p>\n
Seaborn requires that Matplotlib is installed first.<\/p>\n
You can install Matplotlib directly using pip<\/a>, as follows:<\/p>\n
sudo pip install matplotlib<\/pre>\n
Once installed, you can confirm that the library can be loaded and used by printing the version number, as follows:<\/p>\n
# matplotlib\r\nimport matplotlib\r\nprint('matplotlib: %s' % matplotlib.__version__)<\/pre>\nRunning the example prints the current version of the Matplotlib library.<\/p>\n
matplotlib: 3.1.2<\/pre>\n
Next, the Seaborn library can be installed, also using pip:<\/p>\n
sudo pip install seaborn<\/pre>\n
Once installed, we can also confirm the library can be loaded and used by printing the version number, as follows:<\/p>\n
# seaborn\r\nimport seaborn\r\nprint('seaborn: %s' % seaborn.__version__)<\/pre>\nRunning the example prints the current version of the Seaborn library.<\/p>\n
seaborn: 0.10.0<\/pre>\n
To create Seaborn plots, you must import the Seaborn library and call functions to create the plots.<\/p>\n
Importantly, Seaborn plotting functions expect data to be provided as Pandas DataFrames<\/a>. This means that if you are loading your data from CSV files, you must use Pandas functions like read_csv()<\/a> to load your data as a DataFrame. When plotting, columns can then be specified via the DataFrame name or column index.<\/p>\n
To show the plot, you can call the show() function<\/a> on Matplotlib library.<\/p>\n
...\r\n# display the plot\r\npyplot.show()<\/pre>\n
Alternatively, the plots can be saved to file, such as a PNG formatted image file. The savefig() Matplotlib function<\/a> can be used to save images.<\/p>\n
...\r\n# save the plot\r\npyplot.savefig('my_image.png')<\/pre>\nNow that we have Seaborn installed, let’s look at some common plots we may need when working with machine learning data.<\/p>\n
Line Plots<\/h2>\n
A line plot is generally used to present observations collected at regular intervals.<\/p>\n
The x-axis represents the regular interval, such as time. The y-axis shows the observations, ordered by the x-axis and connected by a line.<\/p>\n
A line plot can be created in Seaborn by calling the lineplot() function<\/a> and passing the x-axis data for the regular interval, and y-axis for the observations.<\/p>\n
We can demonstrate a line plot using a time series dataset of monthly car sales<\/a>.<\/p>\n
The dataset has two columns: “Month<\/em>” and “Sales<\/em>.” Month will be used as the x-axis and Sales will be plotted on the y-axis.<\/p>\n
...\r\n# create line plot\r\nlineplot(x='Month', y='Sales', data=dataset)<\/pre>\n
Tying this together, the complete example is listed below.<\/p>\n
# line plot of a time series dataset\r\nfrom pandas import read_csv\r\nfrom seaborn import lineplot\r\nfrom matplotlib import pyplot\r\n# load the dataset\r\nurl = 'https:\/\/raw.githubusercontent.com\/jbrownlee\/Datasets\/master\/monthly-car-sales.csv'\r\ndataset = read_csv(url, header=0)\r\n# create line plot\r\nlineplot(x='Month', y='Sales', data=dataset)\r\n# show plot\r\npyplot.show()<\/pre>\n
Running the example first loads the time series dataset and creates a line plot of the data, clearly showing a trend and seasonality in the sales data.<\/p>\n
![\"Line](\"https:\/\/machinelearningmastery.com\/wp-content\/uploads\/2020\/02\/Line-Plot-of-a-Time-Series-Dataset.png\")
<\/p>\nLine Plot of a Time Series Dataset<\/p>\n<\/div>\n
For more great examples of line plots with Seaborn, see: Visualizing statistical relationships<\/a>.<\/p>\n
Bar Chart Plots<\/h2>\n
A bar chart is generally used to present relative quantities for multiple categories.<\/p>\n
The x-axis represents the categories that are spaced evenly. The y-axis represents the quantity for each category and is drawn as a bar from the baseline to the appropriate level on the y-axis.<\/p>\n
A bar chart can be created in Seaborn by calling the countplot() function<\/a> and passing the data.<\/p>\n
We will demonstrate a bar chart with a variable from the breast cancer classification dataset<\/a> that is comprised of categorical input variables.<\/p>\n
We will just plot one variable, in this case, the first variable which is the age bracket.<\/p>\n
...\r\n# create line plot\r\ncountplot(x=0, data=dataset)<\/pre>\n
Tying this together, the complete example is listed below.<\/p>\n
# bar chart plot of a categorical variable\r\nfrom pandas import read_csv\r\nfrom seaborn import countplot\r\nfrom matplotlib import pyplot\r\n# load the dataset\r\nurl = 'https:\/\/raw.githubusercontent.com\/jbrownlee\/Datasets\/master\/breast-cancer.csv'\r\ndataset = read_csv(url, header=None)\r\n# create bar chart plot\r\ncountplot(x=0, data=dataset)\r\n# show plot\r\npyplot.show()<\/pre>\n
Running the example first loads the breast cancer dataset and creates a bar chart plot of the data, showing each age group and the number of individuals (samples) that fall within reach group.<\/p>\n
![\"Bar](\"https:\/\/machinelearningmastery.com\/wp-content\/uploads\/2020\/02\/Bar-Chart-Plot-of-Age-Range-Categorical-Variable.png\")
<\/p>\nBar Chart Plot of Age Range Categorical Variable<\/p>\n<\/div>\n
We might also want to plot the counts for each category for a variable, such as the first variable, against the class label.<\/p>\n
This can be achieved using the countplot()<\/em> function and specifying the class variable (column index 9) via the “hue<\/em>” argument, as follows:<\/p>\n
...\r\n# create bar chart plot\r\ncountplot(x=0, hue=9, data=dataset)<\/pre>\n
Tying this together, the complete example is listed below.<\/p>\n
# bar chart plot of a categorical variable against a class variable\r\nfrom pandas import read_csv\r\nfrom seaborn import countplot\r\nfrom matplotlib import pyplot\r\n# load the dataset\r\nurl = 'https:\/\/raw.githubusercontent.com\/jbrownlee\/Datasets\/master\/breast-cancer.csv'\r\ndataset = read_csv(url, header=None)\r\n# create bar chart plot\r\ncountplot(x=0, hue=9, data=dataset)\r\n# show plot\r\npyplot.show()<\/pre>\n
Running the example first loads the breast cancer dataset and creates a bar chart plot of the data, showing each age group and the number of individuals (samples) that fall within each group separated by the two class labels for the dataset.<\/p>\n
![\"Bar](\"https:\/\/machinelearningmastery.com\/wp-content\/uploads\/2020\/02\/Bar-Chart-Plot-of-Age-Range-Categorical-Variable-by-Class-Label.png\")
<\/p>\nBar Chart Plot of Age Range Categorical Variable by Class Label<\/p>\n<\/div>\n
For more great examples of bar chart plots with Seaborn, see: Plotting with categorical data<\/a>.<\/p>\n
Histogram Plots<\/h2>\n
A histogram plot is generally used to summarize the distribution of a numerical data sample.<\/p>\n
The x-axis represents discrete bins or intervals for the observations. For example, observations with values between 1 and 10 may be split into five bins, the values [1,2] would be allocated to the first bin, [3,4] would be allocated to the second bin, and so on.<\/p>\n
The y-axis represents the frequency or count of the number of observations in the dataset that belong to each bin.<\/p>\n
Essentially, a data sample is transformed into a bar chart where each category on the x-axis represents an interval of observation values.<\/p>\n
A histogram can be created in Seaborn by calling the distplot() function<\/a> and passing the variable.<\/p>\n
We will demonstrate a boxplot with a numerical variable from the diabetes classification dataset<\/a>. We will just plot one variable, in this case, the first variable, which is the number of times that a patient was pregnant.<\/p>\n
...\r\n# create histogram plot\r\ndistplot(dataset[[0]])<\/pre>\n
Tying this together, the complete example is listed below.<\/p>\n
# histogram plot of a numerical variable\r\nfrom pandas import read_csv\r\nfrom seaborn import distplot\r\nfrom matplotlib import pyplot\r\n# load the dataset\r\nurl = 'https:\/\/raw.githubusercontent.com\/jbrownlee\/Datasets\/master\/pima-indians-diabetes.csv'\r\ndataset = read_csv(url, header=None)\r\n# create histogram plot\r\ndistplot(dataset[[0]])\r\n# show plot\r\npyplot.show()<\/pre>\n
Running the example first loads the diabetes dataset and creates a histogram plot of the variable, showing the distribution of the values with a hard cut-off at zero.<\/p>\n
The plot shows both the histogram (counts of bins) as well as a smooth estimate of the probability density function.<\/p>\n
![\"Histogram](\"https:\/\/machinelearningmastery.com\/wp-content\/uploads\/2020\/02\/Histogram-Plot-of-Number-of-Times-Pregnant-Numerical-Variable.png\")
<\/p>\nHistogram Plot of Number of Times Pregnant Numerical Variable<\/p>\n<\/div>\n
For more great examples of histogram plots with Seaborn, see: Visualizing the distribution of a dataset<\/a>.<\/p>\n
Box and Whisker Plots<\/h2>\n
A box and whisker plot, or boxplot for short, is generally used to summarize the distribution of a data sample.<\/p>\n
The x-axis is used to represent the data sample, where multiple boxplots can be drawn side by side on the x-axis if desired.<\/p>\n
The y-axis represents the observation values. A box is drawn to summarize the middle 50 percent of the dataset starting at the observation at the 25th percentile and ending at the 75th percentile. This is called the interquartile range<\/a>, or IQR. The median, or 50th percentile, is drawn with a line.<\/p>\n
Lines called whiskers are drawn extending from both ends of the box, calculated as (1.5 * IQR) to demonstrate the expected range of sensible values in the distribution. Observations outside the whiskers might be outliers and are drawn with small circles.<\/p>\n
A boxplot can be created in Seaborn by calling the boxplot() function<\/a> and passing the data.<\/p>\n
We will demonstrate a boxplot with a numerical variable from the diabetes classification dataset<\/a>. We will just plot one variable, in this case, the first variable, which is the number of times that a patient was pregnant.<\/p>\n
...\r\n# create box and whisker plot\r\nboxplot(x=0, data=dataset)<\/pre>\n
Tying this together, the complete example is listed below.<\/p>\n
# box and whisker plot of a numerical variable\r\nfrom pandas import read_csv\r\nfrom seaborn import boxplot\r\nfrom matplotlib import pyplot\r\n# load the dataset\r\nurl = 'https:\/\/raw.githubusercontent.com\/jbrownlee\/Datasets\/master\/pima-indians-diabetes.csv'\r\ndataset = read_csv(url, header=None)\r\n# create box and whisker plot\r\nboxplot(x=0, data=dataset)\r\n# show plot\r\npyplot.show()<\/pre>\n
Running the example first loads the diabetes dataset and creates a boxplot plot of the first input variable, showing the distribution of the number of times patients were pregnant.<\/p>\n
We can see the median just above 2.5 times, some outliers up around 15 times (wow!).<\/p>\n
![\"Box](\"https:\/\/machinelearningmastery.com\/wp-content\/uploads\/2020\/02\/Box-and-Whisker-Plot-of-Number-of-Times-Pregnant-Numerical-Variable.png\")
<\/p>\nBox and Whisker Plot of Number of Times Pregnant Numerical Variable<\/p>\n<\/div>\n
We might also want to plot the distribution of the numerical variable for each value of a categorical variable, such as the first variable, against the class label.<\/p>\n
This can be achieved by calling the boxplot()<\/em> function and passing the class variable as the x-axis and the numerical variable as the y-axis.<\/p>\n
...\r\n# create box and whisker plot\r\nboxplot(x=8, y=0, data=dataset)<\/pre>\n
Tying this together, the complete example is listed below.<\/p>\n
# box and whisker plot of a numerical variable vs class label\r\nfrom pandas import read_csv\r\nfrom seaborn import boxplot\r\nfrom matplotlib import pyplot\r\n# load the dataset\r\nurl = 'https:\/\/raw.githubusercontent.com\/jbrownlee\/Datasets\/master\/pima-indians-diabetes.csv'\r\ndataset = read_csv(url, header=None)\r\n# create box and whisker plot\r\nboxplot(x=8, y=0, data=dataset)\r\n# show plot\r\npyplot.show()<\/pre>\n
Running the example first loads the diabetes dataset and creates a boxplot of the data, showing the distribution of the number of times pregnant as a numerical variable for the two-class labels.<\/p>\n
![\"Box](\"https:\/\/machinelearningmastery.com\/wp-content\/uploads\/2020\/02\/Box-and-Whisker-Plot-of-Number-of-Times-Pregnant-Numerical-Variable-by-Class-Label.png\")
<\/p>\nBox and Whisker Plot of Number of Times Pregnant Numerical Variable by Class Label<\/p>\n<\/div>\n
Scatter Plots<\/h2>\n
A scatter plot, or scatterplot, is generally used to summarize the relationship between two paired data samples.<\/p>\n
Paired data samples mean that two measures were recorded for a given observation, such as the weight and height of a person.<\/p>\n
The x-axis represents observation values for the first sample, and the y-axis represents the observation values for the second sample. Each point on the plot represents a single observation.<\/p>\n
A scatterplot can be created in Seaborn by calling the scatterplot() function<\/a> and passing the two numerical variables.<\/p>\n
We will demonstrate a scatterplot with two numerical variables from the diabetes classification dataset<\/a>. We will plot the first versus the second variable, in this case, the first variable, which is the number of times that a patient was pregnant, and the second is the plasma glucose concentration after a two hour oral glucose tolerance test (more details of the variables here<\/a>).<\/p>\n
...\r\n# create scatter plot\r\nscatterplot(x=0, y=1, data=dataset)<\/pre>\n
Tying this together, the complete example is listed below.<\/p>\n
# scatter plot of two numerical variables\r\nfrom pandas import read_csv\r\nfrom seaborn import scatterplot\r\nfrom matplotlib import pyplot\r\n# load the dataset\r\nurl = 'https:\/\/raw.githubusercontent.com\/jbrownlee\/Datasets\/master\/pima-indians-diabetes.csv'\r\ndataset = read_csv(url, header=None)\r\n# create scatter plot\r\nscatterplot(x=0, y=1, data=dataset)\r\n# show plot\r\npyplot.show()<\/pre>\n
Running the example first loads the diabetes dataset and creates a scatter plot of the first two input variables.<\/p>\n
We can see a somewhat uniform relationship between the two variables.<\/p>\n
![\"Scatter](\"https:\/\/machinelearningmastery.com\/wp-content\/uploads\/2020\/02\/Scatter-Plot-of-Number-of-Times-Pregnant-vs-Plasma-Glucose-Numerical-Variables.png\")
<\/p>\nScatter Plot of Number of Times Pregnant vs. Plasma Glucose Numerical Variables<\/p>\n<\/div>\n
We might also want to plot the relationship for the pair of numerical variables against the class label.<\/p>\n
This can be achieved using the scatterplot() function and specifying the class variable (column index 8) via the “hue<\/em>” argument, as follows:<\/p>\n
...\r\n# create scatter plot\r\nscatterplot(x=0, y=1, hue=8, data=dataset)<\/pre>\n
Tying this together, the complete example is listed below.<\/p>\n
# scatter plot of two numerical variables vs class label\r\nfrom pandas import read_csv\r\nfrom seaborn import scatterplot\r\nfrom matplotlib import pyplot\r\n# load the dataset\r\nurl = 'https:\/\/raw.githubusercontent.com\/jbrownlee\/Datasets\/master\/pima-indians-diabetes.csv'\r\ndataset = read_csv(url, header=None)\r\n# create scatter plot\r\nscatterplot(x=0, y=1, hue=8, data=dataset)\r\n# show plot\r\npyplot.show()<\/pre>\n
Running the example first loads the diabetes dataset and creates a scatter plot of the first two variables vs. class label.<\/p>\n
![\"Scatter](\"https:\/\/machinelearningmastery.com\/wp-content\/uploads\/2020\/02\/Scatter-Plot-of-Number-of-Times-Pregnant-vs-Plasma-Glucose-Numerical-Variables-by-Class-Label.png\")
<\/p>\nScatter Plot of Number of Times Pregnant vs. Plasma Glucose Numerical Variables by Class Label<\/p>\n<\/div>\n
Further Reading<\/h2>\n
This section provides more resources on the topic if you are looking to go deeper.<\/p>\n
Tutorials<\/h3>\n
- \n
- A Gentle Introduction to Data Visualization Methods in Python<\/a><\/li>\n<\/ul>\n
APIs<\/h3>\n
- \n
- Seaborn Homepage<\/a>.<\/li>\n
- Official seaborn tutorial<\/a>.<\/li>\n<\/ul>\n
Summary<\/h2>\n
In this tutorial, you discovered a gentle introduction to Seaborn data visualization for machine learning.<\/p>\n
Specifically, you learned:<\/p>\n
- \n
- How to summarize the distribution of variables using bar charts, histograms, and box and whisker plots.<\/li>\n
- How to summarize relationships using line plots and scatter plots.<\/li>\n
- How to compare the distribution and relationships of variables for different class values on the same plot.<\/li>\n<\/ul>\n
Do you have any questions?<\/strong>
\nAsk your questions in the comments below and I will do my best to answer.<\/p>\nThe post How to use Seaborn Data Visualization for Machine Learning<\/a> appeared first on Machine Learning Mastery<\/a>.<\/p>\n<\/div>\n
Go to Source<\/a><\/p>\n","protected":false},"excerpt":{"rendered":"
- Official seaborn tutorial<\/a>.<\/li>\n<\/ul>\n
- Seaborn Homepage<\/a>.<\/li>\n
- A Gentle Introduction to Data Visualization Methods in Python<\/a><\/li>\n<\/ul>\n