Pandas DataFrame：バイナリ列を1つのカテゴリ型列に変換する方法は？

Question

パンダのDataFrameがあると、いくつかのバイナリ列（1は値が存在することを示し、0はそうでないことを示します）を1つのカテゴリ型の列に変換する方法はありますか？

「逆pd.get_dummies()」を実行する方法はもう1つ考えてください。ここで

は、いくつかのバイナリ列にカテゴリ列を変換する例です：

import pandas as pd 
s = pd.Series(list('ABCDAB')) 
df = pd.get_dummies(s) 
df 
    A B C D 
0 1 0 0 0 
1 0 1 0 0 
2 0 0 1 0 
3 0 0 0 1 
4 1 0 0 0 
5 0 1 0 0 

私が達成したい何

df1 
    A B C D 
0 1 0 0 0 
1 0 1 0 0 
2 0 0 1 0 
3 0 0 0 1 
4 1 0 0 0 
5 0 1 0 0 

が

にそれをI変換んでしたデータフレームを与えています

df1 
    A B C D category 
0 1 0 0 0 A 
1 0 1 0 0 B 
2 0 0 1 0 C 
3 0 0 0 1 D 
4 1 0 0 0 A 
5 0 1 0 0 B 

Answer 1

1つの方法は、idxmaxを使用して1秒を検索することです。

In [32]: df["category"] = df.idxmax(axis=1) 

In [33]: df 
Out[33]: 
    A B C D category 
0 1 0 0 0  A 
1 0 1 0 0  B 
2 0 0 1 0  C 
3 0 0 0 1  D 
4 1 0 0 0  A 
5 0 1 0 0  B