為數據框中的每一行收集值為 True 的列名

Question

我有一個看起來像這樣的數據框：

**col_A col_B col_C**
False True  False
True  False False
False False True
False False False

我需要收集每一行的值為 True 的列名並創建另一個數據框：

**col**
col_B
col_A
col_C
nan

注意：一行中最多有一個 True 值。

Answer 1

在這里你用idxmax和where ：

df.idxmax(1).where(df.any(1))

輸出：

0    col_B
1    col_A
2    col_C
3      NaN
dtype: object