Data Science Plots with Seaborn#

Watch a walk-through of this lesson on YouTube

Learning Objectives#

  • Be familiar with the use-cases for Seaborn plots

  • Generate a variety of plots in Seaborn useful for EDA

  • Understand the advantages and disadvantages of different types of plots in Seaborn

  • Recognize and appropriately use strategies for visualizing continuous and categorical data

  • Visualize and reason about the distributions of data


Introduction#

Seaborn is a great Python plotting package that is built on top of Matplotlib. That is, it uses Matplotlib “under the hood”, but it offers the user a much simpler API (set of commands) that enable us to generate a variety of great-looking plots that are particularly useful in data science. You can check out Seaborn’s examples gallery to see some of the cool stuff you can do (including plotting neuroscience data!). Seaborn was written by Michael Waskom.

Comparing Seaborn to Matplotlib, you could say that Matplotlib gives you every sharp, hard-to-use tool in the shed, and you have to figure out how to learn them all without chopping off a finger. Fortunately in data science you aren’t likely to lose any fingers, but Seaborn gives you some DIY-friendly power tools to get data science jobs done easily and looking great.

Overview of Seaborn Plotting Functions#

Seaborn is organized into three categories of plot types, as shown below:

Seaborn overview

Image source: seaborn.pydata.org

You can see that Seaborn’s functions are organized around the different types of data, and questions we typically ask of data in data science: the relpolt function plots relationships between different (continuous) variables; the displot function plots distributions of data (such as histograms), and thecatplot function.

Importing the Seaborn package#

By convention, the alias we use when importing Seaborn is sns.

We’ll also be using some other packages, so we’ll import those packages at the same time.

import seaborn as sns
import pandas as pd
import matplotlib.pyplot as plt
import glob

Plotting Gapminder data#

Mostly we’ll work with long-format data from here on, but since we have worked with the Gapminder data up until now, let’s see how we can generate a plot similar to what we produced in the Introduction to Plotting with Matplotlib lesson.

First we’ll load the Ocieania GDP data, and strip the leading text off column names so that they are just the years:

df = pd.read_csv('data/gapminder_gdp_oceania.csv', index_col='country')
df.columns = df.columns.str.strip('gdpPercap_')

In the previous lesson, we plotted GDP for each country as a function of year. That is, year is on the x axis, GDP is on the y axis, and there are separate lines for each country. As discussed previously, in this dataset, year is a continuous variable, as is GDP, but country is a categorical variable.

From the figure of Seaborn plot types above, you can see that relplot() is the function to use for relational plots — that is, when you want to show the relationships between two continuous variables (year and GDP in this case).

sns.relplot(data=df.T, kind='line')  # For wide-format data we need to transpose rows & columns
plt.show() # Always good practice to have this as the last line in a plotting cell
../_images/935fd147017d245a0caf084390cf58c5c7934c1d0fd51eea3057ef03bcafe5ee.png

Compared to plotting the data in Matplotlib (below), the results are similar, but Seaborn adds some aesthetic touches by default, such as reducing the number of irrelevant frame lines, putting the legend outside the plot, and making the line for the second condition dashed (which aids in discriminating lines, especially if there are several of them, or if someone has vision problems). As we’ll see, Seaborn isn’t just another pretty face — it also makes a wide variety of plots easy to make with a consistent and relatively intuitive syntax.

# Matplotlib version of the same plot
df.T.plot()
plt.show()
../_images/3b7de75f8230b0b01363868fb1eb3c08e7995af962248c94166bbcddb112ac32.png

Plotting Long-Format Data#

Let’s move to working some some data of the sort we might find in cognitive psychology. Imagine a simple lexical decision task, an experiment in which each trial involves presenting a printed string of letters to a participant, and asking them to make a response as quickly as possible: pressing one button of the word is a real word in their native language, and another button if it is not a real word. The dependent variable in the experiment is reaction time (RT) — the amount of time that passed between the onset of the word on the screen, and the button press.

Lexical decision tasks are commonly used in psycholinguistics to explore properties of the real words, such as their length or how frequently they occur in normal usage. However, for our simple example, we’ll focus on the simpler fact that, on average, people are slower in responding to nonwords than real words (especially if the nonwords look plausibly like real words, for example words like blorp or furtle that follow English word formation patterns).

We’ve created some data, using a random number generator, to use in our plotting examples. The data were deliberately created with certain properties, so that the expected difference between RTs. We have data from 5 hypothetical participants, stored in separate data files in the data folder. The naming format for the files is ldt_sXX_data.csv, where XX stands for the subject ID number (two digits).

First we’ll load the data from the files, practicing our list comprehension, and storing the result in a pandas DataFrame called df. We need to include the ignore_index=True kwarg when concatenating the data, because otherwise there will be multiple copies of the same index in the merged DataFrame (since each data file that is read will start at index 0).

df = pd.concat([pd.read_csv(f) for f in glob.glob('data/ldt_s??_data.csv')],
              ignore_index=True)

Let’s look at a sample of the data:

df.sample(10)
ID condition RT
141 ldt_s02 nonword 0.640225
187 ldt_s04 nonword 0.727510
18 ldt_s01 word 0.578504
45 ldt_s01 nonword 0.786928
248 ldt_s05 nonword 0.830974
118 ldt_s02 word 0.623708
219 ldt_s05 word 0.658619
99 ldt_s03 nonword 1.024528
188 ldt_s04 nonword 0.726648
146 ldt_s02 nonword 0.727669

We can see that there are three columns:

  • ID is the unique participant identifier

  • condition is the type of word (real or nonword)

  • RT is reaction time, measured in seconds

Descriptive statistics#

We can use the pandas .describe() method to examine some basic properties of our data:

df.describe()
RT
count 250.000000
mean 0.710869
std 0.159536
min 0.239836
25% 0.597494
50% 0.708552
75% 0.818883
max 1.194955

Since RT is the only continuous variable in this DataFrame, we only get descriptive statistics for this variable.

Examining distributions with histograms#

Histograms are a type of plot that allows us to view distributions of values. The x axis of a histogram represents the values in the data, and the y axis shows the count of how many trials had that value on the x axis. Values on the x axis are grouped into bins — ranges of values.

In Seaborn, we use the displot() function to plot distributions. Histograms are one type of displot, but they are the default, so we can just run the following:

sns.displot(data=df, x='RT')
plt.show()
../_images/79a47a50f87d3d3a97685a8f549d7c3da37a4478236d3fb8296c75734b62d637.png

Comparing the range of values on the x axis with the min and max values from df.describe(), we see that the entire range of values is represented. The range of RT values have been binned into 11 categories, in order to draw the 11 bars in the histogram. We can see that the most common values of RT are between approximately 0.5 – 0.6 s, which corresponds to the median (50%) value shown by pd.describe(). We can also see that the data are slightly skewed — there is a wider range of values to the right than the left of the peak. Visualizing distributions is thus a useful way of examining your data, and you can likely glean more information from this plot, more quickly and easily, than from the descriptive statistics table above.

Tweaking Seaborn plots#

We can adjust the parameters of the histogram with a few kwargs, including adjusting technical aspects of how the data are treated to create the plot (like the number of bins), and aesthetic aspects of how it is represented visually (like the color).

sns.displot(data=df, x='RT',
           bins=25,
           color='red')
plt.show()
../_images/f053da788ab95e2de44c3af86a646fe858d8021a4d7350d6b7e86905bf04afb8.png

In the above plot we increased the number of bins from the default to 25. You can see that the histogram looks less smooth than it did with fewer bins — increasing the number of bins decreases the number of data points in the bins, which in turn results in greater random variability.

By default (when you don’t specify a number of bins), the software makes a decision based on characteristics of the data, which usually results in sensible bin sizes.

What to plot on the histogram y axis#

By default, on the y axis Seaborn plots the count of the number of data points in each bin. This isn’t always the best measure. It’s fine if you just want to look at how values are distributed over the variable on the x axis, but if you want to compare histograms, the scale can be deceiving if the total number of data points in different plots is different. For this reason, we can scale the data in different ways.

Particularly useful is to normalize the data. For example, you can use stat='probability' to make every histograms bar heights sum to 1. This means that the y axis now reflects the proportion (percentage) of the data in each bin.

sns.displot(data=df, x='RT',
           stat='probability')
plt.show()
../_images/7c400b3e5ee5887e016a32d4455439dca4e46f06ad6eda0ab7842d4eca5bc8ab.png

Kernel density estimate#

An alternative form of ditribution plot in Seaborn is the kernel density estimate (kde; often more simply called a “density plot”). This is like a histogram, but plots smooth curves over the range of values rather than bins as bars. This is another kind of normalization, but rather than making the heights of the bars sum to 1, the density estimate makes the area under the smoothed curve equal to 1.

Note however that while histograms represent, literally, the number of data points in each bin, the KDEs are estimates of the true distribution, and are effectively smoothed and thus slightly less accurate representations of the data you have. On the other hand, KDEs are models that predict what the true distirbution would be, if more data was collected.

In practice, the difference between histograms and KDEs is relatively small in terms of how one might interpret what they show, but this distinction between literal presentations of data and estimates meant to generalize beyond the data is an important concept that cuts through all of data science.

To override the default kind of plot generated by displot() (a histogram), we add a kind kwarg:

sns.displot(kind='kde',
           data=df, x='RT')
plt.show()
../_images/f9f1644b409c9b72bec32d567cb23136ce307792bac8b408955f610671a826c4.png

Density plots arguably make more sense for continuous variables like reaction time, since they represent continuous variables in a more continuous fashion.

Viewing the data by condition#

As noted, the experiment had two conditions, words and nonwords. So far we’ve looked at the data overall, combining RTs from those two conditions. But it’s easy to look at the data broken down by condition.

Firstly, we can use pandas’ .groupby() method to get descriptive statistics:

df.groupby('condition').describe()
RT
count mean std min 25% 50% 75% max
condition
nonword 125.0 0.815138 0.130013 0.504684 0.746378 0.815493 0.887449 1.194955
word 125.0 0.606600 0.110844 0.239836 0.526647 0.605477 0.691412 0.862992

We can add a hue kwarg to sns.displot() to separate and color-code the conditions in a histogram:

sns.displot(data=df, x='RT', hue='condition')
plt.show()
../_images/e0cd1679d2a907a0d2ebdfd2c84de0a32bd8b413f7a4e8af391590029df19655.png

The grey area in the plot represents overlap.

You can see that Seaborn does lots of nice things to the plot automatically, that make it easy to interpret: there are meaningful labels on both axes, and a legend telling us what each color represents.

color Choice#

Seaborn has a default palette, or set of colors, that it uses in sequence. You can see this with: sns.color_palette()

sns.color_palette()

So the first category plotted will be blue, the next orange, then green, etc..

Seaborn provides a number of different color palettes, although the default is a good choice as the hues are designed to be easily distinguished from each other, even for people with different forms of colorblindness.

One alternative is the Paired palette (note the capital P), which is designed for when your data are naturally organized in pairs of categories (e.g., imagine if we had the conditions word and nonword, but we wanted to compare these in two different groups, such as native speakers and late learners of a language).

sns.color_palette('Paired')
sns.color_palette('Paired')

Let’s re-plot our 2-condition histogram with the Paired palette:

sns.displot(data=df, x='RT', hue='condition', palette='Paired')
plt.show()
../_images/a8d082b0f2797f61234eecb97aa22f25b73c057cc712e2664f1c6e09abc3ac9c.png

Some palettes are better for some things than others, though — for example, in the histogram above, the overlap is not very distinct from the other shades of blue.

We can use the hue kwarg with other Seaborn plots as well, like kde:

sns.displot(kind='kde',
           data=df,
           x='RT', hue='condition', 
           palette='colorblind')
plt.show()
../_images/1f26c0630a5abe5fd887f77df505cfb479d68bc96e47cbd64c265451d4a2046f.png

Categorical Plots#

Histograms and KDEs are used to visualize the distribution of continuous variables, like RT in our example data set. We can also plot categorical variables, by drawing different histograms/KDEs for each category, like for word vs. nonword in the plots above. These plots are most useful for visualizing the shape of the distributions of the data. This is an important first step in working with data, because some ways of treating data (including statistical tests like t-tests and ANOVAs) assume that the data are normally distributed, (the classic “bell curve”), and may not work as well or as accurately on data that have very different distributions. On the other hand, in such cases other approaches to analyzing the data could be more appropriate.

That said, once we understand the distribution of the data, we might want to focus more on assessing whether there are differences between different categories (such as experimental conditions, or groups). This is where Seaborn’s catplot() family of plots comes into play.

sns.catplot() has 6 different plotting options, as shown in the figure at the top of this lesson. All of them allow us to view the distribution of the data, but in different ways, and often with a focus more on showing the mean, or median. This allows us to more easily compare differences between conditions, while considering the extent to which their distributions overlap or not. We’ll show a few of them in this lesson.

Box plots#

Box plots (sometimes called box-and-whisker plots) are a classic and widely-used approach to comparing between categorical variables. The “box” shows the interquartile range (IQR) — the range between the 25th and 75th percentiles of the data. In other words, half of the data falls within this box. The line inside the box represents the median, or the value exactly in the middle of all the values in the data set. The whiskers (lines extending out from the box) end at ±1.5 x the IQR (more on that in a minute), and then any points plotted outside the whiskers are considered outliers. The idea behind this is that the whiskers end at what might reasonably be predicted to reflect the minimum and maximum values in the distribution, and any outliers are likely to be inaccurate measurements or other types of “noise”. Below is a figure showing how the box plot mas on to the normal distribution, which is a helpful way to think about interpreting box plots.

Image from Jhguch and shared under the Creative Commons Attribution-Share Alike 2.5 Generic license

Drawing box plots with Seaborn#

We use sns.catplot() with the kind='box' kwarg:

sns.catplot(kind='box',
           data=df, 
           x='condition', y='RT')

plt.show()
../_images/2f2977f1ae68f0c813c2a72c35badb4c9ff07520af977c81345da12238b49870.png

We can see from this that the distributions of the two conditions appear quite different, and indeed their IQRs do not overlap at all. This gives us good confidence that RTs are probably different in the two conditions. You can also see that there are a few outliers (the diamonds outside the whiskers).

Although we lose some of the details of the distribution in going from a histogram to a box plot, we gain some simplicity, and it’s easier to assess how different our different conditions are.

Violin Plots#

A violin plot is in some ways the “best of both worlds”, combining a box plot with a KDE. Note that the syntax of this plot is identical to that for the box plot above, except the kind kwarg.

sns.catplot(kind='violin',
           data=df, 
           x='condition', y='RT')

plt.show()
../_images/ab714dad5a00d34dd9126133e9bcff6ead68ef692bbc36198df0d04258934231.png

The box plot has been made more narrow, but the thinker black line inside each “violin” is the IQR, and the thinner lines the whiskers. The colored, shaded area is the KDE. For aesthetic reasons, the KDE is shown on both sides of the box plot (although if you have different sub-categories, you could plot different distributions on each side).

In adding the distribution plot to the box plot, you gain detail, but perhaps at the cost of the simplicity of a box plot, and its usefulness in making inferences about differences between conditions.

I'll comment for the record that "violin" plot is a terrible name — if the distribution of your data was actually shaped like a violin that would be very suspicious! I prefer Salem Waligura-Newman's name for them, which is "stingray plots". But sadly, once a name is out there, it sticks so you should get used to calling them violin plots.

Strip and Swarm plots#

These are quite similar, and both plot the actual individual data points.

sns.catplot(kind='strip',
           data=df, 
           x='condition', y='RT')

plt.show()
../_images/9c1fd95bf854f0e2ab6ef9337a9fc30ba6d51c0a3148ad0e6060488177c9aef1.png

Swarm plots shift data points laterally when they are dense, reducing overlap and making it easier to see where the data are more dense.

sns.catplot(kind='swarm',
           data=df, 
           x='condition', y='RT')

plt.show()
../_images/a26a5945d2803533e01327a0af07033c81ff00e40545fc00c3d01794f041dcd7.png

Bar plots#

These are classic ways of representing data, that focus more on showing and comparing means for different conditions, than on the distribution. The “error bars” (black bars at the top of each bar below) represent 95% confidence intervals, which are a way of representing variance that is better-suited for comparing between means, than for assessing the shape or breadth of the distribution. You can see below that the bar plots lose a lot of the information about the distributions of data, relative to box plots or any of the other plots above.

Another limitation of bar plots is that they are “anchored” at zero. For some data this makes sense, especially if the data values relative to zero are meaningful. But in many cases — including our current RT data — this doesn’t make a lot of sense. An RT of zero would be considered invalid data, since the human nervous system can’t execute a response to a stimulus at the speed of light. So our experiment is not really concerned with how much slower than zero a person responded. As well, if your data values are far from zero, and the differences between conditions are relatively small, then bar plots make make the differences look meaninglessly small, because the y axis has to span the entire range from zero to the maximum value in the plot.

sns.catplot(kind='bar',
           data=df, 
           x='condition', y='RT')

plt.show()
../_images/ada788c400a11810f981c1e848ddf8c876fee9b9fc3a4a201acb04e4e781a21f.png

Point and Line Plots#

Point plots generally are a better choice when you want to focus on comparing means between categories, because they simply show the means as points, with 95% confidence intervals. The scale of the plots will be more appropriate, without the visual bias that bar plots create by filling the range from zero to the mean of each condition with a big block of color.

By default, Seaborn draws lines connecting the points in a plot. This is only really appropriate if your categories are “connected” in some meaningful way, such as different time points (e.g., pre- vs. post-treatment for some intervention), or when you have multiple categorical variables.

sns.catplot(kind='point',
           data=df, 
           x='condition', y='RT')

plt.show()
../_images/56a4ec047d516bb789df45e52ae6c9875c39f8344de7e0e7cd4548ba5f635879.png

We can remove the lines with join=False, and color-code the different levels of condition with hue='condition'

sns.catplot(kind='point', join=False, hue='condition',
           data=df, 
           x='condition', y='RT')

plt.show()
/var/folders/nq/chbqhzjx5hb6hfjx708rh2_40000gn/T/ipykernel_10888/2034374860.py:1: UserWarning: 

The `join` parameter is deprecated and will be removed in v0.15.0. You can remove the line between points with `linestyle='none'`.

  sns.catplot(kind='point', join=False, hue='condition',
../_images/5c0a58d0d47c603a1d54f72f5616a80890640355b960aae4d6a9794509a41935.png

Summary#

  • Seaborn makes it easy to generate plots useful for EDA

  • relplot() is used for visualizing relationships between two continuous variables

  • displot() is used for visualizing distributions of data, as histograms or kernel density estimates (KDEs)

  • catplot() is used for making comparisons between different levels of categorical data, such as different experimental conditions or groups

  • Different kinds of catplot() plots vary in how much they communicate information about the distributions of data in each category, versus on differences between the means of the categories

  • In EDA, it’s often useful to visualize the data many ways to really understand it. Even still, it’s important to have an understanding of the strengths and limitations of different kinds of plots

  • Seaborn makes it relatively easy to generate plots and adjust properties such as color and font size, and as well as making plots accessible to the widest range of viewers