将字符型分类变量数据转换为数字格式

最新推荐文章于 2024-07-06 03:01:05 发布

Mr.Daozhi

最新推荐文章于 2024-07-06 03:01:05 发布

阅读量3.8k

点赞数

分类专栏：数据挖掘文章标签：数据结构

本文链接：https://blog.csdn.net/alphachx/article/details/105240727

版权

数据挖掘专栏收录该内容

22 篇文章 0 订阅

订阅专栏

假设我们有如下dataframe，名字为student_weight：

ordered_height = ['Short', 'Medium', 'Tall']#定义顺序，这里是这个次序可以对应，0：short；1,：Medium；2：Tall；

最后一列Out1是字符型分类变量，无法直接被python函数使用，需要进行transform。步骤如下：

student_weight_target=pd.DataFrame(student_weight["Out1"])#提取“Out1”列；
#利用astype将该列变为数字格式；
st=student_weight_target.Out1.astype("category",  ordered=True,  categories=ordered_height).cat.codes
student_weight_target["o1"]=pd.DataFrame(st)

重新transform的dataframe中的“o1”列如下：