使用单个和多个键进行 GroupBy
使用 groupby() 按一个或多个列分组,并应用求和、均值、计数以及最小值/最大值聚合。
使用单个和多个键进行 GroupBy 是 CoddyKit 上的免费 Pandas & NumPy Academy 课时。 这是第 2 节课,共 4 节。 你可以在下方免费阅读本课时的完整内容 — 然后在浏览器中使用内置代码编辑器和全天候 AI 导师进行实践。 这是 Pandas & NumPy Academy 学习路径的一部分,你的进度在网页和 CoddyKit 应用中同步。 Pandas & NumPy Academy 课程共包含 4 节课。
本课时的部分内容尚未翻译,以英文显示。
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.0Choosing 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.500All 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.0Accessing 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.0Using 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.0Counting 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 2Sorting 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.0Chaining 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 62000Grouping 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.0Quick 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.
常见问题解答
「使用单个和多个键进行 GroupBy」课时是免费的吗?
是的 — 「使用单个和多个键进行 GroupBy」的完整文本可在网页上免费阅读。要进行交互式练习(内置代码编辑器和全天候 AI 导师)并解锁 Pandas & NumPy Academy 课程的其余内容,请升级到 CoddyKit PRO。 Pandas & NumPy Academy 课程共包含 4 节课。
「使用单个和多个键进行 GroupBy」这节课中我会学到什么?
使用 groupby() 按一个或多个列分组,并应用求和、均值、计数以及最小值/最大值聚合。 你通过在浏览器中直接运行的动手代码来练习 Pandas & NumPy Academy,全天候 AI 导师会在你学习这节课的过程中回答你的问题。
学习 Pandas & NumPy Academy 需要有经验吗?
无需任何先前经验。CoddyKit 上的 Pandas & NumPy Academy 课程适合初学者到高级学习者,你可以从这里开始或从头开始,按照自己的节奏学习。 这是第 2 节课,共 4 节。
「使用单个和多个键进行 GroupBy」课时需要多长时间?
大多数 CoddyKit 课程大约需要 5–10 分钟。每节课都很精短且互动,所以你能稳步进步,并在网页和应用中从离开的地方继续。
我能在这节 Pandas & NumPy Academy 课中编写并运行代码吗?
能。每节 Pandas & NumPy Academy 课都包含内置代码编辑器,你可以在浏览器中直接编写并运行真实代码,并获得即时 AI 反馈 — 无需本地设置。
此课程中的所有课时
- 拆分—应用—合并模式
- 使用单个和多个键进行 GroupBy
- agg() 方法
- GroupBy 变换与筛选