The text is released under the CC-BY-NC-ND license, and code is released under the MIT license. If you find this content useful, please consider supporting the work by buying the book!
No changes were made to the contents of this notebook from the original.
An essential piece of analysis of large data is efficient summarization: computing aggregations like sum()
, mean()
, median()
, min()
, and max()
, in which a single number gives insight into the nature of a potentially large dataset.
In this section, we'll explore aggregations in Pandas, from simple operations akin to what we've seen on NumPy arrays, to more sophisticated operations based on the concept of a groupby
.
For convenience, we'll use the same display
magic function that we've seen in previous sections:
In [1]:
import numpy as np
import pandas as pd
class display(object):
"""Display HTML representation of multiple objects"""
template = """<div style="float: left; padding: 10px;">
<p style='font-family:"Courier New", Courier, monospace'>{0}</p>{1}
</div>"""
def __init__(self, *args):
self.args = args
def _repr_html_(self):
return '\n'.join(self.template.format(a, eval(a)._repr_html_())
for a in self.args)
def __repr__(self):
return '\n\n'.join(a + '\n' + repr(eval(a))
for a in self.args)
Here we will use the Planets dataset, available via the Seaborn package (see Visualization With Seaborn). It gives information on planets that astronomers have discovered around other stars (known as extrasolar planets or exoplanets for short). It can be downloaded with a simple Seaborn command:
In [2]:
import seaborn as sns
planets = sns.load_dataset('planets')
planets.shape
Out[2]:
In [3]:
planets.head()
Out[3]:
This has some details on the 1,000+ extrasolar planets discovered up to 2014.
Earlier, we explored some of the data aggregations available for NumPy arrays ("Aggregations: Min, Max, and Everything In Between").
As with a one-dimensional NumPy array, for a Pandas Series
the aggregates return a single value:
In [4]:
rng = np.random.RandomState(42)
ser = pd.Series(rng.rand(5))
ser
Out[4]:
In [5]:
ser.sum()
Out[5]:
In [6]:
ser.mean()
Out[6]:
For a DataFrame
, by default the aggregates return results within each column:
In [7]:
df = pd.DataFrame({'A': rng.rand(5),
'B': rng.rand(5)})
df
Out[7]:
In [8]:
df.mean()
Out[8]:
By specifying the axis
argument, you can instead aggregate within each row:
In [9]:
df.mean(axis='columns')
Out[9]:
Pandas Series
and DataFrame
s include all of the common aggregates mentioned in Aggregations: Min, Max, and Everything In Between; in addition, there is a convenience method describe()
that computes several common aggregates for each column and returns the result.
Let's use this on the Planets data, for now dropping rows with missing values:
In [10]:
planets.dropna().describe()
Out[10]:
This can be a useful way to begin understanding the overall properties of a dataset.
For example, we see in the year
column that although exoplanets were discovered as far back as 1989, half of all known expolanets were not discovered until 2010 or after.
This is largely thanks to the Kepler mission, which is a space-based telescope specifically designed for finding eclipsing planets around other stars.
The following table summarizes some other built-in Pandas aggregations:
Aggregation | Description |
---|---|
count() |
Total number of items |
first() , last() |
First and last item |
mean() , median() |
Mean and median |
min() , max() |
Minimum and maximum |
std() , var() |
Standard deviation and variance |
mad() |
Mean absolute deviation |
prod() |
Product of all items |
sum() |
Sum of all items |
These are all methods of DataFrame
and Series
objects.
To go deeper into the data, however, simple aggregates are often not enough.
The next level of data summarization is the groupby
operation, which allows you to quickly and efficiently compute aggregates on subsets of data.
Simple aggregations can give you a flavor of your dataset, but often we would prefer to aggregate conditionally on some label or index: this is implemented in the so-called groupby
operation.
The name "group by" comes from a command in the SQL database language, but it is perhaps more illuminative to think of it in the terms first coined by Hadley Wickham of Rstats fame: split, apply, combine.
This makes clear what the groupby
accomplishes:
DataFrame
depending on the value of the specified key.While this could certainly be done manually using some combination of the masking, aggregation, and merging commands covered earlier, an important realization is that the intermediate splits do not need to be explicitly instantiated. Rather, the GroupBy
can (often) do this in a single pass over the data, updating the sum, mean, count, min, or other aggregate for each group along the way.
The power of the GroupBy
is that it abstracts away these steps: the user need not think about how the computation is done under the hood, but rather thinks about the operation as a whole.
As a concrete example, let's take a look at using Pandas for the computation shown in this diagram.
We'll start by creating the input DataFrame
:
In [11]:
df = pd.DataFrame({'key': ['A', 'B', 'C', 'A', 'B', 'C'],
'data': range(6)}, columns=['key', 'data'])
df
Out[11]:
The most basic split-apply-combine operation can be computed with the groupby()
method of DataFrame
s, passing the name of the desired key column:
In [12]:
df.groupby('key')
Out[12]:
Notice that what is returned is not a set of DataFrame
s, but a DataFrameGroupBy
object.
This object is where the magic is: you can think of it as a special view of the DataFrame
, which is poised to dig into the groups but does no actual computation until the aggregation is applied.
This "lazy evaluation" approach means that common aggregates can be implemented very efficiently in a way that is almost transparent to the user.
To produce a result, we can apply an aggregate to this DataFrameGroupBy
object, which will perform the appropriate apply/combine steps to produce the desired result:
In [13]:
df.groupby('key').sum()
Out[13]:
The sum()
method is just one possibility here; you can apply virtually any common Pandas or NumPy aggregation function, as well as virtually any valid DataFrame
operation, as we will see in the following discussion.
The GroupBy
object is a very flexible abstraction.
In many ways, you can simply treat it as if it's a collection of DataFrame
s, and it does the difficult things under the hood. Let's see some examples using the Planets data.
Perhaps the most important operations made available by a GroupBy
are aggregate, filter, transform, and apply.
We'll discuss each of these more fully in "Aggregate, Filter, Transform, Apply", but before that let's introduce some of the other functionality that can be used with the basic GroupBy
operation.
In [14]:
planets.groupby('method')
Out[14]:
In [15]:
planets.groupby('method')['orbital_period']
Out[15]:
Here we've selected a particular Series
group from the original DataFrame
group by reference to its column name.
As with the GroupBy
object, no computation is done until we call some aggregate on the object:
In [16]:
planets.groupby('method')['orbital_period'].median()
Out[16]:
This gives an idea of the general scale of orbital periods (in days) that each method is sensitive to.
In [17]:
for (method, group) in planets.groupby('method'):
print("{0:30s} shape={1}".format(method, group.shape))
This can be useful for doing certain things manually, though it is often much faster to use the built-in apply
functionality, which we will discuss momentarily.
Through some Python class magic, any method not explicitly implemented by the GroupBy
object will be passed through and called on the groups, whether they are DataFrame
or Series
objects.
For example, you can use the describe()
method of DataFrame
s to perform a set of aggregations that describe each group in the data:
In [18]:
planets.groupby('method')['year'].describe().unstack()
Out[18]:
Looking at this table helps us to better understand the data: for example, the vast majority of planets have been discovered by the Radial Velocity and Transit methods, though the latter only became common (due to new, more accurate telescopes) in the last decade. The newest methods seem to be Transit Timing Variation and Orbital Brightness Modulation, which were not used to discover a new planet until 2011.
This is just one example of the utility of dispatch methods.
Notice that they are applied to each individual group, and the results are then combined within GroupBy
and returned.
Again, any valid DataFrame
/Series
method can be used on the corresponding GroupBy
object, which allows for some very flexible and powerful operations!
The preceding discussion focused on aggregation for the combine operation, but there are more options available.
In particular, GroupBy
objects have aggregate()
, filter()
, transform()
, and apply()
methods that efficiently implement a variety of useful operations before combining the grouped data.
For the purpose of the following subsections, we'll use this DataFrame
:
In [19]:
rng = np.random.RandomState(0)
df = pd.DataFrame({'key': ['A', 'B', 'C', 'A', 'B', 'C'],
'data1': range(6),
'data2': rng.randint(0, 10, 6)},
columns = ['key', 'data1', 'data2'])
df
Out[19]:
In [20]:
df.groupby('key').aggregate(['min', np.median, max])
Out[20]:
Another useful pattern is to pass a dictionary mapping column names to operations to be applied on that column:
In [21]:
df.groupby('key').aggregate({'data1': 'min',
'data2': 'max'})
Out[21]:
In [22]:
def filter_func(x):
return x['data2'].std() > 4
display('df', "df.groupby('key').std()", "df.groupby('key').filter(filter_func)")
Out[22]:
The filter function should return a Boolean value specifying whether the group passes the filtering. Here because group A does not have a standard deviation greater than 4, it is dropped from the result.
While aggregation must return a reduced version of the data, transformation can return some transformed version of the full data to recombine. For such a transformation, the output is the same shape as the input. A common example is to center the data by subtracting the group-wise mean:
In [23]:
df.groupby('key').transform(lambda x: x - x.mean())
Out[23]:
The apply()
method lets you apply an arbitrary function to the group results.
The function should take a DataFrame
, and return either a Pandas object (e.g., DataFrame
, Series
) or a scalar; the combine operation will be tailored to the type of output returned.
For example, here is an apply()
that normalizes the first column by the sum of the second:
In [24]:
def norm_by_data2(x):
# x is a DataFrame of group values
x['data1'] /= x['data2'].sum()
return x
display('df', "df.groupby('key').apply(norm_by_data2)")
Out[24]:
apply()
within a GroupBy
is quite flexible: the only criterion is that the function takes a DataFrame
and returns a Pandas object or scalar; what you do in the middle is up to you!
In [25]:
L = [0, 1, 0, 1, 2, 0]
display('df', 'df.groupby(L).sum()')
Out[25]:
Of course, this means there's another, more verbose way of accomplishing the df.groupby('key')
from before:
In [26]:
display('df', "df.groupby(df['key']).sum()")
Out[26]:
In [27]:
df2 = df.set_index('key')
mapping = {'A': 'vowel', 'B': 'consonant', 'C': 'consonant'}
display('df2', 'df2.groupby(mapping).sum()')
Out[27]:
In [28]:
display('df2', 'df2.groupby(str.lower).mean()')
Out[28]:
In [29]:
df2.groupby([str.lower, mapping]).mean()
Out[29]:
In [30]:
decade = 10 * (planets['year'] // 10)
decade = decade.astype(str) + 's'
decade.name = 'decade'
planets.groupby(['method', decade])['number'].sum().unstack().fillna(0)
Out[30]:
This shows the power of combining many of the operations we've discussed up to this point when looking at realistic datasets. We immediately gain a coarse understanding of when and how planets have been discovered over the past several decades!
Here I would suggest digging into these few lines of code, and evaluating the individual steps to make sure you understand exactly what they are doing to the result. It's certainly a somewhat complicated example, but understanding these pieces will give you the means to similarly explore your own data.