Pandas - 是否有一种“开箱即用”的方式来组合表示枚举数据的布尔列？

Question

是否有库函数可以进行以下转换：      Enum    Column  Vals     =>       NewCol     -----------------------   =>      --------      True    False   False    =>        Enum      False   True    False    =>       Column      False   False   True     =>        Vals我确信我可以编写自己的函数来完美地完成此操作。感觉应该已经有一个库函数能够做到这一点，但我在文档中看不到。

Answer 1

假设每True行总是有 1 个，您可以使用DataFrame.idxmaxover columns：

df['NewCol'] = df.idxmax('columns')

[出去]

    Enum  Column   Vals  NewCol

0   True   False  False    Enum

1  False    True  False  Column

2  False   False   True    Vals

或者，您可以使用DataFrame.dot做点积：

df['NewCol'] = df.dot(df.columns)

Answer 2

这应该有效：

df.join(df.stack().loc[df.stack()==True].reset_index()[['level_1']]).rename(columns={'level_1':'New_Col'})

Answer 3

您可以融化然后查询数据框：

df['NewCol'] = (

    df.reset_index()

      .melt(id_vars=['index'])

      .query('value')

      .set_index('index')

      .drop(columns='value')

)