2017-10-24 10 views
0

私はPython Pandas Dataframeを使用していて、データセット内の各行の列のリストを印刷しようとしていますが、各列は0または1の値を持つことができます。例えば:私は関係なく、列名または番号の一般的な機能を準備する必要がありますのでご注意ください特定の条件でPandasデータフレームの各行の列名を表示

id output 
0 A,B,C,D 
1 B,D 
2 A,B 
3 A 

id A B C D 
0 1 1 1 1 
1 0 1 0 1 
2 1 1 0 0 
3 1 0 0 0 

は今、私は私の出力をする必要があります。

答えて

3

あなたは行うことができます。

df = df.assign(output=df.dot(df.columns)) 

df[['output']] 

id output 
0 A,B,C,D 
1 B,D 
2 A,B 
3 A 
+0

素晴らしい答えを! –

+0

これをupvoteしないでください。私が旗を立てた答えのバリエーションです。 – GiantsLoveDeathMetal

+0

ああ、あなたは結局最高のことから学ぶよ。 –

関連する問題