Pandas有序分类排序技巧

本文深入解析了如何用 pandas 的 `pd.Categorical` 实现真正可靠、可控的自定义字符串排序——尤其适用于订单状态、评分等级、月份等具有明确业务顺序的字段；强调必须显式指定 `categories` 列表和 `ordered=True`，才能避免 `map/argsort` 等常见方案因缺失值、未映射项或大小写问题导致的排序错乱，同时揭示了 `astype('category')` 的陷阱、NaN 处理细节、安全转回字符串的最佳实践，以及 Categorical 在分组、绘图和链式操作中不可替代的一致性优势。

用 Categorical 实现固定顺序排序，不是靠 sort_values(key=...)

直接用 sort_values 配合自定义函数很难控制“完全按你列的顺序排”，尤其当数据里有缺失值、新类别或大小写混杂时。正确路径是先转成 pd.Categorical，再让 pandas 按分类的内部序数排序——这才是真正可控的有序分类。

常见错误现象：sort_values(key=lambda x: x.map({'low':0,'mid':1,'high':2})) 一遇到 NaN 或未映射的值就变 NaN，整列排乱；或者用 reindex 强行对齐，但原始数据顺序被破坏。

• 必须显式传入 categories 参数，且顺序即排序顺序，比如 ['draft', 'review', 'approved', 'rejected']
• ordered=True 是关键开关，不设它，Categorical 只是分组工具，不支持比较和排序
• 原始列中不存在的值会变成 NaN（除非加 errors='ignore'，但不推荐）

astype('category') 不行，必须用 pd.Categorical 构造

df['status'].astype('category') 只是把列转成无序分类，内部顺序按首次出现顺序定，跟你想排的业务顺序无关。要强制指定顺序，得手动构造 Categorical 对象再塞回去。

使用场景：状态字段（如订单状态、工单阶段）、评分等级（'S','A','B','C'）、月份缩写（'Jan','Feb',...）等有明确业务先后关系的字符串列。

• 正确写法：df['status'] = pd.Categorical(df['status'], categories=['draft','review','approved'], ordered=True)
• 之后直接 df.sort_values('status') 就按你给的列表顺序排，无需额外参数
• 如果原始列含空值，Categorical 默认保留为 NaN，排序时自动排在最前（pandas 1.4+ 默认行为），如需排最后，加 na_position='last'

排序后想还原成字符串？别用 astype(str)，用 astype('string')

排序完如果立刻 astype(str)，会把 NaN 变成字符串 'nan'，而且丢失分类信息；更糟的是，如果后续还要再排一次，又得重做 Categorical 构造。

性能影响：反复构造 Categorical 开销不小，尤其大数据量。最好只在排序前构造一次，排序后保持类型，仅在导出或展示时临时转。

• 安全转回字符串：df['status'].astype('string')（返回 nullable string 类型，NaN 仍是 NaN）
• 如果必须用普通 str，先填空：df['status'].fillna('unknown').astype(str)
• 兼容性注意：'string' 类型要求 pandas ≥ 1.0，老版本只能用 astype(object) + fillna

和 map + argsort 对比：为什么不用纯 NumPy 方案？

有人用 df.iloc[df['level'].map({'L1':0,'L2':1,'L3':2}).argsort()] 硬算索引，看似简洁，但隐患多：映射字典漏项 → NaN → argsort 把 NaN 当最小值；大小写不敏感需求难处理；无法复用排序逻辑做分组或可视化。

容易踩的坑：这种写法在 groupby 后再排序会失效，因为 map 返回的是 Series，argsort 是全局索引，打乱了分组内顺序。

• Categorical 天然支持 groupby(...).apply(lambda g: g.sort_values('x'))，顺序不丢
• 绘图时（如 seaborn 的 catplot），ordered=True 分类能自动按序排横轴
• 如果真要高性能硬解，至少用 pd.factorize 配合 categories 映射表，而不是裸 map

终于介绍完啦！小伙伴们，这篇关于《Pandas有序分类排序技巧》的介绍应该让你收获多多了吧！欢迎大家收藏或分享给更多需要学习的朋友吧~golang学习网公众号也会发布文章相关知识，快来关注吧！