如何在 Pandas groupby 多列聚合时同时添加分组计数列

阿墨小哥_4102

阿墨小哥_4102

2026-10-10

899人浏览

原创

如何在 Pandas groupby 多列聚合时同时添加分组计数列

本文详解如何在 Pandas 中对多列进行分组(groupby),对指定数值列执行不同聚合(如均值、最小值),并原生、高效地添加分组计数列(count),全程不依赖 apply,保持向量化性能。

本文详解如何在 pandas 中对多列进行分组(groupby),对指定数值列执行不同聚合(如均值、最小值),并**原生、高效地添加分组计数列(count)**,全程不依赖 `apply`,保持向量化性能。

在使用 pandas.DataFrame.groupby() 进行多列分组与聚合时,常需同时计算统计指标(如均值、最小值)和每组样本数量。关键在于:count 是一种标准聚合函数,可与其他 NamedAgg 并列定义,无需额外 apply 或 size() 等易混淆操作。

以您的数据为例——筛选 2018 年及以后的纯电动车(BEV),按 State 和 Make 分组,需计算:

  • 平均续航(Electric Range 的均值)
  • 最早车型年份(Model Year 的最小值)
  • 每组车辆数量(即 count)

正确写法如下(推荐使用 pd.NamedAgg 显式声明,语义清晰且兼容性好):

import pandas as pd

# 假设 data 已加载为 DataFrame
filt = (data["Model Year"] >= 2018) & \
       (data["Electric Vehicle Type"] == "Battery Electric Vehicle (BEV)")

result = (data[filt]
          .groupby(["State", "Make"], sort=False, observed=True, as_index=False)
          .agg(
              avg_electric_range=pd.NamedAgg(column="Electric Range", aggfunc="mean"),
              oldest_model_year=pd.NamedAgg(column="Model Year", aggfunc="min"),
              count=pd.NamedAgg(column="Model Year", aggfunc="count")  # ✅ 安全:Model Year 为非空 int64,无 NaN 风险
          ))

? 为什么选 Model Year 做 count?
因其类型为 int64 且筛选后无缺失值(filt 已确保 Model Year >= 2018),故 count 统计的是该组有效记录数(等价于 size())。若某列含 NaN,count 会自动忽略缺失值,而 size() 返回总行数(含 NaN),二者语义不同,需按需选择。

你也可以用任意非空列(如 "VIN (1-10)" 或 "DOL Vehicle ID")替代 Model Year,只要确保该列在筛选后无缺失即可:

# 等效写法(使用 VIN 列计数)
count=pd.NamedAgg(column="VIN (1-10)", aggfunc="count")

✅ 输出结果将包含 State、Make、avg_electric_range、oldest_model_year 和新增的 count 列,结构规整,可直接用于后续过滤(例如:result[result['count'] >= 10] 筛出高频品牌-州组合)。

⚠️ 注意事项:

  • 避免使用 .size() 方法返回 Series 后再 reset_index(name='count'),虽可行但冗余,破坏链式表达;
  • 不要误用 agg('count')(未指定 column)——这会尝试对所有列计数,引发 KeyError;
  • 若需严格计数(含 NaN 行),应改用 size(作为聚合函数需配合 lambda x: len(x) 或通过 groupby(...).size().reset_index(name='count')),但本场景中 count 更符合“有效样本数”的业务含义。

综上,在 agg() 中并列添加 pd.NamedAgg(column=..., aggfunc='count') 是最简洁、高效、可读性强的标准解法,完全遵循 Pandas 最佳实践,兼顾性能与可维护性。

PHP速学视频免费教程(入门到精通)
PHP速学视频免费教程(入门到精通)

PHP怎么学习?PHP怎么入门?PHP在哪学?PHP怎么学才快?不用担心,这里为大家提供了PHP速学教程(入门到精通),有需要的小伙伴保存下载就能学习啦!

下载

相关标签:

本站声明:本文内容由网友自发贡献,版权归原作者所有,本站不承担相应法律责任。如您发现有涉嫌抄袭侵权的内容,请联系admin@php.cn

相关专题

更多
Python 时间序列分析与预测
Python 时间序列分析与预测

本专题专注讲解 Python 在时间序列数据处理与预测建模中的实战技巧,涵盖时间索引处理、周期性与趋势分解、平稳性检测、ARIMA/SARIMA 模型构建、预测误差评估,以及基于实际业务场景的时间序列项目实操,帮助学习者掌握从数据预处理到模型预测的完整时序分析能力。

2025.12.04

320

15

Python 数据清洗与预处理实战
Python 数据清洗与预处理实战

本专题系统讲解 Python 在数据清洗与预处理中的核心技术,包括使用 Pandas 进行缺失值处理、异常值检测、数据格式化、特征工程与数据转换,结合 NumPy 高效处理大规模数据。通过实战案例,帮助学习者掌握 如何处理混乱、不完整数据,为后续数据分析与机器学习模型训练打下坚实基础。

2026.01.31

232

21

Python数据分析与Pandas高级实战
Python数据分析与Pandas高级实战

本专题围绕 Python 数据分析展开,系统讲解 Pandas 的高级用法,包括数据清洗、透视表、时间序列分析以及多表合并与分组操作。通过实战案例,帮助开发者掌握高效处理与分析数据的方法,提高数据处理效率与分析能力。

2026.04.13

451

25

Kratos框架Protobuf接口定义与代码生成合集
Kratos框架Protobuf接口定义与代码生成合集

本专题讲解Kratos框架接口定义体系,涵盖proto编写规范、proto add/client/server生成命令、http注解路由、validate校验、OpenAPI文档生成、跨服务proto复用与兼容性设计。

2026.10.10

0

15

C++虚函数怎么定义和调用
C++虚函数怎么定义和调用

C++虚函数是实现运行时多态的重要机制。本专题从virtual关键字的基本用法入手,介绍基类与派生类之间的函数重写、基类指针调用派生类方法,以及动态绑定的执行过程,帮助初学者掌握虚函数的核心语法。

2026.10.10

0

26

C++类与对象的封装方法教程
C++类与对象的封装方法教程

C++封装是面向对象编程的核心特性之一,通过类将数据与操作数据的函数组织在一起,并利用访问权限控制外部访问。本专题介绍类的定义、成员变量、成员函数以及public、private和protected的使用方法,帮助初学者掌握封装的基本原理。

2026.10.10

0

32

C++构造函数定义与调用方法
C++构造函数定义与调用方法

C++构造函数用于初始化类对象,是面向对象编程的重要基础。本专题从构造函数的定义、声明和调用入手,介绍默认构造函数、带参数构造函数、拷贝构造函数及成员初始化列表,帮助初学者掌握对象创建与初始化的基本方法。

2026.10.10

0

16

Kratos框架零基础入门教程
Kratos框架零基础入门教程

本专题整理Kratos框架入门内容,涵盖Go环境准备、kratos CLI安装升级、new命令创建项目、目录结构分层说明、服务启动与双协议端口、依赖下载报错排查,帮助开发者快速跑通第一个Kratos框架微服务应用。

2026.10.10

0

15

C++条件判断语句怎么写
C++条件判断语句怎么写

C++条件判断是控制程序执行流程的重要基础。本专题介绍if、if-else、else if和switch等常见分支语句,结合条件表达式、比较运算符与代码示例,帮助初学者掌握不同场景下的判断逻辑。

2026.10.10

0

13

热门下载

更多
网站特效
/
网站源码
/
网站素材
/
前端模板

精品课程

更多
热门推荐
/
最新课程
phpStudy极速入门视频教程
phpStudy极速入门视频教程

共6课时 | 54.6万人学习

独孤九贱(4)_PHP视频教程
独孤九贱(4)_PHP视频教程

共89课时 | 133.4万人学习