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

假设我们有如下dataframe,名字为student_weight:

将字符型分类变量数据转换为数字格式_第1张图片

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”列如下:

将字符型分类变量数据转换为数字格式_第2张图片

 

 

 

你可能感兴趣的:(数据挖掘)