Python 选择列中值为1的列名

我有这个数据框：

    Course_2013     AP  Advanced    Vocational  Basic   ESL 
0   7th Math Enrich  0    1             0         0      0   
1   8th Math Enrich  0    1             0         0      0   
2   8th Science Enr  0    1             0         0      0   
3   Accounting I     0    0             0         1      0               
4   Accounting II    0    0             0         0      1   

我需要创建一个名为“category”的新列，该列在包含1的每行中包含列名称：

    Course_2013     AP  Advanced    Vocational  Basic   ESL category
0   7th Math Enrich  0    1             0         0      0   Advanced
1   8th Math Enrich  0    1             0         0      0   Advanced
2   8th Science Enr  0    1             0         0      0   Advanced
3   Accounting I     0    0             0         1      0   Basic              
4   Accounting II    0    0             0         0      1   ESL

---

让我们使用

apply

和

idxmax

：

df['category'] = df.iloc[:,1:].apply(lambda x: x.idxmax(), axis=1)

输出：

       Course_2013  AP  Advanced  Vocational  Basic  ESL  category
0  7th Math Enrich   0         1           0      0    0  Advanced
1  8th Math Enrich   0         1           0      0    0  Advanced
2  8th Science Enr   0         1           0      0    0  Advanced
3     Accounting I   0         0           0      1    0     Basic
4    Accounting II   0         0           0      0    1       ESL