<分区>
我有一个具有以下总体布局的数据框:
id,ind_1,ind_2_ind_3
1,0,1,0
1,1,0,0
2,0,1,0
2,0,0,1
3,0,0,1
3,1,0,0
我想添加一个额外的列，其值为“1”时的原始指标名称，应该如下所示:
id,ind_1,ind_2,ind_3,ind_all
1,0,1,0,ind_2
1,1,0,0,ind_1
2,0,1,0,ind_2
2,0,0,1,ind_3
3,0,0,1,ind_3
3,1,0,0,ind_1
欢迎任何提示!
最佳答案
你需要
df['ind_all'] = (df.iloc[:, 1:] == 1).idxmax(1)
id ind_1 ind_2 ind_3 ind_all
0 1 0 1 0 ind_2
1 1 1 0 0 ind_1
2 2 0 1 0 ind_2
3 2 0 0 1 ind_3
4 3 0 0 1 ind_3
5 3 1 0 0 ind_1
关于 python Pandas : How to Reverse One-Hot Encoding Back to Categorical，我们在Stack Overflow上找到一个类似的问题： https://stackoverflow.com/questions/49372640/