2017-06-01 60 views
1

我有以下數據框:編碼字符串列數值通過預定義的規則

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

什麼是Pandas方式通過以下規則{'Bad':0, 'Good':1, 'Not Sure':2}和encode voting_result列整數得到以下resut

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

謝謝

回答

5

使用map作者:dictionary

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