pandas 为什么要用 Category 类型而不是 Str？

__Lingyue__

于 2020-04-21 00:10:58 发布

阅读量1.2k

点赞数 1

分类专栏：数据处理文章标签： pandas 机器学习

本文链接：https://blog.youkuaiyun.com/cFarmerReally/article/details/105648222

版权

当数据列的唯一值较少且需要排序时，使用pandas的Category类型而非Str可以提升groupby操作速度并节省内存。Category类型通过存储index优化内存使用，并支持有序操作。了解更多详情，请参阅官方文档。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

category是一种pandas的类型

pd.Series(["a", "b", "c", "a"], dtype="category")
pd.Series(["a", "b", "c", "a"], dtype="str")

之前自己的代码都是用str，或者直接就是python自己的object类型，那么我们什么时候需要用category而不是str呢？

当这一列的值的种类(number of unique)非常少的时候，当需要给这一列的值表示某一个顺序的时候，用category。

在做groupby操作的时候可以提速
可以节省内存
支持order

为什么可以节省内存和提速groupby呢？
猜测是如果你表明了category，它存储的其实是index，而不是每个值，进行groupby的时候不需要先进行sort操作，所以可以提速和节省内存，这也是它建议你的值的种类比较少的时候用。

官方文档：

This is an introduction to pandas categorical data type, including a short comparison with R’s factor.

Categoricals are a pandas data type corresponding to categoric

最低0.47元/天解锁文章

pandas 为什么要用 Category 类型 而不是 Str？

pandas 为什么要用 Category 类型而不是 Str？