本文详解Pandas中groupby作用于全量DataFrame与子集列的差异，指出对无关列执行nunique()会造成性能浪费，并推荐最简洁、高效且语义清晰的写法：df.groupby(key)[target_col].nunique()。
本文详解pandas中`groupby`作用于全量dataframe与子集列的差异，指出对无关列执行`nunique()`会造成性能浪费，并推荐最简洁、高效且语义清晰的写法：`df.groupby(key)[target_col].nunique()`。
在实际数据分析工作中，我们常需按某列分组后统计目标列的唯一值数量（如统计每个部门有多少种职级）。此时，一个常见但易被忽视的性能陷阱是：在groupby前或后对非必要列执行聚合运算。
例如，以下写法看似无害，实则低效：
# ❌ 不推荐：先选取子集再聚合，且对所有列调用 nunique()
df[cols].groupby('dept').nunique()['job_level']
# ❌ 更差：对整个 DataFrame 调用 nunique()，再索引目标列
df.groupby('dept').nunique()['job_level']这两者虽在结果上可能一致（当cols == ['dept', 'job_level']时），但存在本质问题：
- nunique() 会对 groupby 后的每一列分别计算唯一值个数；
- 若 df 有 100 列，而你只关心 'job_level'，那么其余 99 列的 nunique 计算完全是冗余开销；
- 这不仅拖慢执行速度，还增加内存占用，尤其在大数据集上影响显著。
✅ 正确做法是将列筛选逻辑前置到聚合操作内部，即直接指定目标列参与聚合：
# ✅ 推荐：语义清晰、性能最优
result = df.groupby('dept')['job_level'].nunique()
# ✅ 等价写法（显式强调 Series 分组）
result = df['job_level'].groupby(df['dept']).nunique()这种写法的优势在于：
- 计算精准：仅对 'job_level' 列执行去重计数，零冗余；
- 可读性强：明确表达了“按 dept 分组，统计 job_level 的不重复数量”这一业务意图；
- 兼容性好：支持 some_column 为 DataFrame 列名或外部 Series（如 df['dept'].map(region_map)）；
- 返回类型友好：默认返回 pd.Series，索引为分组键，值为对应唯一值数量，便于后续链式操作（如 .sort_values() 或 .to_frame()）。
⚠️ 注意事项：
- 若 cols 包含多余列（如 ['dept', 'job_level', 'salary']），df[cols].groupby(...).nunique() 仍会计算 salary 的唯一值——除非你确实需要它，否则纯属浪费；
- 避免链式调用中多次 .groupby() 或 .nunique()，应优先使用 groupby(...)[col].agg(...) 模式统一入口；
- 对于多列聚合需求（如同时统计 job_level 和 team 的唯一数），可用元组解包或字典映射：
df.groupby('dept')[['job_level', 'team']].nunique() # 或 df.groupby('dept').agg({'job_level': 'nunique', 'team': 'nunique'})
总结：Pandas 的链式操作虽灵活，但“先切片再聚合”并非银弹。真正的优化不在于减少代码行数，而在于让计算严格聚焦于业务目标列。坚持 df.groupby(key)[target].method() 这一范式，既是性能最佳实践，也是代码可维护性的关键保障。
