python - 有没有办法使用熊猫将多个标志列汇总为一个？

Question

我在数据框中有这些列：

a   b   c   d   x
0   1   0   0   kevin
1   0   0   0   john
0   0   1   0   mary
1   0   0   0   jessica
0   0   0   1   joe

我需要将前四列变成只有一列。像这样的东西：

new_col   x 
b         kevin
a         john
c         mary
a         jessica
d         joe

有熊猫功能吗？

Answer 1

与dot

df['new'] = df.loc[:,'a':'d'].dot(df.columns[:-1])
Out[306]: 
0    b
1    a
2    c
3    a
4    d
dtype: object

如果每行只有一个 1 可以检查idxmax

df.loc[:,'a':'d'].idxmax(1)
Out[307]: 
0    b
1    a
2    c
3    a
4    d
dtype: object