Python groupby 和 rank 函数排序错误：count 变量缺失

在使用 groupby 对数据进行分组排序时，遇到 count 变量不存在的错误。

错误分析

rank 函数用于对数据进行排名，需要指定排序依据的列名。错误提示表明 ok 数据中缺少名为 count 的列。

立即学习“Python免费学习笔记（深入）”；

macOS UI 自动化技能包 —— 用 Python 控制你的 Mac

让 macOS 自动化像说话一样简单 —— 用代码控制你的 Mac，无需学习 AppleScript 系统要求 macOS 14+ (Sonoma) Python 3.9+ 需要开启：辅助功能 + 屏幕录制权限

下载

解决方案

确认 ok 数据中确实没有 count 列后，可采用如下方法解决：

在对 ok 数据应用 groupby 和 rank 之前，需要先创建 count 列。 创建方法取决于 count 列的含义，例如，如果 count 代表每个 DISTRICT 的记录数，则可以使用如下代码：

ok['count'] = ok.groupby('DISTRICT')['DISTRICT'].transform('count')
ok['rank'] = ok.groupby('DISTRICT')['count'].transform(lambda x: x.rank(method='first', ascending=False).astype(int))

这段代码首先使用 groupby('DISTRICT')['DISTRICT'].transform('count') 计算每个 DISTRICT 的记录数，并将结果存储在新的 count 列中。然后，使用 groupby('DISTRICT')['count'].transform(...) 对每个 DISTRICT 内的 count 值进行排名，method='first' 保证相同值排名一致，ascending=False 表示降序排列，astype(int) 将排名转换为整数。

如果 count 列的含义不同，需要根据实际情况修改创建 count 列的代码。 确保在使用 rank 函数前，count 列已正确创建并包含有效数据。 这样就能避免 "count 变量不存在" 的错误。