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

Question

我有以下数据帧：

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} 将列encodevoting_result为整数并获得以下资源Pandas方法是什么

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