Altcademy - a Forbes magazine logo Best Coding Bootcamp 2023

How to use groupby in Pandas

Understanding GroupBy in Pandas

When you're diving into data analysis with Python, one of the most powerful tools at your disposal is the Pandas library. It's like a Swiss Army knife for data manipulation and analysis. One of the essential functionalities provided by Pandas is the groupby operation, which allows you to group large amounts of data and compute operations on these groups.

What is GroupBy?

Imagine you're sorting a collection of colored balls into buckets where each bucket is dedicated to one color. This is essentially what groupby does; it sorts data into groups based on some criteria. After grouping the data, you can apply a function to each group independently, such as summing up numbers, calculating averages, or finding the maximum value.

Simple GroupBy Example

Let's start with a simple example. Suppose you have a dataset of students with their respective grades in different subjects. Your task is to find the average grade for each subject. Here's how you can do that using groupby in Pandas:

import pandas as pd

# Create a DataFrame
data = {
    'Subject': ['Math', 'Science', 'Math', 'Science', 'English', 'English'],
    'Grade': [90, 80, 85, 88, 92, 95]

df = pd.DataFrame(data)

# Group the data by the 'Subject' column and calculate the mean grade for each subject
grouped = df.groupby('Subject')
average_grades = grouped.mean()


When you run this code, Pandas groups the grades by subject and then calculates the average grade for each group:

English   93.5
Math      87.5
Science   84.0

How Does GroupBy Work?

To understand how groupby works, let's break it down into steps:

  1. Split: The groupby function starts by splitting the DataFrame into groups based on the given criteria (e.g., the 'Subject' column in our example).
  2. Apply: Then, it applies a function to each group independently (e.g., calculating the mean of grades).
  3. Combine: Finally, it combines the results into a new DataFrame where the index is the groups and the columns are the computed values.

Digging Deeper: GroupBy With Multiple Columns

You can also group by multiple columns. Let's say you want to find the average grade for each subject, separated by gender. Here's how you would do it:

# Add a 'Gender' column to our dataset
data['Gender'] = ['Female', 'Male', 'Female', 'Male', 'Female', 'Male']

df = pd.DataFrame(data)

# Group by both 'Subject' and 'Gender'
grouped = df.groupby(['Subject', 'Gender'])
average_grades = grouped.mean()


The output will show the average grades for each subject, separated by gender:

Subject Gender       
English Female   92.0
        Male     95.0
Math    Female   87.5
        Male      NaN
Science Female    NaN
        Male     84.0

Here, NaN (Not a Number) indicates that there were no data points for that particular group.

Applying Different Functions to Groups

You don't have to limit yourself to calculating the mean. You can apply different functions to your groups:

# Calculate different statistics for each subject
max_grades = grouped.max()
min_grades = grouped.min()
sum_grades = grouped.sum()

print("Maximum Grades:\n", max_grades)
print("\nMinimum Grades:\n", min_grades)
print("\nSum of Grades:\n", sum_grades)

More Power With agg() Function

The agg() function, short for aggregate, gives you the ability to apply multiple functions at once to your groups. Here's an example:

# Apply multiple functions to each subject group
statistics = grouped.agg(['mean', 'max', 'min', 'sum'])


This will give you a DataFrame with the mean, maximum, minimum, and sum of the grades for each subject and gender.

GroupBy With Custom Functions

You can also apply your custom functions to groups. Let's say you want to define a function that calculates the range of grades (max - min) for each group:

def grade_range(group):
    return group['Grade'].max() - group['Grade'].min()

range_grades = grouped.apply(grade_range)


This will apply your grade_range function to each group and return the range of grades.

Intuition and Analogies

To help solidify your understanding of groupby, think of it like organizing a library. Books (data) can be grouped by genre (category), and then you can count how many books there are in each genre (applying a function). Similarly, with groupby, you organize your data into categories and then perform operations on each category.


Mastering the groupby function in Pandas can elevate your data analysis skills significantly. It's a bit like learning to sort and organize your thoughts; once you get the hang of it, you'll find it easier to navigate through complex data and extract meaningful insights. Remember that groupby is all about splitting your data into meaningful groups, applying functions to understand those groups better, and then combining the results for analysis. Keep practicing with different datasets and operations, and soon you'll be grouping and analyzing data with confidence and creativity!