通过预定义规则将字符串列编码为数值

Question

我有以下DataFrame：

   voting_result project   scene
    Bad          ccus      345943
    Good         ccus      311129
    Bad          ccus      309082
    Bad          ccus      331613
    Good         ccus      331615 
    Not Sure     ccus      331616   

通过遵循规则{'Bad':0, 'Good':1, 'Not Sure':2 }并获得以下结果的Pandas方法是将表决结果列encode为整数

voting_result    project    scene
        0         ccus      345943
        1         ccus      311129
        0         ccus      309082
        0         ccus      331613
        1         ccus      331615 
        2         ccus      331616

谢谢

Answer 1

按dictionary使用map ：

d = {'Bad':0, 'Good':1, 'Not Sure':2}
df['voting_result'] = df['voting_result'].map(d)
print (df)
   voting_result project   scene
0              0    ccus  345943
1              1    ccus  311129
2              0    ccus  309082
3              0    ccus  331613
4              1    ccus  331615
5              2    ccus  331616