Pandas & Seaborn - A guide to handle & visualize data in Python
Last Updated: July..
Pandas
Why is Pandas great? It is built on top on NumPy. It uses its multi-dimensional arrays and fast operations internally to provide higher level methods for manipulation and analysis.
It is also easy to use. Almost every Pandas method returns a (modified) copy of the data, which allows you to chain transformations, and perform complex modifications in one line. The overview is divided into sections, each one with code examples and an explanation of what is being done.
First we’ll need to make some imports, which will be necessary through all the examples. We’ll use the well known Titanic dataset (available in Seaborn), which holds data of the Titanic passengers, such as their age, paid fare, and if they survived or not.
Basic aspects
These are must knows that will make your life easier when dealing with Pandas for the first time.
Data Structures
Pandas’ data structures can hold mixed typed values as well as labels, and their axes can have names set. The data structures are the following.
The most basic Data Structure available in Pandas is the Series. This is basically a 1-dimensional labeled array. Therefore, Series have only one axis (axis == 0) called “index”.
Then, we have DataFrames. These are 2-dimensional structures, with two axes, the “index” axis (axis == 0), and the “columns” axis (axis == 1). DataFrames can be thought of as Python dictionaries where the keys are the column labels, and the values are the column Series.
Lastly, we have Panels. These are 3-dimensional data structures, that are rarely used, in comparison with DataFrames. Analogously to DataFrames, they can be thought of as Python dictionaries of DataFrames. Instead of “index” and “columns”, Panels’ axes are named as follow:
items(axis == 0)
major_axis(
axis== 1)
minor_axis(
axis== 2)
The axes distinction is vital, since a lot of methods need to have this specified properly in order to work as expected. We’ll see it’s usage in following examples.
From here on, we will use the Series/DataFrame as the data structure of choice in the examples when explaining things. Keep in mind that anything that applies to Series probably applies to DataFrames too, but it may not be the case the other way around.
Getting information and basic calculations
Pandas offers some methods to get information of a data structure:
info, index, columns, axes, where you can see the memory usage of the data, information about the axes such as the data types involved, and the number of not-null values.
It also offers methods to do basic calculations such as
count, mean, max, min, cumsum, imax, imin.
There’s a method called
describe which is great, because it does almost all of the calculations just mentioned, and presents them in a nicely formatted table.
We also have sorting and ranking methods. For example, to sort the data based on its index, or on any column we have
sort_index and
sort_values. To calculate the ranking of a row based on a value, there’s the
rank method, which is pretty straightforward to use.
In order to perform operations between data structures, we can use arithmetic and boolean operators. Amongst them we have
+, -, *, /, **, &, |, ~, etc.
An alternative to some of these are the
div,
add,
multiply and
subtract methods, which are equivalent to the corresponding operators, but with
fill_value support in case of
NaN values.
Accessing and setting data
There are several methods to retrieve data from a DataFrame, they can be separated by whether they return a scalar/subset of data, or if they are label/position based.
There’s a method which is label and posibion based, the
ix method. It returns a subset of data and is primarily label based, with a fallback to integer positional access, with an exception. If the axis’ labels are integer typed, then the method won’t fall back to integer positional access. In this case, if you want to use positions, you need to use the
iat or
iloc methods.
Update (2017-08-28): The Pandas indexer method
ix was deprecated in release 0.20.0/0.20.1 since its capability of working both with
position and label based indexes caused confusion to users over the years. It’s now recommended to use
at/loc with label based indexes and
iat/iloc with position based indexes. For this reason, the
ix invocations have been commented out.
Positional access allows you to slice a DataFrame, just like you do with lists in Python. Slice notation works in the
iat,
iloc, and
ix methods.
These methods can also be used to set data, from scalars to a set of rows.
Boolean Queries
If you want to get a subset of the data but you don’t know the associated labels/positions, a boolean query is what you want.
To make a boolean query, you need to pass the DataFrame a True/False Series whose index aligns with the DataFrame being queried’s index. That is, for every label in the DataFrame, there must be True/False value associated to that same label in the Series.
The True/False values in the Series indicate if the value associated to a certain label should be returned in the query or not. In this example, we run a boolean query on a Series, instead of a DataFrame, and of course, it worked as expected.
The Series can be assembled combining other boolean queries (the query “terms”) using boolean operators.
Example query
- The query has four
True/Falseseries (query terms) combined with the
&(element to element) operator.
- The first Series was constructed using the
titanic.sexSeries ([‘female’, ‘male’, ‘female’…]) and comparing each value to the ‘female’ string, returning
True/Falsefor each value.
- The second Series used the
pandas.Series.isinfunction (which is pretty straightforward), and also accesses the series using square brackets for a reason; “class” is a reserved word in Python for defining classes, and using it with the dot notation raises a
SyntaxErrorexception.
- The third and fourth “terms” work just like the first one.
Some cool features
Indexing
One of the benefits of Pandas data structures being in-memory is that it allows fast indexing.
You can set, and remove a DataFrame’s index with the methods
set_index, and
reset_index respectively.
There’s a method called
reindex that allows you to conform a DataFrame to a new index both for rows and columns, with an optional filling logic (by default, new row/column labels will have
NaN values associated in the new DataFrame).
If the original DataFrame has duplicated labels in its row/column index then reindex will fail. You will need to either reset the index prior to the reindexing, or remove rows/columns with duplicated labels.
Alignment
One property of data stored in Pandas’ data structures is that it is aligned to the structure’s axes. This means that, when performing an operation, between two data structures, what matters is the associated label, and not the order of the data.
Example
As we can see, the method took values whose labels matched in order to perform the sum in spite of them being in different positions in the original Series. The result Series’ index is aligned with both the series s1 and s2.
Pandas offers a way to perform aligning of data structures artificially, the
align method.
Calling the align method with the two series from the previous example will result in a tuple of two new Series, the series s1 with its index extended containing the values in s2’s index, and the series s2 extended in the same way.
When aligning structures of higher dimensions, you can specify the axis along which the alignment will be made. For example, when aligning two DataFrames you can align along the ‘rows’ (axis=0) axis, the ‘columns’ (axis=1) axis or both (default behaviour).
Combining Data Frames
Pandas offers several methods to combine DataFrames, that can be separated into two approaches, which are concatenation and merging.
Concatenate DataFrames basically means “sticking them together” along an axis, regardless of the values contained within them. Pandas’ main method for concatenation is
concat, although you can use
append too!
concat receives a list of DataFrames and uses axis = 0 (‘rows’) by default, that means that it sticks one DataFrame “below” the ones before. If axis = 1 (‘columns’) is passed, then it will stick each DataFrame at the right side of the ones before.
concat, as opposed to several other methods, is only available in the main pandas namespace.
A namespace basically is “where a method or attribute” lives. This means, that e.g.
concat, can not be invoked from a DataFrame object (it does not live in the
pandas.DataFrame namespace).
Merging is another way of combining DataFrames, but unlike
concat it combines them looking for matching values in columns of said DataFrames (you can merge by index too).
The main method to perform merging in Pandas is
merge which lives both in the main pandas namespace and in the DataFrame namespace (unlike
concat).
Its behaviour as is is just like an INNER JOIN in SQL
Example: To show how to merge two DataFrames and some convenient arguments the
merge function has, I’ll create a
towns_df DataFrame to merge with the titanic dataset DataFrame.
Merge example
left_onand
right_on: We want to combine both DataFrames’ rows when there’s a match between the “embark_town” column in the titanic DataFrame, and the “name” column in the
towns_dfDataFrame, so we indicate this with the
left_onand
right_onarguments.
- how: By default,
mergeperforms an inner join, but we don’t want to lose the rows in the original DataFrame that didn’t have an associated embark_town value, so we indicate that we want a
"left"join with the
howparameter.
indicator: We also want to know, for each row, if there was a match on both DataFrames, or if the resulting row was only on either one. The indicator parameter adds a “_merge” column that indicates this.
suffixes: Since we have “age” columns in both DataFrames, we want to be able to distinguish them properly in the resulting merged DataFrame, we add proper suffixes for this purpose, and get “age_passenger” and “age_city” columns in the merged DataFrame as a result.
GroupBy
Pandas’ GroupBy is exactly what you’d expect and much more. It is not just a
groupby method that works like SQL’s “GROUP BY” but a whole set of methods to perform splitting into groups, transforming them (perhaps independently) and combining the results. You should definitely check out the Group By: split-apply-combine section in the Pandas docs to really get to know (and appreciate) Pandas’ GroupBy capabilities.
The main method for grouping data is
groupby. It exists in the
pandas.DataFrame namespace so you can invoke it directly from a DataFrame object, simply by passing a list of the columns you wish to group the DataFrame by.
You can group by any axis. Of course, by default the grouping is made via the index (rows) axis, but you could group by the columns axis.
The
groupby method is lazy, that is, it doesn’t really perform the data splitting until the group is really needed, which is the most practical/efficient way to go in the majority of cases.
In the example, I’ll show a really cool Pandas method called
cut that will allow us to bin the data according to a column. Then we’ll split the titanic dataset into several groups with
groupby.
- In the example, we want to classify each titanic passenger by their age. Firstly we define the bins, i.e. the values that will delimit the age intervals we want.
- Secondly, we define the labels, i.e. the name of each bin/interval
- Thirdly, we use
pandas.cutto map the titanic’s passengers ages to a label.
- Finally, we set the result Series from the last step as a new column in our titanic DataFrame
Now, we can use the newly created “age_group” column to group by the titanic passengers.
“groups” is an object of type DataFrameGroupBy. It can be iterated in order to inspect or work with the different groups (each one a DataFrame object)
You can also use the
get_group method (in this case passing it a 2-sized tuple) to get the corresponding DataFrame group.
The first thing I’d want to see about the resulting groups is the size of each one. To do this you can use the
size method, which will return a multi-level Series. Each level corresponds to a grouping field, and the values correspond to the number of rows associated to the group.
Since this result is a Series, we can easily get the size of each group in relation to the whole dataset.
Now we can see very easily that, for example, most children did not travel alone.
Calculated columns
When you’re working with Pandas, there is something you most certainly will want to do, and that is adding a column with calculated values to your DataFrame.
There’re several ways to do this, in fact we’ve already done it with
pandas.cut in the “Group By” section, but that was a particular case.
A more general approach is to use the
apply function. If we wanted to add a column “is_old” with boolean values, for every row (passenger) of the titanic, we should do the following
What we did there was to create a Series by applying a custom function to each row of the DataFrame. Then we assigned the series to a new column named “is_old” in the titanic DataFrame.
Notice how we use the axis argument in the
DataFrame.apply invocation. A good way to remember how to use this argument is the following:
axis='columns'makes the custom function receive a Series with one value per column (i.e. a row) in each invocation.
axis='rows'makes the custom function receive a Series with one value per row (i.e. a column) in each invocation.
This approach is good if we need to use multiple values of a row. But in this case, we only use the “age” value of every row. So, in this case it would seem unnecessary to use
apply for the whole DataFrame.
Series.applydoes not receive an
axisargument since Series have only one axis.
- We defined a new
is_old_func_seriescustom function since
Series.applypasses scalars (the Series values) to the custom function, and trying to access the attribute “age” of a scalar would raise an
AttributeErrorexception.
- This approach seems to just use the necessary data for the calculation, and as we’ll see it is orders of magnitude faster.
In the presence of “a lot” of rows (more than 10.000) and simple calculations for the new column, there’s an alternative whose execution time is orders of magnitude lower than
Series.apply's. This alternative is the
eval method.
eval is extremely fast in this case, because it performs arithmetic and boolean expression evaluations “all at once” using the underlying engine numexpr, which is a high performance numerical evaluator for NumPy.
Its syntax is a little more friendly than that of
apply, but for now does not support if operations. To do assignments to a subset of the DataFrame you could use a combination of
loc/ix and
eval.
Reshaping
Reshaping is, broadly speaking transforming the structure of the data to make it suitable for further analysis. In the context of Pandas, we can reshape a DataFrame by using one column’s values as the index, and another column’s values as new columns, this is called pivoting.
Pivoting
There are two main ways to apply pivoting in Pandas, the
pivot and
pivot_table methods.
The
pivot function is more restrictive than
pivot_table since it needs the DataFrame’s column set as “index” to have unique values only.
Our data doesn’t fit the
pivot input quite properly, which is “stacked” or “record” formatted data (as indicated in the Pandas docs), but for the sake of demonstrating its usage, we’ll tweak our titanic DataFrame a little bit.
We first removed all rows with duplicated “age” values (keeping the first appearance of each one) and pivoted that DataFrame. The index of that new DataFrame are the labels in the “age” column (all unique now), the columns of the new DataFrame are the unique labels of the “class” column in the original DataFrame, and the values are the corresponding “fare” values.
A better alternative for this is to use the
pivot_table method, since it saves us from having to drop duplicates (which in this case makes no sense), and it aggregates the DataFrame values (by default performs an average).
In the
pivot example we have only one row with “age” == 71 and “class” == “First”, therefore the value ( 34.6542 ) is the “fare” value of that row. Whilst in the
pivot_table example, we have several rows with “age” == 71 and “class” == “First”, and the value ( 42.0792 ) is the average of the “fare” values of those rows.
Something cool we could calculate, for example, is the median of the paid fares per embarking town, per age group. We can do this easily with
pivot_table in one line. We need to indicate that the aggregating function to be used is
np.median.
The inverse process of pivoting, unpivoting we might call it, is implemented in Pandas by the
pandas.melt function.
This function takes a DataFrame and “melts it”, that is, it takes one or more columns and uses them as “indentifier variables” (keeps them the way they are). The rest of the columns are “converted” into two “unidentifier (or measured) variable” columns “variable” and “value”.
In the example:
p_titanic.reset_index(): We do this, because
p_titanicdoes not have a column named “age” with the age values, but an index named “age” with the age values, and for the unpivoting process we’re going to want to have the “age” column.
id_vars='age': This indicates that our “identifying variable” is going to be the column “age” (we can have more than one identifying variable/column when melting)
- var_name: This argument simply sets the name for the “variable” column. We want it to be named “class_renamed”. If None, the
p_titanic.reset_index().columns.name(“class” in this case) value is used, and if that’s not set, the column will just be named “variable”.
value_vars: These are the columns that are going to be “unpivoted” to the rows axis, and are going to be values of the “variable” column. If not set, the method considers all non identifying variable columns.
value_name: This simply sets the name for the “value” column. We want it to be named “fare”. If None, the name “value” is used.
Stacking
We can also reshape a DataFrame by stacking/unstacking. This is related to Hierarchical indexes or MultiIndexes, which are a way to represent higher (than 1 or 2) dimensional data in Series or DataFrames.
If you recall the groupby snippet where we called
DataFrameGroupBy.size, the result of that invocation was a Series with a MultiIndex with two levels, the first one (level 0) corresponding to the “age_group”, and the second one (level 1) corresponding to the “alone” flag.
You can see how MultiIndexes work very easily, simply by removing the
values argument in the
pivot/pivot_table examples. This will create a DataFrame with MultiIndexed columns with two levels; the first one being the value (“fare”, “age”, “survived”, etc.) and the second one, the “age_group”. Therefore, you can get an equivalent result with the following line:
But back to the point, stacking and unstacking. Stacking is to take a level of the columns index and transfer it to the rows index. Therefore, unstacking is the opposite; transferring a level from the rows level to the columns level.
This works even with DataFrames with single level axes. Stacking or unstacking a DataFrame of this characteristics would turn it into a MultiIndex Series!
We can take the MultiIndex Series from the groupby snippet mentioned earlier, and unstack any of its levels.
By default, stack/unstack takes the highest level, in this case the level 1, but we could e.g. unstack the level 0. You can try running stack on the results of these examples to see it in action.
TimeSeries
Last but not least, Pandas offers a lot of high level methods to work with TimeSeries. These methods are as powerful as they’re useful, and they deserve a blog post on their own.
Nevertheless, they shouldn’t pass unnoticed in this overview. TimeSeries are basically Series where the index is of a special type (DateTimeIndex), and the labels are timestamps!
With Pandas, you can resample these TimeSeries (upsample or downsample), generate Date ranges, perform non-standard increments (e.g. shift X number of business days), and much more.
We recommend you check out the TimeSeries section in the Pandas docs whenever you need a reference, as well as reading some of the great blog posts out there, where you’ll see most of these features applied in cool and meaningful ways to real data.
Seaborn
If you’ve gotten this far, first of all congratulations 😁. Secondly, don’t worry, we won’t be giving you an extensive description of Seaborn’s features. Instead we’ll just go over what it is, some of its benefits, and show you some cool plots you can make with it.
Seaborn is a Python library for making statistical visualizations. It’s built to provide eye candy plots and at the same time it makes developers’ life easier. We’ve all been in the situation where we needed to build a simple decent looking histogram and ended up making several function invocations and setting arguments without fully knowing what we’re doing.
Seaborn tackles this not by reinventing the wheel but by improving it. It is built on top of Matplotlib and provides a high level API that makes “a well-defined set of hard things easy” (as stated in the docs), amongst other things by making that its methods work greatly by passing a minimal set of arguments.
The eye candy comes from the built in themes, the possibility to build custom attractive color palettes, and the witty way they’re utilized to display statistical sugar (e.g. the kernel density estimation in a violin plot)
Seaborn is part of the PyData stack, and accepts Pandas’ data structures as inputs in its API (thank goodness 😄)
Update (2017-08-28): In the Seaborn examples, we access Matplotlib through Seaborn when doing
sns.plt.show(). Seaborn depends on Matplotlib, but accessing a library as a submodule of another library that imports it is a bad practice. This behavior was prohibited in release 0.8.0 of Seaborn. Nevertheless, when doing it properly (through Matplotlib), the plots look a little different. Information on how to configure plots’ aesthetics in Seaborn is available here.
Let’s get to the plots!
distplot: The first thing you want to see when exploring your data is the distribution of your variables. For example, let’s see the Titanic’s passengers’ ages distribution.
Distribution of Titanic's passengers' ages.
- If we want to see the raw number of rows in each bin, we can pass
kde=False(kernel density estimation = False)
We need to drop
NaNvalues for
distplotnot to raise a ValueError exception.
FacetGrid: If we wanted to break down a plot (e.g. the last one) by some categories, we needn’t perform boolean queries, nor groupbys, we can use
FacetGrid.
Distribution of ages in groups of passengers
jointplot: This method is used to display data points according to two variables, along with both their distributions, kernel density estimators, and an optional regression that fits the data. With
regwe indicate that we want a regression fit to the data.
- In this case, although there appears to be a small tendency upwards shown by the regression, there appears to be almost no correlation between the variables “age” and “fare”, as shown by the Pearson correlation coefficient.
Data points by their "age" and "fare" values, along with the independent variable distributions
heatmap: Heatmaps are ideal to plot “rectangular data” such as matrixes. They’re great to visualize when some values, or calculated values, such as averages, counts, etc. are more extreme.
- We can take the
pt_titanicDataFrame from the
pivot_table, which held data of the median fares paid by passengers per embark_town per age_group, and build a heatmap very easily. Most times, we like out heatmaps annotated to catch some subtelties that may pass by me with the colors. The
fmtvalue is pretty straightforward.
Annotated heatmap of the paid fares' median by groups of passengers
Finally, something really cool that you can put into a heatmap is a correlation matrix. Pandas DataFrame has a
corrmethod that calculates Pearson’s (can be another) correlation coefficient between all couples of numeric columns of the DataFrame.
Annotated heatmap of Pearson correlation coefficients between variables
This’s it! Hopefully this post was useful for you to get to know Pandas in a gradual and ordered way, and motivates you to dig deeper into the features without fear. With much less detail, the post also attempted to give a glance at Seaborn’s beautiful graphs; we highly encourage you to dive into the Seaborn API and the tutorials so as to really get acquainted with it.
Have fun using this tools to better analyze and explore data!
Update (2017-03-21): In a previous version of this post, we were using dot notation when assigning a new column to a DataFrame. As indicated in the comments, using dot notation carelessly for this can cause an undesired overwrite of DataFrame attributes for that object. But even more so, the assignment of a new column this way does not work. What works is accessing/updating an existing column, which is why we probably missed this in the first place._
Thinking of implementing a machine learning project in your organization? Here are 11 essential questions to ask before kicking off an ML initiative.
Like what you read?
Subscribe to our newsletter and get updates on Deep Learning, NLP, Computer Vision & Python.No spam, ever. We'll never share your email address and you can opt out at any time.