将1列数据转换为多重热编码

时间:2019-04-16 08:54:55

标签: python pandas dataframe binary dummy-variable

以该问题为例,假设我们有一个数据框:

    Name Class
0   Aci  FB 
1   Dan  TWT
2   Ann  GRS
3   Aci  GRS
4   Dan  FB 

结果数据框为   df

   Name  FB   TWT  GRS
0  Aci   1    0   1
0  Dan   1    1   0
0  Ann   0    0   1

2 个答案:

答案 0 :(得分:0)

get_dummiesDataFrame.set_index一起使用并汇总secondMethod()max

sum

答案 1 :(得分:0)

您也可以为此使用Series.str.get_dummiesDataFrame.groupby

df.Class.str.get_dummies().groupby(df['Name']).max().reset_index()

[出]

  Name  FB  GRS  TWT
0  Aci   1    1    0
1  Ann   0    1    0
2  Dan   1    0    1