0Pricing
Pandas & NumPy Academy · Lección

GroupBy con una o varias claves

Agrupe por una o varias columnas con groupby() y aplique las agregaciones sum, mean, count y min/max.

GroupBy con una o varias claves es una lección gratuita de Pandas & NumPy Academy en CoddyKit. Esta es la lección 2 de 4. Puedes leer la lección completa abajo gratuitamente — luego la practicas en el navegador con un editor de código integrado y un tutor de IA 24/7. Forma parte de la ruta de aprendizaje de Pandas & NumPy Academy, y tu progreso se sincroniza en la web y la app de CoddyKit. El curso de Pandas & NumPy Academy incluye 4 lecciones en total.

Partes de esta lección aún no han sido traducidas y se muestran en inglés.

GroupBy with a Single Key

The simplest GroupBy call uses a single column as the grouping key. You call df.groupby('column_name') and chain an aggregation. Pandas creates one group for each unique value in that column and applies the aggregation to every numeric column (or the selected column). This is the most common form of GroupBy in day-to-day analysis.

import pandas as pd

df = pd.DataFrame({
    'dept': ['Eng', 'HR', 'Eng', 'HR', 'Eng'],
    'salary': [90000, 60000, 95000, 62000, 88000],
    'years': [3, 5, 7, 2, 4]
})

print(df.groupby('dept')['salary'].mean())
# dept
# Eng    91000.0
# HR     61000.0

Choosing Which Columns to Aggregate

After calling groupby() you can select one column with bracket notation to get a SeriesGroupBy, or select multiple columns with a list to get a DataFrameGroupBy. If you skip selection entirely, Pandas aggregates all numeric columns, which is convenient but can produce unexpected results when you have irrelevant numeric columns.

# Single column result -> SeriesGroupBy
print(df.groupby('dept')['salary'].sum())

# Multiple columns result -> DataFrameGroupBy
print(df.groupby('dept')[['salary', 'years']].mean())
#        salary  years
# dept
# Eng   91000.0    4.667
# HR    61000.0    3.500

All Common Aggregations

Pandas supports a full set of built-in aggregation methods on GroupBy objects: sum(), mean(), median(), min(), max(), count(), std(), var(), first(), and last(). Each returns a scalar per group, giving you a clean summary table indexed by the grouping key.

g = df.groupby('dept')['salary']
print('sum:   ', g.sum())
print('mean:  ', g.mean())
print('min:   ', g.min())
print('max:   ', g.max())
print('count: ', g.count())
print('std:   ', g.std().round(2))

GroupBy with Multiple Keys

Pass a list of column names to groupby() to group by more than one dimension at once. Each unique combination of values in those columns becomes its own group. The result has a MultiIndex with one level per grouping column, allowing you to drill into cross-dimensional summaries in a single step.

df2 = pd.DataFrame({
    'dept':   ['Eng', 'Eng', 'HR',  'HR',  'Eng', 'HR'],
    'level':  ['L1',  'L2',  'L1',  'L2',  'L1',  'L1'],
    'salary': [80000, 100000, 55000, 70000, 82000, 58000]
})

result = df2.groupby(['dept', 'level'])['salary'].mean()
print(result)
# dept  level
# Eng   L1       81000.0
#       L2      100000.0
# HR    L1       56500.0
#       L2       70000.0

Accessing MultiIndex Results

When you group by multiple keys the result index is a MultiIndex. You can access a specific outer level with .loc['value'] and navigate inner levels with tuples. Calling reset_index() flattens the MultiIndex into regular columns, which is often more convenient for further operations or export.

result = df2.groupby(['dept', 'level'])['salary'].mean()

# Access Engineering rows only
print(result.loc['Eng'])
# level
# L1     81000.0
# L2    100000.0

# Flatten to a regular DataFrame
print(result.reset_index())
#   dept level    salary
# 0  Eng    L1   81000.0
# 1  Eng    L2  100000.0

Using as_index=False

By default, the grouping columns become the index of the result. Passing as_index=False keeps them as regular columns instead, giving you a flat DataFrame that is easier to pass to further Pandas operations or to export. This is equivalent to calling reset_index() on the result.

flat = df2.groupby(['dept', 'level'], as_index=False)['salary'].mean()
print(flat)
#   dept level    salary
# 0  Eng    L1   81000.0
# 1  Eng    L2  100000.0
# 2   HR    L1   56500.0
# 3   HR    L2   70000.0

Counting Rows Per Group

count() returns the number of non-null values in each group. If you want the total number of rows per group regardless of nulls, use size() instead. This distinction matters when your data has missing values, because count() will undercount groups that have NaN entries.

import numpy as np

df3 = pd.DataFrame({
    'dept': ['Eng', 'Eng', 'HR', 'HR'],
    'bonus': [5000, np.nan, 3000, 4000]
})

print(df3.groupby('dept')['bonus'].count())  # ignores NaN
# dept
# Eng    1
# HR     2

print(df3.groupby('dept')['bonus'].size())   # includes NaN rows
# dept
# Eng    2
# HR     2

Sorting GroupBy Results

By default, GroupBy sorts the result by the group key. You can disable sorting with sort=False to preserve the original order of first appearance, which is slightly faster on large datasets. Once you have the result as a DataFrame, you can sort it further with sort_values() on any column.

result = (df.groupby('dept', sort=False)['salary']
            .mean()
            .reset_index()
            .sort_values('salary', ascending=False))
print(result)
#   dept    salary
# 0  Eng   91000.0
# 1   HR   61000.0

Chaining GroupBy with Other Methods

GroupBy results are regular Series or DataFrames, so you can chain further Pandas methods on them immediately. A common pattern is to aggregate, then reset_index(), then rename columns, then sort_values() — all in a single readable method chain without storing intermediate variables.

summary = (
    df
    .groupby('dept')['salary']
    .agg(['mean', 'count'])
    .rename(columns={'mean': 'avg_salary', 'count': 'headcount'})
    .reset_index()
    .sort_values('avg_salary', ascending=False)
)
print(summary)

Applying sum, mean, count Together

You frequently need more than one statistic per group. Pass a list of function names to .agg() to compute several at once. The result is a DataFrame with a column for each function. This is more efficient than calling each aggregation separately because Pandas processes all of them in a single pass over the data.

result = df.groupby('dept')['salary'].agg(['sum', 'mean', 'count', 'max'])
print(result)
#           sum      mean  count    max
# dept
# Eng    273000   91000.0      3  95000
# HR     122000   61000.0      2  62000

Grouping by Categorical Columns

When a grouping column has the Categorical dtype, Pandas includes all categories in the result by default, even those with no rows. This can be useful to ensure your summary table always shows every category, but it creates rows with NaN or 0 for empty groups. Control this with the observed parameter (set to True to skip empty categories).

df['dept'] = df['dept'].astype('category')
# With observed=True, only groups that appear in data are included
print(df.groupby('dept', observed=True)['salary'].mean())
# dept
# Eng    91000.0
# HR     61000.0

Quick Check

Test your understanding of GroupBy with single and multiple keys from this lesson.

Lesson Recap

In this lesson you learned: how to group by a single column and apply common aggregations, how to group by multiple columns to produce cross-dimensional summaries, and the difference between count() and size() when nulls are present. Next up we explore the powerful agg() method for computing multiple different statistics in a single call.

Preguntas frecuentes

¿La lección «GroupBy con una o varias claves» es gratis?

Sí — el texto completo de «GroupBy con una o varias claves» es gratis para leer aquí en la web. Para practicarla de forma interactiva (editor de código integrado y tutor de IA 24/7) y desbloquear el resto del curso de Pandas & NumPy Academy, actualiza a CoddyKit PRO. El curso de Pandas & NumPy Academy incluye 4 lecciones en total.

¿Qué aprenderé en «GroupBy con una o varias claves»?

Agrupe por una o varias columnas con groupby() y aplique las agregaciones sum, mean, count y min/max. Practicas Pandas & NumPy Academy con código real que ejecutas directamente en el navegador, y un tutor de IA 24/7 responde tus preguntas mientras trabajas en la lección.

¿Necesito experiencia previa para empezar Pandas & NumPy Academy?

No se requiere experiencia previa. Pandas & NumPy Academy en CoddyKit está estructurado para principiantes hasta estudiantes avanzados, así que puedes empezar aquí o desde el inicio y avanzar a tu ritmo. Esta es la lección 2 de 4.

¿Cuánto tiempo toma la lección «GroupBy con una o varias claves»?

La mayoría de las lecciones de CoddyKit toman alrededor de 5–10 minutos. Cada una es compacta e interactiva, así que avanzas constantemente y retomas exactamente por donde dejaste en la web y la app.

¿Puedo escribir y ejecutar código en esta lección de Pandas & NumPy Academy?

Sí. Cada lección de Pandas & NumPy Academy incluye un editor de código integrado, así que escribes y ejecutas código real directamente en tu navegador y obtienes retroalimentación instantánea de IA — sin configuración local necesaria.

Todas las lecciones de este curso

  1. El patrón Split-Apply-Combine
  2. GroupBy con una o varias claves
  3. El método agg()
  4. Transform y Filter de GroupBy
← Volver a Pandas & NumPy Academy