Python Dataframe 将 bool 列数据合并为一列数据

Question

我有一个包含多列的数据框。我想将列合并为一列数据。

我的代码:

df = 
     A   foo   goo
0   10   Y     NaN
1   40   NaN   Y
2   80   Y     NaN

预期答案:

df = 
     A   Group   
0   10   foo     
1   40   goo   
2   80   foo 

我的方法:

df['foo'].replace('Y','foo',inplace=True)
df['goo'].replace('Y','goo',inplace=True)
df['Group'] = df['foo']+df['goo']
df =     
     A   foo   goo   Group
0   10   foo   NaN   NaN
1   40   NaN   goo   NaN
2   80   foo   NaN   NaN

在我的回答中，所有元素都变成了 NaN。

Answer 1

我们可以在 axis=1 上尝试 idxmax

df['Group'] = df.drop('A', 1).eq('Y').idxmax(1)

---

    A  foo  goo Group
0  10    Y  NaN   foo
1  40  NaN    Y   goo
2  80    Y  NaN   foo